Create mobileauth.py

steam/mobileauth.py

@@ -0,0 +1,135 @@
+# -*- coding: utf-8 -*-
+from time import time
+import sys
+from base64 import b64encode
+import requests
+
+from cryptography.hazmat.primitives.asymmetric.rsa import RSAPublicNumbers
+from cryptography.hazmat.primitives.asymmetric.padding import PKCS1v15
+from steam.core.crypto import backend
+
+from steam.util.web import make_requests_session
+from steam import SteamID
+
+if sys.version_info < (3,):
+    intBase = long
+else:
+    intBase = int
+
+
+class MobileAuth(object):
+    key = None
+    complete = False  #: whether authentication has been completed successfully
+    session = None    #: :class:`requests.Session` (with auth cookies after auth is complete)
+    captcha_gid = -1
+    steamid = None    #: :class:`steam.steamid.SteamID` (after auth is complete)
+
+    def __init__(self, username, password):
+        self.__dict__.update(locals())
+        self.session = make_requests_session()
+
+    @property
+    def captcha_url(self):
+        if self.captcha_gid == -1:
+            return None
+        else:
+            return "https://store.steampowered.com/login/rendercaptcha/?gid=%s" % self.captcha_gid
+
+    def get_rsa_key(self, username):
+        try:
+            resp = self.session.post('https://steamcommunity.com/mobilelogin/getrsakey/',
+                                    timeout=15,
+                                    data={
+                                        'username': username,
+                                        'donotchache': int(time() * 1000),
+                                        },
+                                    ).json()
+        except requests.exceptions.RequestException as e:
+            raise HTTPError(str(e))
+
+        return resp
+
+    def _load_key(self):
+        if not self.key:
+            resp = self.get_rsa_key(self.username)
+
+            nums = RSAPublicNumbers(intBase(resp['publickey_exp'], 16),
+                                    intBase(resp['publickey_mod'], 16),
+                                    )
+
+            self.key = backend.load_rsa_public_numbers(nums)
+            self.timestamp = resp['timestamp']
+
+    def login(self, captcha='', email_code='', twofactor_code='', language='english'):
+        if self.complete:
+            return self.session
+            
+        for domain in ['store.steampowered.com', 'help.steampowered.com', 'steamcommunity.com']:
+            self.session.cookies.set('forceMobile', '1', domain=domain, secure=False)
+            self.session.cookies.set('mobileClientVersion', '0 (2.1.3)', domain=domain, secure=False)
+            self.session.cookies.set('mobileClient', 'android', domain=domain, secure=False)
+            self.session.cookies.set('Steam_Language', 'english', domain=domain, secure=False)
+            self.session.cookies.set('dob', '', domain=domain, secure=False)
+
+        self._load_key()
+
+        data = {
+            'username' : self.username,
+            "password": b64encode(self.key.encrypt(self.password.encode('ascii'), PKCS1v15())),
+            "emailauth": email_code,
+            "emailsteamid": str(self.steamid) if email_code else '',
+            "twofactorcode": twofactor_code,
+            "captchagid": self.captcha_gid,
+            "captcha_text": captcha,
+            "loginfriendlyname": "python-steam webauth",
+            "rsatimestamp": self.timestamp,
+            "remember_login": 'true',
+            "donotcache": int(time() * 100000),
+        }
+        data['oauth_client_id'] = 'DE45CD61'
+        data['oauth_scope'] = 'read_profile write_profile read_client write_client'
+        data['loginfriendlyname'] = '#login_emailauth_friendlyname_mobile'
+
+        try:
+            resp = self.session.post('https://steamcommunity.com/mobilelogin/dologin/', data=data, timeout=15).json()
+        except requests.exceptions.RequestException as e:
+            raise HTTPError(str(e))
+
+        self.captcha_gid = -1
+
+        if resp['success'] and resp['login_complete']:
+            self.complete = True
+            self.password = None
+            return resp
+        else:
+            if resp.get('captcha_needed', False):
+                self.captcha_gid = resp['captcha_gid']
+
+                raise CaptchaRequired(resp['message'])
+            elif resp.get('emailauth_needed', False):
+                self.steamid = SteamID(resp['emailsteamid'])
+                raise EmailCodeRequired(resp['message'])
+            elif resp.get('requires_twofactor', False):
+                raise TwoFactorCodeRequired(resp['message'])
+            else:
+                raise LoginIncorrect(resp['message'])
+
+        return None
+
+class MobileWebAuthException(Exception):
+    pass
+
+class HTTPError(MobileWebAuthException):
+    pass
+
+class LoginIncorrect(MobileWebAuthException):
+    pass
+
+class CaptchaRequired(MobileWebAuthException):
+    pass
+
+class EmailCodeRequired(MobileWebAuthException):
+    pass
+
+class TwoFactorCodeRequired(MobileWebAuthException):
+    pass