o ߿.g/@sRdZddlZddlZddlZddlZddlZddlZddlZdZGdddZ dS)z} This file contains a class to manage tokens Coded by Tyler Bowers Github: https://github.com/tylerebowers/Schwab-API-Python Nz/var/www/html/flask_project/c@sxeZdZeddfddZdedefddZd ed ed efd d Z ddZ dddZ ddZ ddZ ddZddZdS)Tokensz 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)rFrG redirect_urir%)rFr%zGInvalid grant type; options are 'authorization_code' or 'refresh_token'z(https://api.schwabapi.com/v1/oauth/token)headersdata) base64 b64encodebytesr!r"decoder#rrequestspost)rrFrGrLrMrrr_post_oauth_token[s( 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_issuedrWF) ensure_asciiindentNz'[Schwabdev] Could not write tokens file) getr$r%r&r,r-openr0 isoformatjsondumprr5)rrUrVrWfto_writeerrr _write_tokensss$  &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 rrWr$r%r&rYrZN)NNN)r_r0raloadr^r$r%r&r' fromisoformatr,r-rr5)rrcdrWrerrrr1s       ( 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 r rrrz S remaining)z*[Schwabdev] The refresh token has expired!=zA[Schwabdev] The access token has expired, updating automatically.N)r/r'r3r*r+r-r4r5r6r7r9r.r,r r2update_access_token)rforcerDrrrrs 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)rTr%okr'r3r*r+rfr-rar r2r5r,text)rresponserUrrrrls  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@rJz-[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) startswithindexrTrnr'r3r*r+rfrar r2r5ro)r url_or_coderGrpr3rrrupdate_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)rqr5 webbrowserr_inputrv)rauth_url response_urlrrrr9s  zTokens.update_refresh_tokenN)F)__name__ __module__ __qualname__ PATH_UBUNTUrEr8rTr'dictrfr1rrlrqrvr9rrrrrsI  r) __doc__rarrNrRr'r:rwr~rrrrrs