Garmelon/euphoria-plusone
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Initialize before starting bot

Browse source
This commit is contained in:
Joscha 2017-09-10 12:54:52 +00:00
parent 8b976e038b
commit c53e0b6919
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
plusone.py
View file

@ -1,5 +1,6 @@
import yaboli import yaboli
from yaboli.utils import * from yaboli.utils import *
import asyncio
import re import re
import sys import sys
@ -66,10 +67,6 @@ class PlusOne(yaboli.Bot):
self.register_command("points", self.command_points, specific=False) self.register_command("points", self.command_points, specific=False)
self.register_trigger(self.PLUSONE_RE, self.trigger_plusone) self.register_trigger(self.PLUSONE_RE, self.trigger_plusone)
async def on_connected(self):
await super().on_connected()
await self.db.initialize()
async def trigger_plusone(self, message, match): async def trigger_plusone(self, message, match):
specific = re.match(self.MENTION_RE, match.group(2)) specific = re.match(self.MENTION_RE, match.group(2))
if specific: if specific:
@ -111,6 +108,7 @@ class PlusOne(yaboli.Bot):
def main(): def main():
if len(sys.argv) == 3: if len(sys.argv) == 3:
db = PointDB(sys.argv[2]) db = PointDB(sys.argv[2])
asyncio.get_event_loop().run_until_complete(db.initialize())
run_bot(PlusOne, sys.argv[1], db) run_bot(PlusOne, sys.argv[1], db)
else: else:
print("USAGE:") print("USAGE:")