o iH@szddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZddl mZddl mZdd l mZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlm Z Gddde!Z"Gddde#Z$Gddde"Z%Gddde"Z&Gddde"Z'dd Z(d!d"Z)d#d$Z*Gd%d&d&ee+Z,dS)'N)input)buildwraps)ServiceAccountCredentials)FlowExchangeError)AccessTokenRefreshError)OAuth2WebServerFlow)OOB_CALLBACK_URN)Storage)ClientRedirectHandler)ClientRedirectServer)scopes_to_string) ApiAttribute)ApiAttributeMixin)LoadSettingsFile)ValidateSettings) SettingsError)InvalidConfigErrorc@eZdZdZdS) AuthErrorz3Base error for authentication/authorization errors.N__name__ __module__ __qualname____doc__rrG/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/pydrive/auth.pyrrc@r)InvalidCredentialsErrorz&Error trying to read credentials file.Nrrrrrr rr c@r)AuthenticationRejectedzUser rejected authentication.Nrrrrrr!"rr!c@r)AuthenticationErrorzGeneral authentication error.Nrrrrrr"&rr"c@r) RefreshErrorzAccess token refresh error.Nrrrrrr#*rr#ctfdd}|S)z>Decorator to check if the auth is valid and loads auth if not.cs|jdur t|_|jjr t|dddkr|jn|j|jjdur+|j|durTd|vrT|ddurTd|dvrT|dddurT|dd|_|dd=n|j |_|g|Ri|S)N auth_methodFserviceparamhttp) auth GoogleAuthaccess_token_expiredgetattr ServiceAuthLocalWebserverAuthr& Authorizer(Get_Http_Object)selfargskwargs decorateerr _decorated/s"         zLoadAuth.._decoratedrr5r6rr4rLoadAuth-sr8cr$)z'Decorator to authorize service account.csd|_d}|jd}|jdur|r||jdur-|g|Ri||d}n|jrL|jjdur;|n|g|Ri||d}|rV|rX| dSdSdS)Nr&Fsave_credentialsT) r%settingsget credentialsLoadCredentialsr/r+ refresh_tokenRefreshSaveCredentials)r1r2r3dirtyr9r4rrr6Ps$     z$CheckServiceAuth.._decoratedrr7rr4rCheckServiceAuthNsrBcr$)z6Decorator to check if it requires OAuth2 flow request.csd}d}|jd}|jdur|r||jdur||jdur1|g|Ri|}d}n|jrL|jjdur?|n |g|Ri|}d}|durU| ||r_|ra| dSdSdS)NFr9T) r:r;r<r=flowGetFlowr+r>r?Authr@)r1r2r3rAcoder9r4rrr6is(       zCheckAuth.._decoratedrr7rr4r CheckAuthgsrGc@s,eZdZdZddddgdZgdZgdZed Zed Z ed Z ed Z ed Z edZ edZd;ddZeddZe  dr*zWrapper class for oauth2client library in google-api-python-client. Loads all settings and credentials from one 'settings.yaml' file and performs common OAuth2.0 related functionality such as authentication and authorization. filezclient_secrets.jsonFz%https://www.googleapis.com/auth/drive)client_config_backendclient_config_filer9 oauth_scope) client_id client_secretauth_uri token_uri revoke_uri redirect_uri)client_service_emailclient_user_emailclient_pkcs12_file_pathr: client_configrCr<r(r&r% settings.yamlNcCsf||_t|i|_zt||_Wnty |j|_YdSw|jdur,|j|_dSt|jdS)zCreate an instance of GoogleAuth. This constructor just sets the path of settings file. It does not actually read the file. :param settings_file: path of settings file. 'settings.yaml' by default. :type settings_file: str. N) http_timeoutr__init__rUrr:rDEFAULT_SETTINGSr)r1 settings_filerWrrrrXs    zGoogleAuth.__init__cCs|jdurdS|jjS)zChecks if access token doesn't exist or is expired. :returns: bool -- True if access token doesn't exist or is expired. NT)r<r+r1rrrr+s zGoogleAuth.access_token_expired localhostc Cs"|durddg}d}d}|D]#}|}z t||ft}Wntjy.}zWYd}~qd}~wwd}|r;d||f}ntdtd td t||j_|} t j | d dd td ttd| t| d|j vr{tdt dd|j vr|j dStdtdtd)aAuthenticate and authorize from user by creating local web server and retrieving authentication code. This function is not for web server application. It creates local web server for user from standalone application. :param host_name: host name of the local web server. :type host_name: str. :param port_numbers: list of port numbers to be tried to used. :type port_numbers: list. :returns: str -- code returned from local web server :raises: AuthenticationRejected, AuthenticationError NiiFrTz http://%s:%s/z>Failed to start a local web server. Please check your firewallz=settings and locally running programs that may be blocking orz8using configured ports. Default ports are 8080 and 8090.r)new autoraisez&Your browser has been opened to visit: errorz#Authentication request was rejectedzUser rejected authenticationrFz>Failed to find "code" in the query parameters of the redirect.zTry command-line authenticationzNo code found in redirect)r r socketr`printr"rCrQ GetAuthUrl webbrowseropenhandle_request query_paramsr!) r1 host_name port_numberssuccess port_numberporthttpdeoauth_callback authorize_urlrrrr.sH     zGoogleAuth.LocalWebserverAuthcCs<t|j_|}tdttd|ttdS)zAuthenticate and authorize from user by printing authentication url retrieving authentication code from command-line. :returns: str -- code returned from commandline. z)Go to the following link in your browser:r_zEnter verification code: )r rCrQrcrbrstrip)r1rprrrCommandLineAuths  zGoogleAuth.CommandLineAuthcCsxt|jt|jr|t|jd}|jd}|jd}|jd}tj|||d|_ |r:|j j |d|_ dSdS)zAuthenticate and authorize using P12 private key, client id and client email for a Service account. :raises: AuthError, InvalidConfigError rKrSrRrT)service_account_emailfilenamescopes)subN) setSERVICE_CONFIGS_LISTrULoadServiceConfigSettingsrr:r;rfrom_p12_keyfiler<create_delegated)r1ru user_email service_email file_pathrrrr-s   zGoogleAuth.ServiceAuthcC@|dur|jd}|durtd|dkr|dStd)zLoads credentials or create empty credentials if it doesn't exist. :param backend: target backend to save credential to. :type backend: str. :raises: InvalidConfigError Nsave_credentials_backend!Please specify credential backendrH Unknown save_credentials_backend)r:r;rLoadCredentialsFiler1backendrrrr=s  zGoogleAuth.LoadCredentialscCsT|dur|jd}|durtdz t|}||_WdSty)tdw)a%Loads credentials or create empty credentials if it doesn't exist. Loads credentials file from path in settings if not specified. :param credentials_file: path of credentials file to read. :type credentials_file: str. :raises: InvalidConfigError, InvalidCredentialsError Nsave_credentials_file'Please specify credentials file to read(Credentials file cannot be symbolic link)r:r;rr r<IOErrorr r1credentials_filestoragerrrr%s  zGoogleAuth.LoadCredentialsFilecCr)a$Saves credentials according to specified backend. If you have any specific credentials backend in mind, don't use this function and use the corresponding function you want. :param backend: backend to save credentials. :type backend: str. :raises: InvalidConfigError NrrrHr)r:r;rSaveCredentialsFilerrrrr@8s  zGoogleAuth.SaveCredentialscCst|jdur td|dur|jd}|durtdzt|}||j|j|WdSty9tdw)zSaves credentials to the file in JSON format. :param credentials_file: destination to save file to. :type credentials_file: str. :raises: InvalidConfigError, InvalidCredentialsError NzNo credentials to saverrr) r<r r:r;rr put set_storerrrrrrKs    zGoogleAuth.SaveCredentialsFilecCsh|dur|jd}|durtd|dkr|dS|dkr&|dS|dkr0|dStd)aQLoads client configuration according to specified backend. If you have any specific backend to load client configuration from in mind, don't use this function and use the corresponding function you want. :param backend: backend to load client configuration from. :type backend: str. :raises: InvalidConfigError NrIz$Please specify client config backendrHr:r&zUnknown client_config_backend)r:r;rLoadClientConfigFileLoadClientConfigSettingsryrrrrLoadClientConfig_s    zGoogleAuth.LoadClientConfigc Cs|dur |jd}z t|\}}Wntjy%}ztd|d}~ww|tjtjfvr2tdz#gd}|D] }|||j|<q9|d|jd<|dd|jd <Wn t y`td wd g}z|D] }|||j|<qgWdSt y}YdSw) aLoads client configuration file downloaded from APIs console. Loads client config file from path in settings if not specified. :param client_config_file: path of client config file to read. :type client_config_file: str. :raises: InvalidConfigError NrJzInvalid client secrets file %sz)Unknown client_type of client config file)rLrMrNrOrP redirect_urisrrQz"Insufficient client config in file client_email) r: clientsecretsloadfileInvalidClientSecretsErrorrTYPE_WEBTYPE_INSTALLEDrUr;KeyError)r1rJ client_type client_infor` config_indexconfigservice_auth_configrrrrvs:    zGoogleAuth.LoadClientConfigFilec CsP|jD]"}z |jd||j|<Wqty%d}|d|7}t|wdS)zSLoads client configuration from settings file. :raises: InvalidConfigError service_configz'Insufficient service config in settingsz Missing: {} key.N)rxr:rUrformatr)r1rerrrrrrys   z$GoogleAuth.LoadServiceConfigSettingsc Cs>|jD]}z |jd||j|<WqtytdwdS)zTLoads client configuration from settings file. :raises: InvalidConfigError rUz&Insufficient client config in settingsN)CLIENT_CONFIGS_LISTr:rUrr)r1rrrrrs  z#GoogleAuth.LoadClientConfigSettingscstfddjDsjdjdjdd}jddur-jd|d<tjd jd tjd fi|_jd rUjj d dddSdS)zQGets Flow object from client configuration. :raises: InvalidConfigError c3s|]}|jvVqdSN)rU).0rr[rr sz%GoogleAuth.GetFlow..rQrNrO)rQrNrOrPNrLrMrKget_refresh_tokenofflineforce) access_typeapproval_prompt) allrrrUr rr:rCr;paramsupdate)r1constructor_kwargsrr[rrDs.    zGoogleAuth.GetFlowc Csz|jdur td|jjdurtd|jdur tj|jd|_z |j|jWdSty<}ztd|d}~ww)z;Refreshes the access_token. :raises: RefreshError NzNo credential to refresh.zBNo refresh_token found.Please set access_type of OAuth to offline.timeoutzAccess token refresh failed: %s) r<r#r>r(httplib2HttprWrefreshr)r1r`rrrr?s    zGoogleAuth.RefreshcCs|jdur ||jS)zlCreates authentication url where user visits to grant access. :returns: str -- Authentication url. N)rCrDstep1_get_authorize_urlr[rrrrcs  zGoogleAuth.GetAuthUrlcCs|||dS)zAuthenticate, authorize, and build service. :param code: Code for authentication. :type code: str. :raises: AuthenticationError N) Authenticater/)r1rFrrrrEs  zGoogleAuth.Authc CsV|jdur |z |j||_Wnty$}ztd|d}~wwtddS)zAuthenticates given authentication code back from user. :param code: Code for authentication. :type code: str. :raises: AuthenticationError NzOAuth2 code exchange failed: %szAuthentication successful.)rCrDstep2_exchanger<rr"rb)r1rFrnrrrrs   zGoogleAuth.AuthenticatecCsN|jdur tj|jd|_|jrtd|j|j|_tdd|jd|_ dS)zEAuthorizes and builds service. :raises: AuthenticationError Nrz*No valid credentials provided to authorizedrivev2)r() r(rrrWr+r"r< authorizerr&r[rrrr/s zGoogleAuth.AuthorizecCstj|jd}|j|}|S)zCreate and authorize an httplib2.Http object. Necessary for thread-safety. :return: The http object to be used in each call. :rtype: httplib2.Http r)rrrWr<r)r1r(rrrr0 s zGoogleAuth.Get_Http_Object)rVN)r\Nr)&rrrrrYrrxrr:rUrCr<r(r&r%rXpropertyr+rGr.rrrBr-r=rr@rrrryrrDr?rcrErr/r0rrrrr*sT    6        &     r*)-rardroauth2client.clientsecretsr six.movesrapiclient.discoveryr functoolsroauth2client.service_accountroauth2client.clientrrr r oauth2client.filer oauth2client.toolsr r oauth2client._helpersrapiattrrrr:rrrr Exceptionrrr r!r"r#r8rBrGobjectr*rrrrs>                   !