aclgh/maquitous
1
0
Fork 0
forked from Fragrance/maquitous
Code Pull Requests Activity

feat: move requests to httpx

Browse Source
This commit is contained in:
7a1dd609-d238-4580-9d5f-ee8412b0f5bc
2025-01-25 10:57:28 +08:00
parent fd2f93e937
commit 0d27140dee
2 changed files with 20 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@@ -1,4 +1,4 @@
requests httpx
pycryptodome pycryptodome
pytz pytz
flask flask

22
sdgb.py
View File

@@ -3,7 +3,7 @@ import zlib
import pytz import pytz
import base64 import base64
import hashlib import hashlib
import requests import httpx
from datetime import datetime from datetime import datetime
from Crypto.Cipher import AES from Crypto.Cipher import AES
@@ -29,7 +29,6 @@ class aes_pkcs7(object):
def decrypt(self, content): def decrypt(self, content):
cipher = AES.new(self.key, AES.MODE_CBC, self.iv) cipher = AES.new(self.key, AES.MODE_CBC, self.iv)
#content = base64.b64decode(content)
return cipher.decrypt(content) return cipher.decrypt(content)
text = cipher.decrypt(content).decode('utf-8') text = cipher.decrypt(content).decode('utf-8')
return self.pkcs7unpadding(text) return self.pkcs7unpadding(text)
@@ -56,9 +55,10 @@ def sdgb_api(data, useApi, userId):
data = data data = data
data_enc = aes.encrypt(data) data_enc = aes.encrypt(data)
data_def = zlib.compress(data_enc) data_def = zlib.compress(data_enc)
requests.packages.urllib3.disable_warnings()
endpoint = "https://maimai-gm.wahlap.com:42081/Maimai2Servlet/" endpoint = "https://maimai-gm.wahlap.com:42081/Maimai2Servlet/"
r = requests.post(endpoint + get_hash_api(useApi), headers={ r = httpx.post(
endpoint + get_hash_api(useApi),
headers = {
"User-Agent": "%s#%d"%(get_hash_api(useApi), userId), "User-Agent": "%s#%d"%(get_hash_api(useApi), userId),
"Content-Type": "application/json", "Content-Type": "application/json",
"Mai-Encoding": "1.40", "Mai-Encoding": "1.40",
@@ -66,16 +66,15 @@ def sdgb_api(data, useApi, userId):
"Charset": "UTF-8", "Charset": "UTF-8",
"Content-Encoding": "deflate", "Content-Encoding": "deflate",
"Expect": "100-continue" "Expect": "100-continue"
}, data = data_def,verify=False) },
data = data_def
)
resp_def = r.content resp_def = r.content
try: try:
resp_enc = zlib.decompress(resp_def) resp_enc = zlib.decompress(resp_def)
#print(resp_enc)
except: except:
resp_enc = resp_def resp_enc = resp_def
#print("skipped")
#print(resp_enc)
return unpad(aes.decrypt(resp_enc), 16).decode() return unpad(aes.decrypt(resp_enc), 16).decode()
def qr_api(qr_code): def qr_api(qr_code):
@@ -96,7 +95,10 @@ def qr_api(qr_code):
"Host": "ai.sys-all.cn", "Host": "ai.sys-all.cn",
"User-Agent": "WC_AIME_LIB" "User-Agent": "WC_AIME_LIB"
} }
res = requests.post("http://ai.sys-allnet.cn/wc_aime/api/get_data", data=json.dumps(param, separators=(',', ':')), headers=headers) res = httpx.post(
# print(param) # debug打印param数据 "http://ai.sys-allnet.cn/wc_aime/api/get_data",
data = json.dumps(param, separators=(',', ':')),
headers = headers
)
assert res.status_code == 200, "网络错误" assert res.status_code == 200, "网络错误"
return json.loads(res.content) return json.loads(res.content)