103 lines
2.7 KiB
Python
Executable file
103 lines
2.7 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
|
|
import pydle, asyncio, dataset, sys, os, time
|
|
|
|
class Kim(pydle.Client):
|
|
async def on_connect(self):
|
|
print('Connected!')
|
|
|
|
self.modules = {}
|
|
self.cmd = {}
|
|
self.rawm = {}
|
|
self.help = {}
|
|
self.db = dataset.connect('sqlite:///database.db')
|
|
self.t=0
|
|
|
|
print('loading modules...')
|
|
await self.loadMods()
|
|
print('joining channels')
|
|
for i in self.chansjoin:
|
|
await self.join(i)
|
|
print('Done!')
|
|
|
|
# tilde +B bot
|
|
await self.set_mode(self.nickname, '+B')
|
|
|
|
|
|
async def loadMods(self):
|
|
for i in [s for s in os.listdir('modules') if ".py" in s and '.swp' not in s]:
|
|
i = i[:-3]
|
|
print('loading', i)
|
|
m = __import__("modules."+i)
|
|
m = eval('m.'+i)
|
|
await m.init(self)
|
|
self.modules[i] = m
|
|
|
|
async def on_invite(self, channel, by):
|
|
await self.modules['invite'].invite(self, channel, by)
|
|
# print('{} invited me to {}!'.format(by, channel))
|
|
# self.t = time.time()+1
|
|
# await self.join(channel)
|
|
|
|
|
|
async def on_message(self, chan, source, msg):
|
|
if chan == self.nickname:
|
|
chan = source
|
|
if source != self.nickname:
|
|
|
|
|
|
if time.time() > self.t:
|
|
|
|
|
|
if msg == '!botlist':
|
|
await self.message(chan, 'helo im kim, a learning chatbot https://tildegit.org/xfnw/kim/')
|
|
await self.parseCommand(chan, source, msg)
|
|
for i in self.rawm:
|
|
await self.rawm[i](self, chan, source, msg)
|
|
|
|
async def parseCommand(self, chan, source, msg):
|
|
if msg[:len(self.prefix)] == self.prefix:
|
|
|
|
msg = msg[len(self.prefix):]
|
|
cmd = msg.split(' ')[0]
|
|
msg = msg[len(cmd)+1:]
|
|
if len(cmd) < 1:
|
|
return
|
|
|
|
if cmd in self.cmd:
|
|
await self.cmd[cmd](self, chan, source, msg)
|
|
return
|
|
|
|
# fuzzy search for commands
|
|
results = [i for i in self.cmd if i.startswith(cmd)]
|
|
if len(results) == 1:
|
|
await self.cmd[results[0]](self, chan, source, msg)
|
|
|
|
|
|
async def is_admin(self, nickname):
|
|
|
|
# Check the WHOIS info to see if the source has identified with NickServ.
|
|
# This is a blocking operation, so use yield.
|
|
info = await self.whois(nickname)
|
|
if 'account' in info:
|
|
account = info['account']
|
|
else:
|
|
# they are not nickserv registered
|
|
return False
|
|
|
|
return account in self.admins
|
|
|
|
#async def on_private_message(self, trash, source, msg):
|
|
# if source != self.nickname:
|
|
# for i in self.rawm:
|
|
# await self.rawm[i](self, source, source, msg)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
client = Kim('kim', realname='owens bot')
|
|
client.admins = ['lickthecheese', 'ben', 'coffeeowl', 'gbmor', 'tomasino', 'ubergeek', 'deepend', 'calamitous', 'khuxkm']
|
|
client.prefix = 'kim: '
|
|
client.run('team.tilde.chat', tls=True, tls_verify=False)
|
|
|