31 lines
1.3 KiB
Python
31 lines
1.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
import logging
|
|
from vkaudiotoken import CommonParams, TokenReceiverOfficial, supported_clients, TokenException, TwoFAHelper
|
|
from . import wxUI
|
|
|
|
log = logging.getLogger("authenticator.official")
|
|
|
|
class AuthenticationError(Exception): pass
|
|
|
|
def login(user, password):
|
|
""" Generates authentication workflow at VK servers. """
|
|
log.info("Authenticating user account.")
|
|
access_token = None
|
|
try:
|
|
params = CommonParams(supported_clients.VK_OFFICIAL.user_agent)
|
|
receiver = TokenReceiverOfficial(user, password, params, "GET_CODE")
|
|
access_token = receiver.get_token()["access_token"]
|
|
log.debug("got a valid access token for {}".format(user))
|
|
except TokenException as err:
|
|
if err.code == TokenException.TWOFA_REQ and 'validation_sid' in err.extra:
|
|
log.debug("User requires two factor verification. Calling methods to send an SMS...")
|
|
try:
|
|
TwoFAHelper(params).validate_phone(err.extra['validation_sid'])
|
|
except TokenException as err:
|
|
if err.code == TokenException.TWOFA_ERR:
|
|
wxUI.two_auth_limit()
|
|
raise AuthenticationError("Error authentication two factor auth.")
|
|
code, remember = wxUI.two_factor_auth()
|
|
receiver = TokenReceiverOfficial(user, password, params, code)
|
|
access_token = receiver.get_token()["access_token"]
|
|
return access_token |