From e77276ae52b3ce4993169698f1a1b7f4276ba9e6 Mon Sep 17 00:00:00 2001 From: Manuel Cortez Date: Mon, 3 Sep 2018 09:15:52 -0500 Subject: [PATCH] Fixed error when getting VK API error 5. Closes #11 --- src/sessionmanager/session.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/src/sessionmanager/session.py b/src/sessionmanager/session.py index 5c041bc..bf776e7 100644 --- a/src/sessionmanager/session.py +++ b/src/sessionmanager/session.py @@ -9,6 +9,7 @@ import utils import sound from config_utils import Configuration, ConfigurationResetException from pubsub import pub +from vk.exceptions import VkAPIError log = logging.getLogger("session") @@ -218,11 +219,15 @@ class vkSession(object): If the access_token has expired, it will call authorise() too, for getting a new access token.""" if self.settings["vk"]["token"] != None: - result = self.vk.login_access_token(self.settings["vk"]["token"]) - self.logged = True - log.debug("Logged.") - if result == False: - self.authorise() + try: + result = self.vk.login_access_token(self.settings["vk"]["token"]) + self.logged = True + log.debug("Logged.") + if result == False: + self.authorise() + except VkAPIError as err: + if err.code == 5: + self.authorise() else: self.authorise() self.get_my_data()