o ՂFeA@s*dZddlZddlmZddlmZddlmZddlZddlZddlm Z ddl m Z m Z ddl mZmZejr>eZeZneZeZGd d d eZGd d d eZGd ddeZGdddeZGdddeZGdddeZGdddeeZGdddeZ GdddeZ!GdddeZ"dS)ap This module simplifies the process of obtaining an authenticated session for steam websites. After authentication is completed, a :class:`requests.Session` is created containing the auth cookies. The session can be used to access ``steamcommunity.com``, ``store.steampowered.com``, and ``help.steampowered.com``. .. warning:: A web session may expire randomly, or when you login from different IP address. Some pages will return status code `401` when that happens. Keep in mind if you are trying to write robust code. .. note:: If you are using :class:`.SteamClient` take a look at :meth:`.SteamClient.get_web_session()` .. note:: If you need to authenticate as a mobile device for things like trading confirmations use :class:`MobileWebAuth` instead. The login process is identical, and in addition you will get :attr:`.oauth_token`. Example usage: .. code:: python import steam.webauth as wa user = wa.WebAuth('username') # At a console, cli_login can be used to easily perform all login steps session = user.cli_login('password') session.get('https://store.steampowered.com/account/history') # Or the login steps be implemented for other situation like so try: user.login('password') except (wa.CaptchaRequired, wa.LoginIncorrect) as exp: if isinstance(exp, LoginIncorrect): # ask for new password else: password = self.password if isinstance(exp, wa.CaptchaRequired): print user.captcha_url # ask a human to solve captcha else: captcha = None user.login(password=password, captcha=captcha) except wa.EmailCodeRequired: user.login(email_code='ZXC123') except wa.TwoFactorCodeRequired: user.login(twofactor_code='ZXC123') user.session.get('https://store.steampowered.com/account/history/') N)time) b64encode)getpass)SteamID)make_requests_sessiongenerate_session_id) rsa_publickeypkcs1v15_encryptc@s|eZdZdZdZdZdZdZdZdZ dddZ ddZ e d d Z d d Zd dZdddZddZdddZdddZdS)WebAuthNFcCs"|jtt|_|dSN)__dict__updatelocalsrsession_session_setup)selfusernamepasswordr8/usr/local/lib/python3.10/dist-packages/steam/webauth.py__init__Us zWebAuth.__init__cCsdSr rrrrrrZszWebAuth._session_setupcCs|jdkrdSd|jS)zOIf a captch is required this property will return url to the image, or ``None``r Nz6https://steamcommunity.com/login/rendercaptcha/?gid=%s) captcha_gidrrrr captcha_url]s  zWebAuth.captcha_urlc CsVz|jjdd|ttddd}W|Stjjy*}ztt |d}~ww)aGet rsa key for a given username :param username: username :type username: :class:`str` :return: json response :rtype: :class:`dict` :raises HTTPError: any problem with http request, timeouts, 5xx, 4xx etc z+https://steamcommunity.com/login/getrsakey/i)r donotcache)timeoutdataN) rpostintrjsonrequests exceptionsRequestException HTTPErrorstr)rrresperrr get_rsa_keyes    zWebAuth.get_rsa_keycCsD|js ||j}tt|ddt|dd|_|d|_dSdS)N publickey_mod publickey_exp timestamp)keyr*rrintBaser.)rr(rrr _load_key{s  zWebAuth._load_keyc Cs|jtt|j|d||rt|jnd||j|d|jdt t dd }z |j j d|dd  WStjjyF}ztt|d}~ww) Nasciir python-steam webauthtrue順) rr emailauth emailsteamid twofactorcode captchagid captcha_textloginfriendlyname rsatimestampremember_loginr)https://steamcommunity.com/login/dologin/rrr)rrr r/encoder'steam_idrr.r!rrr r"r#r$r%r&rrcaptcha email_codetwofactor_coderr)rrr _send_logins$  zWebAuth._send_logincCs d|_dS)NzSTEAM_0:0:57296955)rA)rlogin_responserrr_finalize_logins zWebAuth._finalize_loginenglishc Cs|jr|jS|r ||_n |jr|j}ntd|s|jr|j}||j||||d}|dr|drd|_d|_|_d|_t|jj D]}dD]}|jj j |j |j ||j d qJqFt|_dD]!}|jj j d ||d |jj j d d |d |jj j d|j|d qb|||jS|ddr|d|_d|_|ddrd|_t|dt|d|ddrt|d|_t|d|ddrt|dd|ddvrt|dd|_t|d)a`Attempts web login and returns on a session with cookies set :param password: password, if it wasn't provided on instance init :type password: :class:`str` :param captcha: text reponse for captcha challenge :type captcha: :class:`str` :param email_code: email code for steam guard :type email_code: :class:`str` :param twofactor_code: 2FA code for steam guard :type twofactor_code: :class:`str` :param language: select language for steam web pages (sets language cookie) :type language: :class:`str` :return: a session on success and :class:`None` otherwise :rtype: :class:`requests.Session`, :class:`None` :raises HTTPError: any problem with http request, timeouts, 5xx, 4xx etc :raises LoginIncorrect: wrong username or password :raises CaptchaRequired: when captcha is needed :raises CaptchaRequiredLoginIncorrect: when captcha is needed and login is incorrect :raises EmailCodeRequired: when email is needed :raises TwoFactorCodeRequired: when 2FA is needed zpassword is not specified)rrCrDrEsuccesslogin_completeTr r zstore.steampowered.comzhelp.steampowered.comzsteamcommunity.comdomainsecureSteam_LanguagerN birthtime-3333 sessionidcaptcha_neededFrclear_password_fieldmessageemailauth_neededr7requires_twofactorztoo many login failures) logged_onrrLoginIncorrect captcha_coder1rFrlistcookiessetnamevaluerOr session_idrHgetCaptchaRequiredLoginIncorrectCaptchaRequiredrrAEmailCodeRequiredTwoFactorCodeRequiredTooManyLoginFailures) rrrCrDrElanguager(cookierNrrrloginsT              z WebAuth.loginc Cs z ||||||WSttfyH}z/d}}t|tr-|s"dnd}t|t|j}t|tr Tr zEnter password for %s: z)Invalid password for %s. Enter password: z"Solve CAPTCHA at %s CAPTCHA code: NzEnter email code: z"Incorrect code. Enter email code: zEnter 2FA code: z Incorrect code. Enter 2FA code: ) rkr[re isinstancerreprrr _cli_inputrfrg)rrrCrDrEriexppromptrrr cli_logins6      zWebAuth.cli_login)r r r r r )r r r r rI)__name__ __module__ __qualname__r/rZrrbrr\rArrpropertyrr*r1rFrHrkrqrrrrr Ls$    Nr c@s0eZdZdZdZd ddZddZd d d ZdS) MobileWebAuthzJIdentical to :class:`WebAuth`, except it authenticates as a mobile device.Nr cCs|jtt|j|d||rt|jnd||j|d|jdt t dddd }|j j d d |j j d d z2z|j jd |ddWW|j j d d|j j d dStjjyj}ztt|d}~ww|j j d d|j j d dw)Nr2r r3r4r5DE45CD61z3read_profile write_profile read_client write_client) rrr6r7r8r9r:r;r<r=roauth_client_id oauth_scopemobileClientVersion 0 (2.1.3) mobileClientandroidr>rr?)rrr r/r@r'rArr.r!rrr^r_r r"popr#r$r%r&rBrrrrF's4  zMobileWebAuth._send_logincCs*t|d}t|d|_|d|_dS)Noauthsteamid oauth_token)r"loadsrrAr)rrGrrrrrHCszMobileWebAuth._finalize_loginrIc Cs|r||_n |jr |j}ntd|rt||_n|js td|jj}d|i}z |jjd|d}Wntjj yF}zt t |d}~wwz| d}Wnt j jyh}z d|jvrbtd |d}~wwt|_d D]Z}|jjjd d |d |jjjd|j|d |jjjdd|d |jjjdd|d |jjjdt |d|d|d |jjjdt |d|d|dd|jjjd||d qod|_|jS)a.Attempts a mobile authenticator login using an oauth token, which can be obtained from a previously logged-in `MobileWebAuth` :param oauth_token: oauth token string, if it wasn't provided on instance init :type oauth_token: :class:`str` :param steam_id: `SteamID` of the account to log into, if it wasn't provided on instance init :type steam_id: :class:`str` or :class:`SteamID` :param language: select language for steam web pages (sets language cookie) :type language: :class:`str` :return: a session on success and :class:`None` otherwise :rtype: :class:`requests.Session`, :class:`None` :raises HTTPError: any problem with http request, timeouts, 5xx, 4xx etc :raises LoginIncorrect: Invalid token or SteamID ztoken is not specifiedzsteam_id is not specified access_tokenz@https://api.steampowered.com/IMobileAuthService/GetWGToken/v0001)rNresponsez-Please verify your 
key=
parameter.z invalid tokenrLrRrSrQrTr{r|r}r~ steamLoginz%7C%7CtokensteamLoginSecure token_secureTrMrP)rr[rrAas_64rr r#r$r%r&r'r"decoderJSONDecodeErrortextrrbr^r_rZ) rrrArirr(r) resp_datarNrrr oauth_loginHsN   $zMobileWebAuth.oauth_loginrr)r r rI)rsrtru__doc__rrFrHrrrrrrw#s  rwc@ eZdZdS)WebAuthExceptionNrsrtrurrrrrrc@r)r&Nrrrrrr&rr&c@r)r[Nrrrrrr[rr[c@r)reNrrrrrrerrec@r)rdNrrrrrrdrrdc@r)rfNrrrrrrfrrfc@r)rgNrrrrrrgrrgc@r)rhNrrrrrrhrrh)#rr"rbase64rrsixr# steam.steamidrsteam.utils.webrrsteam.core.cryptorr PY2longr0 raw_inputrnr!inputobjectr rw Exceptionrr&r[rerdrfrgrhrrrrs47    Xd