Python - Fix after refactor

2 files changed, 7 insertions, 8 deletions

diff --git a/src/blog_node5_net.py b/src/blog_node5_net.py
index 53a8f3e..96fceea 100644
--- a/src/blog_node5_net.py
+++ b/src/blog_node5_net.py
@@ -83,7 +83,7 @@ articles_by_url: typing.Dict[str, article.Article] = {}
 try:
     with open(os.path.join('configs', 'database.yml'), 'r') as file:
         db_conf  = yaml.safe_load(file.read())
-    db: DBHandler = DBHandler(db_conf)
+    db: db_handler.DBHandler = db_handler.DBHandler(db_conf)
 except FileNotFoundError as ex:
     logger.warning("Database config - Not found, running without")
     db = None
diff --git a/src/telegram_handler.py b/src/telegram_handler.py
index 6213445..4ccc83e 100644
--- a/src/telegram_handler.py
+++ b/src/telegram_handler.py
@@ -7,13 +7,12 @@ import yaml
 
 class Telegram:
     def __init__(self, config: dict):
-        config: dict = config
-        bot = telegram.Bot(config['token'])
-        breakpoint()
-        asyncio.run(init_bot())
+        self.config: dict = config
+        self.bot = telegram.Bot(config['token'])
+        asyncio.run(self.init_bot())
 
     def send_message(self, text):
-        asyncio.run(async_send_message(text))
+        asyncio.run(self.async_send_message(text))
 
 
     async def async_send_message(self, text):
@@ -22,6 +21,6 @@ class Telegram:
 
 
     async def init_bot(self):
-        async with bot:
-            print(await bot.get_me())
+        async with self.bot:
+            print(await self.bot.get_me())