lifegpc/pythonscript
1
0
Fork 0
mirror of https://github.com/lifegpc/pythonscript.git synced 2026-06-06 11:28:58 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update pdtest

Browse Source
This commit is contained in:
lifegpc
2023-11-07 09:33:00 +08:00
parent aa349acf29
commit 9a8c4ebd50
1 changed files with 17 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
pdtest.py
View File

@@ -8,6 +8,7 @@ import requests as r
from urllib.parse import urlparse, urlencode from urllib.parse import urlparse, urlencode
from posixpath import basename from posixpath import basename
from json import dumps as _dumps from json import dumps as _dumps
from time import time
HOST = "http://localhost:8080" HOST = "http://localhost:8080"
HEADERS = {} HEADERS = {}
@@ -32,14 +33,9 @@ def post(path: str, token: Union[bytes, str] = None, token_id: int = None,
return r.post(f"{HOST}/{path}", data=data, headers=HEADERS) return r.post(f"{HOST}/{path}", data=data, headers=HEADERS)
if isinstance(token, str): if isinstance(token, str):
token = b64decode(token) token = b64decode(token)
keys = sorted([i for i in data.keys()]) if isinstance(data, dict):
u = sha512() data['t'] = f"{round(time())}"
u.update(token) h = {'X-TOKEN-ID': str(token_id), 'X-SIGN': gen_sign(data, token)}
for key in keys:
u.update(key.encode() if isinstance(key, str) else key)
v = data[key]
u.update(v.encode() if isinstance(v, str) else v)
h = {'X-TOKEN-ID': str(token_id), 'X-SIGN': u.hexdigest()}
h.update(HEADERS) h.update(HEADERS)
return r.post(f"{HOST}/{path}", headers=h, data=data) return r.post(f"{HOST}/{path}", headers=h, data=data)
@@ -63,6 +59,8 @@ def get(path: str, token: Union[bytes, str] = None, token_id: int = None,
return r.get(f"{HOST}/{path}", data=data, headers=HEADERS) return r.get(f"{HOST}/{path}", data=data, headers=HEADERS)
if isinstance(token, str): if isinstance(token, str):
token = b64decode(token) token = b64decode(token)
if isinstance(data, dict):
data['t'] = f"{round(time())}"
h = {'X-TOKEN-ID': str(token_id), 'X-SIGN': gen_sign(data, token)} h = {'X-TOKEN-ID': str(token_id), 'X-SIGN': gen_sign(data, token)}
h.update(HEADERS) h.update(HEADERS)
return r.get(f"{HOST}/{path}", headers=h, data=data) return r.get(f"{HOST}/{path}", headers=h, data=data)
@@ -87,6 +85,17 @@ def add_user(username: str, password: str, name: str,
return check_err(post("auth/user/add", token, token_id, {"username": username, "name": name, "password": b64encode(pas)}).json()) # noqa: E501 return check_err(post("auth/user/add", token, token_id, {"username": username, "name": name, "password": b64encode(pas)}).json()) # noqa: E501
def add_token(username: str, password: str):
re = check_err(post("/auth/pubkey").json())
key = ser.load_pem_public_key(re['key'].encode())
pas = key.encrypt(password.encode(), padding.PKCS1v15())
return check_err(post("/auth/token/add", None, None, {"username": username, "password": b64encode(pas)}).json()) # noqa: E501
def extend_token(token: Union[bytes, str], token_id: int):
return check_err(post("/auth/token/extend", token, token_id).json())
def get_proxy_pixiv_url(url: str, secrets: str): def get_proxy_pixiv_url(url: str, secrets: str):
u = urlparse(url) u = urlparse(url)
if u.scheme not in ['http', 'https']: if u.scheme not in ['http', 'https']: