39 lines
1.7 KiB
Python
39 lines
1.7 KiB
Python
#!/usr/bin/python
|
|
import keys
|
|
import logging
|
|
from authenticator import official
|
|
from vk_api.audio import VkAudio
|
|
from authenticator.wxUI import two_factor_auth
|
|
|
|
log = logging.getLogger("vkSessionHandler")
|
|
|
|
class vkObject(object):
|
|
|
|
def __init__(self):
|
|
self.api_key = keys.keyring.get_api_key()
|
|
|
|
def login(self, user, password, token, secret, device_id, alt_token, filename):
|
|
if alt_token == False:
|
|
log.info("Using kate's token...")
|
|
# Let's import the patched vk_api module for using a different user agent
|
|
from . import vk_api_patched as vk_api
|
|
if token == "" or token == None:
|
|
log.info("Token is not valid. Generating one...")
|
|
original_token = official.login(user, password)
|
|
token = original_token[0]
|
|
secret = original_token[1]
|
|
device_id = original_token[2]
|
|
log.info("Token validated...")
|
|
self.session_object = vk_api.VkApi(app_id=self.api_key, login=user, password=password, token=token, secret=secret, device_id=device_id, scope="offline, wall, notify, friends, photos, audio, video, docs, notes, pages, status, groups, messages, notifications, stats", config_filename=filename)
|
|
else:
|
|
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.client = self.session_object.get_api()
|
|
# print self.client.audio.get()
|
|
log.debug("Getting tokens for 24 hours...")
|
|
# info = self.client.account.getProfileInfo()
|
|
# Add session data to the application statistics.
|
|
self.client.stats.trackVisitor()
|
|
self.client_audio = VkAudio(self.session_object)
|