mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-16 12:02:16 +00:00
64 lines
1.5 KiB
Python
64 lines
1.5 KiB
Python
import hashlib
|
|
import random
|
|
import string
|
|
import time
|
|
import uuid
|
|
|
|
RECOGNIZE_SERVER = {
|
|
"1": "cn_gf01",
|
|
"2": "cn_gf01",
|
|
"5": "cn_qd01",
|
|
"6": "os_usa",
|
|
"7": "os_euro",
|
|
"8": "os_asia",
|
|
"9": "os_cht",
|
|
}
|
|
|
|
|
|
def get_device_id(name: str) -> str:
|
|
return str(uuid.uuid3(uuid.NAMESPACE_URL, name)).replace("-", "").upper()
|
|
|
|
|
|
def md5(text: str) -> str:
|
|
_md5 = hashlib.md5() # nosec B303
|
|
_md5.update(text.encode())
|
|
return _md5.hexdigest()
|
|
|
|
|
|
def random_text(num: int) -> str:
|
|
return "".join(random.sample(string.ascii_lowercase + string.digits, num))
|
|
|
|
|
|
def timestamp() -> int:
|
|
return int(time.time())
|
|
|
|
|
|
def get_ds(salt: str = "", web: int = 1) -> str:
|
|
if salt == "":
|
|
if web == 1:
|
|
salt = "h8w582wxwgqvahcdkpvdhbh2w9casgfl"
|
|
elif web == 2:
|
|
salt = "h8w582wxwgqvahcdkpvdhbh2w9casgfl"
|
|
elif web == 3:
|
|
salt = "fd3ykrh7o1j54g581upo1tvpam0dsgtf"
|
|
i = str(timestamp())
|
|
r = random_text(6)
|
|
c = md5("salt=" + salt + "&t=" + i + "&r=" + r)
|
|
return f"{i},{r},{c}"
|
|
|
|
|
|
def get_recognize_server(uid: int) -> str:
|
|
server = RECOGNIZE_SERVER.get(str(uid)[0])
|
|
if server:
|
|
return server
|
|
else:
|
|
raise TypeError(f"UID {uid} isn't associated with any recognize server")
|
|
|
|
|
|
def get_headers():
|
|
headers = {
|
|
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
|
|
"AppleWebKit/537.36 (KHTML, like Gecko) Chrome/102.0.5005.63 Safari/537.36",
|
|
}
|
|
return headers
|