add database calls + some bugfixes

db.py

@@ -2,7 +2,18 @@ class DuckDB:
     def __init__(self, location=None):
         self.location = location
         self.db = []
-        if location != None: read(location)
+        if location != None: self.read(location)
+
+    def add(self, state, nick, abstime, reltime, channel):
+        self.db.append(DuckEvent(
+            "{}{} {} {} {}".format(
+                state,
+                nick,
+                str(abstime),
+                str(reltime),
+                channel,
+            )
+        ))
 
     def parse(self, fd):
         lines = [i.rstrip() for i in fd.readlines()]
@@ -21,7 +32,7 @@ class DuckDB:
     def write(self, location):
         fd = open(location, "w")
         self.output(fd)
-        fd.close
+        fd.close()
 
 class DuckEvent:
     def __init__(self, line=None):

main.py

@@ -2,6 +2,9 @@ import asyncio
 import random
 import time
 
+from db import DuckDB
+from db import DuckEvent
+
 from irctokens import build, Line
 from ircrobots import Bot as BaseBot
 from ircrobots import Server as BaseServer
@@ -19,33 +22,47 @@ class Server(BaseServer):
     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):
         self.messages = 0
         self.duckactive = True
         self.duckactivetime = time.time()
         await self.msgall(lang["duck"])
+
     async def duck_test(self):
         if self.messages > 1 and random.randint(0, 99) < 10: await self.new_duck()
+
     async def misstime(self):
         return format(time.time() - self.lastduck, '.2f')
+
     async def coughttime(self):
         return format(self.lastduck - self.duckactivetime, '.2f')
+
     async def duck_action(self, user, chan):
+        db = DuckDB(self.db)
         if self.duckactive:
             self.duckactive = False
             self.messages = 0
             self.lastduck = time.time()
-            await self.msgall(lang["duckcought"].format(user, chan, self.coughttime())))
+            await self.msgall(lang["duckcought"].format(user, chan, await self.coughttime()))
+            db.add("B", user, time.time(), float(await self.coughttime()), chan)
         elif self.lastduck != 0:
-            await self.msg(chan, lang["noduck"].format(self.coughttime())), user)
+            await self.msg(chan, lang["noduck"].format(await self.misstime()), user)
+            db.add("M", user, time.time(), float(await self.misstime()), chan)
         else:
             await self.msg(chan, lang["noduckstart"], user)
+            db.add("M", user, time.time(), -1, chan)
+        db.write(self.db)
+
     async def line_read(self, line: Line):
         print(f"{self.name} < {line.format()}")
         if line.command == "001":
@@ -62,9 +79,10 @@ class Server(BaseServer):
                 return
 
             self.messages += 1
-            self.duck_test()
+            await self.duck_test()
         elif line.command == "INVITE":
             await self.send(build("JOIN", [line.params[1]]))
+
     async def line_send(self, line: Line):
         print(f"{self.name} > {line.format()}")