Main

pywidevine/clients/hbomax/client.py

@@ -0,0 +1,106 @@
+import base64, time, requests, os, json
+import pywidevine.clients.hbomax.config as hmaxcfg
+from os.path import join
+
+SESSION = requests.Session()
+HMAXTOKEN_FILE = join(hmaxcfg.COOKIES_FOLDER, 'hmax_login_data.json')
+
+
+login_config = {
+    'username': 'rivas909@me.com',
+    'password': 'NoCambieselPass.12345'
+}
+
+def login(SESSION, login_endpoint, content_url, save_login=True):
+    def get_free_token(token_url):
+        token_data = hmaxcfg.get_token_info()
+        free_token = requests.post(url=token_url, headers=token_data['headers'], json=token_data['data'])
+        if int(free_token.status_code) != 200:
+                print(free_token.json()['message'])
+                exit(1)
+        return free_token.json()['access_token']
+    free_access_tk = get_free_token(login_endpoint)
+    auth_data = hmaxcfg.get_auth_token_info(login_config)
+    headers = auth_data['headers']
+    headers['authorization'] = "Bearer {}".format(free_access_tk)
+    auth_rep = SESSION.post(url=login_endpoint, headers=headers, json=auth_data['data'])
+    if int(auth_rep.status_code) != 200:
+            print(auth_rep.json()['message'])
+            exit(1)
+
+    access_token_js = auth_rep.json()
+    
+    login_grant_access = [
+        {
+            "id": "urn:hbo:privacy-settings:mined",
+            "id": "urn:hbo:profiles:mined",
+            "id": "urn:hbo:query:lastplayed",
+            "id": "urn:hbo:user:me"}
+    ]
+    user_grant_access = {
+        "accept": "application/vnd.hbo.v9.full+json",
+        "accept-encoding": "gzip, deflate, br",
+        "accept-language": hmaxcfg.metadata_language,
+        "user-agent": hmaxcfg.UA,
+        "x-hbo-client-version": "Hadron/50.40.0.111 desktop (DESKTOP)",
+        "x-hbo-device-name": "desktop",
+        "x-hbo-device-os-version": "undefined",
+        "Authorization": f"Bearer {access_token_js['refresh_token']}"
+    }
+    user_grant_req = SESSION.post(content_url, json=login_grant_access, headers=user_grant_access)
+
+    if int(user_grant_req.status_code) != 207:
+        print("failed to list profiles")
+    
+    user_grant_js = user_grant_req.json()
+    user_grant_id = ""
+    
+    for profile in user_grant_js:
+        if profile['id'] == "urn:hbo:profiles:mine":
+            if len(profile['body']['profiles']) > 0:
+                user_grant_id = profile['body']['profiles'][0]['profileId']
+            else:
+                print("no profiles found, create one on hbomax and try again")
+                exit(1)
+
+    profile_headers = {
+        "accept": "application/vnd.hbo.v9.full+json",
+        "accept-encoding": "gzip, deflate, br",
+        "accept-language": hmaxcfg.metadata_language,
+        "user-agent": hmaxcfg.UA,
+        "x-hbo-client-version": "Hadron/50.40.0.111 desktop (DESKTOP)",
+        "x-hbo-device-name": "desktop",
+        "x-hbo-device-os-version": "undefined",
+        "referer": "https://play.hbomax.com/profileSelect",
+        "Authorization": f"Bearer {free_access_tk}" #~ free token
+    }
+
+    user_profile = {
+        "grant_type": "user_refresh_profile",
+        "profile_id": user_grant_id,
+        "refresh_token": f"{access_token_js['refresh_token']}",
+    }
+    
+    user_profile_req = SESSION.post(login_endpoint, json=user_profile, headers=profile_headers)
+
+    if int(user_profile_req.status_code) != 200:
+        error_msg = "failed to obatin the final token"
+        print(error_msg)
+
+    user_profile_js = user_profile_req.json()
+    
+    refresh_token = user_profile_js['refresh_token']
+
+    login_data = {'ACCESS_TOKEN': refresh_token, 'EXPIRATION_TIME': int(time.time())}
+    if save_login:
+        with open(HMAXTOKEN_FILE, 'w', encoding='utf-8') as f:
+            f.write(json.dumps(login_data, indent=4))
+            f.close()
+    return auth_rep.json()['access_token']
+
+
+def get_video_payload(urn):
+    headers = hmaxcfg.generate_payload()
+    payload = []
+    payload.append({"id":urn, "headers": headers['headers']})
+    return payload