diff --git a/dislocker.py b/dislocker.py index aec219a..f0ae3b9 100644 --- a/dislocker.py +++ b/dislocker.py @@ -242,12 +242,11 @@ class Bot(discord.Client): def keyboard_used_check(self, **kwargs): try: + cursor = dislocker.db.cursor() if kwargs["keyboard_number"] == None: return {"result": 0, "about": "ok"} else: keyboard_number = int(kwargs["keyboard_number"]) - - cursor = dislocker.db.cursor() cursor.execute("SELECT * FROM keyboard_list WHERE keyboard_number=%s", (keyboard_number,)) keyboard_list_record = cursor.fetchall() @@ -269,12 +268,11 @@ class Bot(discord.Client): def mouse_used_check(self, **kwargs): try: + cursor = dislocker.db.cursor() if kwargs["mouse_number"] == None: return {"result": 0, "about": "ok"} else: mouse_number = int(kwargs["mouse_number"]) - - cursor = dislocker.db.cursor() cursor.execute("SELECT * FROM mouse_list WHERE mouse_number=%s", (mouse_number,)) mouse_list_record = cursor.fetchall()