mcvsoftware/socializer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use patched VK api only if kate tokens are enabled

Browse Source
This commit is contained in:
Manuel Cortez 2018-12-19 17:34:28 -06:00
parent b9ccc083ae
commit 909fd9d68c
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/sessionmanager/vkSessionHandler.py
View File

@ -1,9 +1,9 @@
#!/usr/bin/python #!/usr/bin/python
import keys import keys
import logging import logging
import vk_api_patched as vk_api
import core import core
from vk_api.audio import VkAudio from vk_api.audio import VkAudio
from wxUI import two_factor_auth
log = logging.getLogger("vkSessionHandler") log = logging.getLogger("vkSessionHandler")
@ -15,6 +15,8 @@ class vkObject(object):
def login(self, user, password, token, alt_token, filename): def login(self, user, password, token, alt_token, filename):
if alt_token == False: if alt_token == False:
log.info("Using kate's token...") log.info("Using kate's token...")
# Let's import the patched vk_api module for using a different user agent
import vk_api_patched as vk_api
if token == "" or token == None: if token == "" or token == None:
log.info("Token is not valid. Generating one...") log.info("Token is not valid. Generating one...")
token = core.requestAuth(user, password) token = core.requestAuth(user, password)
@ -24,7 +26,8 @@ class vkObject(object):
log.info("Token validated...") log.info("Token validated...")
self.session_object = vk_api.VkApi(app_id=self.api_key, login=user, password=password, token=token, scope="offline, wall, notify, friends, photos, audio, video, docs, notes, pages, status, groups, messages, notifications, stats", config_filename=filename) self.session_object = vk_api.VkApi(app_id=self.api_key, login=user, password=password, token=token, scope="offline, wall, notify, friends, photos, audio, video, docs, notes, pages, status, groups, messages, notifications, stats", config_filename=filename)
else: else:
self.session_object = vk_api.VkApi(app_id=self.api_key, login=user, password=password, scope="offline, wall, notify, friends, photos, audio, video, docs, notes, pages, status, groups, messages, notifications, stats", config_filename=filename) import vk_api
self.session_object = vk_api.VkApi(app_id=self.api_key, login=user, password=password, scope="offline, wall, notify, friends, photos, audio, video, docs, notes, pages, status, groups, messages, notifications, stats", config_filename=filename, auth_handler=two_factor_auth)
self.session_object.auth() self.session_object.auth()
self.client = self.session_object.get_api() self.client = self.session_object.get_api()
# print self.client.audio.get() # print self.client.audio.get()