From 43a617394a75789321fc820f9f532a0dec700a66 Mon Sep 17 00:00:00 2001
From: Kylmakalle <iceman9831@gmail.com>
Date: Wed, 5 Jul 2017 21:54:27 +0300
Subject: [PATCH] Stability improvemements [2]

---
 bot.py         | 1 +
 vk_messages.py | 3 +--
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/bot.py b/bot.py
index 7da24e8..0bebc97 100644
--- a/bot.py
+++ b/bot.py
@@ -30,6 +30,7 @@ vk_tokens = redis.StrictRedis(connection_pool=tokens_pool)
 currentchat = {}
 
 bot = telebot.AsyncTeleBot(token)
+bot.remove_webhook()
 
 link = 'https://oauth.vk.com/authorize?client_id={}&' \
        'display=page&redirect_uri=https://oauth.vk.com/blank.html&scope=friends,messages,offline,docs,photos,video' \
diff --git a/vk_messages.py b/vk_messages.py
index 43c5477..df37641 100644
--- a/vk_messages.py
+++ b/vk_messages.py
@@ -4,7 +4,6 @@ import redis
 import requests
 import time
 import vk
-from vk.exceptions import VkAPIError
 import wget
 
 logging.basicConfig(format='%(levelname)-8s [%(asctime)s] %(message)s', level=logging.WARNING, filename='vk.log')
@@ -284,7 +283,7 @@ class VkMessage:
         api = vk.API(self.session)
         try:
             new = api.messages.getLongPollHistory(ts=self.ts, pts=self.pts)
-        except VkAPIError.code == 10:
+        except:
             timeout = 3
             logging.warning('Retrying getLongPollHistory in {} seconds'.format(timeout))
             time.sleep(timeout)