U ARd!*@srdZddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl Z ddl Z ddlZddlmZddlmZmZmZdd lmZeeZejrd Znd Zed d dgZedddZde jddde edZdZ de jdddde jddde jdddd Z!edd!d"d#Z"Gd$d%d%e#Z$eed!d&d'Z%d(d)Z&d*d+Z'd,d-Z(d.d/Z)dS)0z(Manage the user's Streamlit credentials.N) namedtuple)datetime)Optional)uuid4)RequestException)env_util file_utilutil) get_loggerz$%userprofile%/.streamlit/config.tomlz~/.streamlit/config.toml _Activationemailis_validreturncCsPtjjdkotj ptjd}d|r,dndt j dddt j d d d d S) Nzutf-8Z WT_SESSIONu {0}%(welcome)s If you’d like to receive helpful onboarding emails, news, offers, promotions, and the occasional swag, please enter your email address below. Otherwise, leave this field blank. %(email)su👋 zWelcome to Streamlit!TboldzEmail: bluefg)Zwelcomer ) sysstdoutencodingr IS_WINDOWSosenvirongetformatclickstyle)Z show_emojir A/tmp/pip-unpacked-wheel-b9et7o5g/streamlit/runtime/credentials.py email_prompt3s    r"a You can find our privacy policy at %(link)s Summary: - This open source library collects usage statistics. - We cannot see and do not store information contained inside Streamlit apps, such as text, charts, images, etc. - Telemetry data is stored in servers in the United States. - If you'd like to opt out, add the following to %(config)s, creating that file if necessary: [browser] gatherUsageStats = false z#https://streamlit.io/privacy-policyT)Z underline)linkconfigzT Collecting usage statistics. To deactivate, set browser.gatherUsageStats to False. z$ %(start)s %(prompt)s %(hello)s zGet started by typing:r)rr$rzstreamlit hellor)startpromptZhello)r rc Cs|dksd|krdSddddddd }td }dd d d ditt|d|ddd|dd }tjd|t| d}| dS)z1Send the user's email to segment.io, if submittedN@zapi.segment.ioz*/*zen-US,en;q=0.9z text/plainzlocalhost:8501zlocalhost:8501/) authorityacceptzaccept-languagez content-typeoriginZrefererz+00:00Zlibraryzanalytics-pythonz2.2.2)nameversionZsubmittedEmailZprovided_email)Z authoremailsourcetrackZ iCkMy7ymtJ9qYzQRXkQpnAJEq7D4NyMU) Z anonymous_idcontextZ messageId timestampeventZtraitstypeZuserIdZwriteKeyzhttps://api.segment.io/v1/t)headersdata) rutcnow isoformatstrrrequestspostjsondumpsencoderaise_for_status)r r4dtr5responser r r! _send_emailns<  rAc@seZdZUdZdZeded<eddZddZ e dd d Z dddd d Z dddZ eddZddZdeddddZdS) CredentialszCredentials class.N _singletoncCs|jdkrttjS)zReturn the singleton instance.N)rCrB)clsr r r! get_currents zCredentials.get_currentcCs*tjdk rtdd|_t|_|t_dS)zInitialize class.Nz;Credentials already initialized. Use .get_current() instead)rBrC RuntimeError activation_get_credential_file_path _conf_fileselfr r r!__init__s zCredentials.__init__rcCs t|SN)r repr_rJr r r!__repr__szCredentials.__repr__Fc Cs|jdk rtddSzHt|jd}t|d}W5QRX|dkrNtt |d|_Wnzt k r|r|j | dYdSt dYnHtk r|r| |j | dYdSttd|jYnXdS) zLoad from toml file.Nz/Credentials already loaded. Not rereading file.rgeneralr )show_instructionsz7Credentials not found. Please run "streamlit activate".zz Unable to load credentials from %s. Run "streamlit reset" and try again. )rGLOGGERerroropenrItomlloadr Exception _verify_emailFileNotFoundErroractivaterFresettextwrapdedent)rK auto_resolvefr5r r r!rWs8  zCredentials.loadTc Cs`z||Wn2ttfk r@}ztt|W5d}~XYnX|jdksT|jjs\tddS)zTCheck if streamlit is activated. Used by `streamlit run script.py` NzActivation email not valid.)rWrXrF_exitr8rGr )rKr_er r r!_check_activateds zCredentials._check_activatedc CsTt}d|_zt|jWn0tk rN}ztd|W5d}~XYnXdS)zReset credentials by removing file. This is used by `streamlit activate reset` in case a user wants to start over. Nz#Error removing credentials file: %s) rBrErGrremoverIOSErrorrSrT)rDcrbr r r!r\s zCredentials.resetc Cs|jdkrdStjtj|jddd|jji}t|jd}t d|i|W5QRXzt |jjWn2t k r}zt d|W5d}~XYnXdS)z!Save to toml file and send email.NT)exist_okr wrQzError saving email: )rGrmakedirspathdirnamerIr rUrVdumprArrSrT)rKr5r`rbr r r!saves  zCredentials.save)rRrcCsz |Wntk r YnX|jrD|jjr:tdqtdn`d}|stjtdddd}t||_|jjr| t t |rt t d}qHt dqHdS) zCActivate Streamlit. Used by `streamlit activate`. zAlready activatedzUActivation not valid. Please run `streamlit activate reset` then `streamlit activate`Fr)textZ prompt_suffixdefaultZ show_defaultTzPlease try again.N)rWrFrGr rarr'r"rYrmsecho_TELEMETRY_TEXT_INSTRUCTIONS_TEXTrSrT)rKrRZ activatedr r r r!r[s4     zCredentials.activate)F)T)T)__name__ __module__ __qualname____doc__rCr__annotations__ classmethodrErLr8rOrWrcr\rmboolr[r r r r!rBs   "  rBcCs@|}t|dkr6|ddkr6tdtddSt|dS)arVerify the user's email address. The email can either be an empty string (if the user chooses not to enter it), or a string with a single '@' somewhere in it. Parameters ---------- email : str Returns ------- _Activation An _Activation object. Its 'is_valid' property will be True only if the email was validated. rr(z"That doesn't look like an email :(NFT)striplencountrSrTr )r r r r!rY/s   rYcCst|tddS)zExit program with error.N)rSrTrexit)messager r r!raKs racCs tdS)Nzcredentials.toml)rZget_streamlit_file_pathr r r r!rHQsrHcCstjtSrM)rrjexistsrHr r r r!_check_credential_file_existsUsrcCsDddlm}ts4|dr4|ds0ttdSt dS)zCheck credentials and potentially activate. Note ---- If there is no credential file and we are in headless mode, we should not check, since credential would be automatically set to an empty string. rr$zserver.headlesszbrowser.gatherUsageStatsN) streamlitr$r get_optionZis_manually_setrrp_TELEMETRY_HEADLESS_TEXTrBrErcrr r r!check_credentialsYs   r)*rvr;rrr] collectionsrrtypingruuidrrr9rVZrequests.exceptionsrrrrr Zstreamlit.loggerr rsrSrZ_CONFIG_FILE_PATHr r8r"rrqrrrrAobjectrBrYrarHrrr r r r!sZ           +