From 2932a87da39a8832a9c4c99dd2ca4828be9e6b92 Mon Sep 17 00:00:00 2001
From: Hykilpikonna <me@hydev.org>
Date: Sun, 22 Nov 2020 15:41:55 -0500
Subject: [PATCH] [O] Specify types

---
 src/bot.py      |  2 +-
 src/commands.py | 21 ++++++++++++---------
 2 files changed, 13 insertions(+), 10 deletions(-)

diff --git a/src/bot.py b/src/bot.py
index 0c1e367..2656a44 100644
--- a/src/bot.py
+++ b/src/bot.py
@@ -19,7 +19,7 @@ if __name__ == '__main__':
     print("Bot created: ", bot.getMe())
 
     # Create updater
-    updater = Updater(bot=bot)
+    updater = Updater(bot=bot, use_context=True)
     dispatcher = updater.dispatcher
 
     # Register commands
diff --git a/src/commands.py b/src/commands.py
index c9279f8..e5b7def 100644
--- a/src/commands.py
+++ b/src/commands.py
@@ -1,6 +1,9 @@
 import re
 from io import BytesIO
 
+from telegram import Bot, Update
+from telegram.ext import Updater, CallbackContext
+
 from src.database import Database
 from src.utils import toJson, create
 
@@ -35,14 +38,14 @@ urlValidator = re.compile(
 database = Database()
 
 
-def start(update, context):
+def start(update: Update, context: CallbackContext):
     chat = update.effective_chat
     database.checkUser(chat.id)
 
     return helpMsg
 
 
-def ls(update, context):
+def ls(update: Update, context: CallbackContext):
     chat = update.effective_chat
     user = database.checkUser(chat.id)
     requests = database.userRequests[user]
@@ -50,7 +53,7 @@ def ls(update, context):
     return "Your requests: %s" % toJson(requests)
 
 
-def touch(update, context):
+def touch(update: Update, context: CallbackContext):
     chat = update.effective_chat
     user = database.checkUser(chat.id)
 
@@ -82,7 +85,7 @@ def touch(update, context):
     return "%s is successfully created!" % name
 
 
-def rm(update, context):
+def rm(update: Update, context: CallbackContext):
     chat = update.effective_chat
     user = database.checkUser(chat.id)
 
@@ -102,12 +105,12 @@ def rm(update, context):
     return "%s is successfully removed!" % name
 
 
-def nano(update, context):
+def nano(update: Update, context: CallbackContext):
     chat = update.effective_chat
     user = database.checkUser(chat.id)
 
 
-def test(update, context):
+def test(update: Update, context: CallbackContext):
     chat = update.effective_chat
     user = database.checkUser(chat.id)
 
@@ -130,16 +133,16 @@ def test(update, context):
     return 'Done!'
 
 
-def interval(update, context):
+def interval(update: Update, context: CallbackContext):
     chat = update.effective_chat
     user = database.checkUser(chat.id)
 
 
-def enable(update, context):
+def enable(update: Update, context: CallbackContext):
     chat = update.effective_chat
     user = database.checkUser(chat.id)
 
 
-def disable(update, context):
+def disable(update: Update, context: CallbackContext):
     chat = update.effective_chat
     user = database.checkUser(chat.id)