From 96ab3ab2f586be54e6af367fc0bbbbbd8ee13926 Mon Sep 17 00:00:00 2001
From: Rosyid Haryadi <rosyid_haryadi@protonmail.com>
Date: Sun, 28 Jul 2024 00:24:41 +0700
Subject: fix handle connection problem

---
 bot.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

(limited to 'bot.py')

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
-- 
cgit v1.2.3-70-g09d2