add user lookup
This commit is contained in:
parent
9b9d9d600c
commit
fa732a5075
1 changed files with 61 additions and 1 deletions
|
@ -1,5 +1,6 @@
|
|||
import logging
|
||||
import asyncio
|
||||
from typing import Dict
|
||||
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
|
@ -7,7 +8,7 @@ from discord.ext import commands
|
|||
from .common import Cog
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
UserID = int
|
||||
|
||||
async def shell(command: str):
|
||||
process = await asyncio.create_subprocess_shell(
|
||||
|
@ -20,10 +21,61 @@ async def shell(command: str):
|
|||
return f'{out}{err}'.strip()
|
||||
|
||||
|
||||
async def distance(s, len_s, t, len_t):
|
||||
cost = 0
|
||||
|
||||
if not len_s:
|
||||
return len_t
|
||||
|
||||
if not len_t:
|
||||
return len_s
|
||||
|
||||
if s[len_s - 1] == t[len_t - 1]:
|
||||
cost = 0
|
||||
else:
|
||||
cost = 1
|
||||
|
||||
l = [
|
||||
(await distance(s, len_s - 1, t, len_t)) + 1,
|
||||
(await distance(s, len_s, t, len_t - 1)) + 1,
|
||||
(await distance(s, len_s - 1, t, len_t - 1)) + cost
|
||||
]
|
||||
|
||||
return min(l)
|
||||
|
||||
|
||||
class Rsudo(Cog):
|
||||
def __init__(self, bot):
|
||||
super().__init__(bot)
|
||||
|
||||
async def lookupuser(self, uname: str) -> Dict[UserID, float]:
|
||||
lu = len(uname)
|
||||
|
||||
cchan = self.bot.get_channel(
|
||||
self.bot.config.command_channel
|
||||
)
|
||||
|
||||
if not cchan:
|
||||
return None
|
||||
|
||||
members = cchan.guild.members
|
||||
scores = {}
|
||||
for member in members:
|
||||
if member.bot: continue
|
||||
|
||||
lmn = len(member.name)
|
||||
d = await distance(member.name, lmn, uname, lu)
|
||||
scores[member.id] = d
|
||||
|
||||
if uname in member.name or uname in member.nick:
|
||||
scores[member.id] = -1
|
||||
|
||||
users = sorted(scores.keys(), key=lambda k: scores[k])
|
||||
if not users:
|
||||
return None
|
||||
|
||||
return users[0]
|
||||
|
||||
async def create_request(self, message):
|
||||
command_channel = self.bot.get_channel(
|
||||
self.bot.config.command_channel
|
||||
|
@ -120,6 +172,14 @@ class Rsudo(Cog):
|
|||
except discord.HTTPException:
|
||||
await command_channel.send(f'Failed to send denial message to {user}.')
|
||||
|
||||
@commands.command()
|
||||
async def lookup(self, ctx, uname: str):
|
||||
u = await self.lookupuser(uname)
|
||||
if not u:
|
||||
return await ctx.send('none found')
|
||||
|
||||
u = self.bot.get_user(u)
|
||||
await ctx.send(repr(u))
|
||||
|
||||
def setup(bot):
|
||||
bot.add_cog(Rsudo(bot))
|
||||
|
|
Loading…
Reference in a new issue