Update app.py

app.py

@@ -88,6 +88,29 @@ def get_hash(str):
     return hashlib.md5(str.encode()).hexdigest()
 
 
+Name = ["VIVO-", 'OPPO-', 'HUAWEI-', 'Honor-', 'XIAOMI', 'HONGMI-', 'APPLE-']
+Type = ["S ", " pro ", "A "]
+
+
+def B():
+    timestamp = str(int(time.time()) * 1000)
+    device_id = str(uuid.uuid4()).replace("-", "")
+    phoneModel = ""
+    for i in range(3):
+        r = chr(random.randint(65, 90))
+        phoneModel += r
+    phoneModel = phoneModel + "-"
+    for i in range(5):
+        if random.randint(0, 4) != 2:
+            r = chr(random.randint(65, 90))
+        else:
+            r = chr(random.randint(48, 57))
+        phoneModel += r
+    phoneBuilder = Name[random.randint(0, 6)] + str(random.randint(3, 9)) + Type[random.randint(0, 2)] + str(
+        random.randint(5, 6)) + "G"
+    return timestamp, device_id, phoneModel, phoneBuilder
+
+    
 async def get_change_ip():
     m = random.randint(0, 255)
     n = random.randint(0, 255)
@@ -97,15 +120,18 @@ async def get_change_ip():
     return randomIP
 
 
-# 创建随机UA
-async def get_user_agent():
-    tmp1 = random.randrange(90, 120)
-    tmp2 = random.randrange(5200, 5500)
-    tmp3 = random.randrange(90, 180)
-    tmp_version = str(tmp1) + ".0." + str(tmp2) + "." + str(tmp3)
-    ua = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/' + tmp_version + ' Safari/537.36'
-    print(ua)
-    return ua
+def get_UA(id, ua_key, timestamp, phoneModel, phoneBuilder, version):
+    UA = "ANDROID-com.pikcloud.pikpak/" + version + " accessmode/ devicename/" + phoneBuilder + " appname/ android-com.pikcloud.pikpak appid/ action_type/ clientid/ YNxT9w7GMdWvEOKa deviceid/" + id + "refresh_token/ grant_type/ devicemodel/ " + phoneModel + " networktype/WIFI accesstype/ sessionid/ osversion/" + "13." + str(
+        random.randint(0, 9)) + "." + str(random.randint(0,
+                                                         9)) + " datetime/" + timestamp + " protocolversion/200 sdkversion/2.0.1.200200 clientversion/" + version + " providername/NONE clientip/ session_origin/ devicesign/div101." + ua_key + "platformversion/10 usrno/"
+    return UA
+
+
+def get_ua_key(id):
+    k = "com.pikcloud.pikpak1appkey"
+    rank_1 = hashlib.sha1((id + k).encode("utf-8")).hexdigest()
+    rank_2 = get_hash(rank_1)
+    return id + rank_2
 
 
 # 设置请求头基本信息
@@ -131,13 +157,12 @@ async def get_image_d(pid, device_id, f, change_ip):
     }
     response = requests.request("POST", url, json=payload, headers=headers)
     response_data = response.json()
-    # print(f"d:{response_data['d']}")
     return response_data['d']
 
 
 # 安全验证-登录
 async def init(client_id, device_id, User_Agent, mail, change_ip):
-    url = f"https://user.mypikpak.net/v1/shield/captcha/init"
+    url = f"https://user.mypikpak.com/v1/shield/captcha/init?client_id={client_id}"
     payload = {
         "action": "POST:/v1/auth/signin",
         "captcha_token": '',
@@ -148,13 +173,19 @@ async def init(client_id, device_id, User_Agent, mail, change_ip):
         },
     }
     headers = {
-        "x-client-id": "YUMx5nI8ZU8Ap8pm",
         "X-Device-Id": device_id,
         "User-Agent": User_Agent,
         'X-Forwarded-For': str(change_ip)
     }
     headers.update(basicRequestHeaders_1)
     response = requests.request("POST", url, json=payload, headers=headers)
+    if response.status_code != 200:
+        error_response = response.json()
+        message = None
+        if "details" in error_response and len(error_response["details"]) > 0:
+            last_detail = error_response["details"][-1]
+            message = last_detail.get("message", "操作频繁...")
+        raise Exception(f"{message}")
     return json.loads(response.text)
 
 
@@ -482,11 +513,15 @@ async def login():
         return jsonify({"error": "没有指定参数"}), 400
     email = data.get('userName')
     password = data.get('passWord')
+    timestamp = B()[0]
+    phoneModel = B()[2]
+    phoneBuilder = B()[3]
     try:
         change_ip = await get_change_ip()
         client_id1 = "YNxT9w7GMdWvEOKa"
         device_id = str(uuid.uuid4()).replace("-", "")
-        user_agent = await get_user_agent()
+        ua_key = get_ua_key(device_id)
+        user_agent = get_UA(device_id, ua_key, timestamp, phoneModel, phoneBuilder, "1.47.1")
         action_init = await init(client_id1, device_id, user_agent, email, change_ip)
         captcha_token = action_init["captcha_token"]
         if 'url' in action_init.keys():
@@ -504,7 +539,7 @@ async def login():
             return jsonify({"code": 200, "data": login_data, "msg": 'success'}), 200
     except Exception as e:
         print(e)
-        return jsonify({"code": 500, "data": {}, "msg": '登录失败'}), 500
+        return jsonify({"code": 500, "data": {}, "msg": str(e)}), 200
 
 
 config = Config()