move irc duck logic to DuckLogic

This commit is contained in:
randomuser 2021-07-21 00:20:03 -05:00
parent e49deed71a
commit 9dabdd7ac6
1 changed files with 17 additions and 16 deletions

33
main.py
View File

@ -17,21 +17,7 @@ lang = {
"duck": "・゜゜・。。・゜゜\_o< QUACK!", "duck": "・゜゜・。。・゜゜\_o< QUACK!",
} }
class Server(BaseServer): class DuckLogic:
messages = 0
duckactive = False
duckactivetime = 0
lastduck = 0
db = "duckdb"
async def msg(self, chan, msg, usr=None):
if usr != None:
await self.send(build("PRIVMSG", [chan, usr + ": " + msg]))
else: await self.send(build("PRIVMSG", [chan, msg]))
async def msgall(self, msg):
[await self.msg(channel, msg) for channel in self.channels]
async def new_duck(self): async def new_duck(self):
self.messages = 0 self.messages = 0
self.duckactive = True self.duckactive = True
@ -63,6 +49,21 @@ class Server(BaseServer):
db.add("M", user, time.time(), -1, chan) db.add("M", user, time.time(), -1, chan)
db.write(self.db) db.write(self.db)
class Server(BaseServer, DuckLogic):
messages = 0
duckactive = False
duckactivetime = 0
lastduck = 0
db = "duckdb"
async def msg(self, chan, msg, usr=None):
if usr != None:
await self.send(build("PRIVMSG", [chan, usr + ": " + msg]))
else: await self.send(build("PRIVMSG", [chan, msg]))
async def msgall(self, msg):
[await self.msg(channel, msg) for channel in self.channels]
async def line_read(self, line: Line): async def line_read(self, line: Line):
print(f"{self.name} < {line.format()}") print(f"{self.name} < {line.format()}")
if line.command == "001": if line.command == "001":
@ -75,7 +76,7 @@ class Server(BaseServer):
chan = line.params[0] chan = line.params[0]
user = line.hostmask.nickname user = line.hostmask.nickname
if cmd == "bef": await self.duck_action(user, chan) if cmd == "bef": await self.duck_action(user, chan)
if cmd == "trigger": await self.new_duck() elif cmd == "trigger": await self.new_duck()
return return
self.messages += 1 self.messages += 1