o g/@sNdZddlZddlZddlZddlZddlZddlZddlZGdddZdS)z} This file contains a class to manage tokens Coded by Tyler Bowers Github: https://github.com/tylerebowers/Schwab-API-Python Nc@speZdZdddZdedefddZd ed ed efd d ZddZ dddZ ddZ ddZ ddZ ddZdS)Tokens tokens.jsonTc s|durtd|durtd|durtd|dur tdt|dks,t|dkr0td|d d d kr.checkerT)targetdaemonz>[Schwabdev] Warning: Tokens will not be updated automatically.) Exceptionlen_client_app_key _app_secret _callback_url access_token refresh_tokenid_tokendatetimeminreplacetimezoneutc_access_token_issued_refresh_token_issued_access_token_timeout_refresh_token_timeout _tokens_file _read_tokensrverbosenow total_secondsprintintabsstrupdate_refresh_token threadingThreadstart) rclientapp_key app_secret callback_url tokens_fileupdate_tokens_autoat_deltart_deltarrrr__init__s^    Z Z    zTokens.__init__ grant_typecodecCstdtt|jd|jdddd}|dkr$d||jd}n|dkr.d|d }ntd tj d ||d S) a Makes API calls for auth code and refresh tokens :param grant_type: 'authorization_code' or 'refresh_token' :type grant_type: str :param code: authorization code :type code: str :return: response :rtype: requests.Response zBasic :zutf-8z!application/x-www-form-urlencoded) Authorizationz Content-Typeauthorization_code)rGrH redirect_urir&)rGr&zGInvalid grant type; options are 'authorization_code' or 'refresh_token'z(https://api.schwabapi.com/v1/oauth/token)headersdata) base64 b64encodebytesr"r#decoder$rrequestspost)rrGrHrMrNrrr_post_oauth_tokenYs( zTokens._post_oauth_token at_issued rt_issuedtoken_dictionaryc Cs|d|_|d|_|d|_||_||_z.t|jd}|||d}t j ||dddWd Wd S1s?wYWd St ya}zt |t d WYd }~d Sd }~ww) a6 Writes token file and sets variables :param at_issued: access token issued :type at_issued: datetime.pyi :param rt_issued: refresh token issued :type rt_issued: datetime.pyi :param token_dictionary: token dictionary :type token_dictionary: dict r%r&r'w)access_token_issuedrefresh_token_issuedrXF) ensure_asciiindentNz'[Schwabdev] Could not write tokens file) getr%r&r'r-r.openr1 isoformatjsondumprr6)rrVrWrXfto_writeerrr _write_tokensqs$  &zTokens._write_tokensc CszQt|jd@}t|}|d}|d|_|d|_|d|_tj |d|_ tj |d|_ |j |j |fWdWS1sJwYWdSt yh}z t |WYd}~d Sd}~ww) z Reads token file and sets variables :return: access token issued, refresh token issued, token dictionary :rtype: datetime.pyi, datetime.pyi, dict rrXr%r&r'rZr[N)NNN)r`r1rbloadr_r%r&r'r( fromisoformatr-r.rr6)rrddrXrfrrrr2s       ( zTokens._read_tokensFc Cs|jtjtjj|j}|dkrAtd|dkrdndtt |dddtt |dd dd tt |d dd |dksG|rQtd | dS|j tjtjj|j d krq|j jrktd|dSdS)z Checks if tokens need to be updated and updates if needed (only access token is automatically updated) :param force: force update of refresh token (also updates access token) :type force: bool iz1[Schwabdev] The refresh token will expire soon! (rr r r rrrrz S remaining)z*[Schwabdev] The refresh token has expired!=zA[Schwabdev] The access token has expired, updating automatically.N)r0r(r4r+r,r.r5r6r7r8r:r/r-r!r3update_access_token)rforcerErrrrs Z  $ zTokens.update_tokenscCsn|d|j}|jr,tjtjj}|||j| |j j r*t d|j dSdSt |jt ddS)zD "refresh" the access token using the refresh token r&z"[Schwabdev] Access token updated: zS[Schwabdev] Could not get new access token, refresh_token probably expired randomlyN)rUr&okr(r4r+r,rgr.rbr!r3r6r-text)rresponserVrrrrms   zTokens.update_access_tokencCsd|jd|jS)z' Get authorization url z7https://api.schwabapi.com/v1/oauth/authorize?client_id=z&redirect_uri=)r"r$rrrrget_refresh_token_auth_urlsz!Tokens.get_refresh_token_auth_urlcCs|dr||dd|dd}n|}|d|}|jr?tjtjj}|||| |j j r=t dd Sd St |j t dd S) z Get new access and refresh tokens using callback url or authorization code. :param url_or_code: callback url (full url) or authorization code (the code=... in url) :type url_or_code: str zhttps://zcode=rz%40@rKz-[Schwabdev] Refresh and Access tokens updatedz[Schwabdev] Could not get new refresh and access tokens, check these: 1. App status is "Ready For Use". 2. App key and app secret are valid. 3. You pasted the whole url within 30 seconds. (it has a quick expiration)N) startswithindexrUror(r4r+r,rgrbr!r3r6rp)r url_or_coderHrqr4rrrupdate_refresh_token_from_codes $   z%Tokens.update_refresh_token_from_codecCs6|}td|t|td}||dS)zM Get new access and refresh tokens using authorization code. z"[Schwabdev] Open to authenticate: z?[Schwabdev] After authorizing, paste the address bar url here: N)rrr6 webbrowserr`inputrw)rauth_url response_urlrrrr:s  zTokens.update_refresh_tokenN)rT)F)__name__ __module__ __qualname__rFr9rUr(dictrgr2rrmrrrwr:rrrrrs I  r) __doc__rbrrOrSr(r;rxrrrrrs