fix handle connection problem

1 files changed, 8 insertions, 4 deletions

diff --git a/bot.py b/bot.py
index b17a3bd..c66d6a0 100644
--- a/bot.py
+++ b/bot.py
@@ -1,5 +1,6 @@
 import os
 import sqlite3
+import urllib.error
 from urllib import parse
 import requests
 from dotenv import load_dotenv
@@ -59,10 +60,13 @@ class Bot:
     def pollUpdate(self):
         self.isProcessingPolling = True
 
-        url = f'https://api.telegram.org/bot{self.token}/getUpdates'
-        response = requests.get(url)
-        if response.status_code == 200:
-            self._dbUpdate(response.json())
+        try:
+            url = f'https://api.telegram.org/bot{self.token}/getUpdates'
+            response = requests.get(url)
+            if response.status_code == 200:
+                self._dbUpdate(response.json())
+        except requests.exceptions.ConnectionError as e:
+            logger.error("Connection error during polling update")
         self._checkUnreadDb()
 
         self.isProcessingPolling = False