backup/SMSBoom
1
0
Fork 0
mirror of https://github.com/WhaleFell/SMSBoom.git synced 2026-04-13 18:10:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

✨ feat: 接口在线web基本完工.

Browse Source
This commit is contained in:
AdminWhaleFall
2022-04-05 02:16:09 +08:00
parent 0b6bea4c87
commit 38389ab41e
6 changed files with 181 additions and 227 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
flask_app/utils.py
View File

@@ -42,31 +42,35 @@ class API(BaseModel):
return v
def replace_data(self, content: Union[str, dict], phone) -> str:
if not phone:
return content
if isinstance(content, dict):
for key, value in content.items():
content[key] = value.replace("{phone}", phone).replace(
"{timestamp}", self.timestamp_new())
content[key] = value.replace("[phone]", phone).replace(
"[timestamp]", self.timestamp_new())
else:
if isinstance(content, str):
content.replace("{phone}", phone).replace(
"{timestamp}", self.timestamp_new())
content.replace('[phone]', phone).replace(
'[timestamp]', self.timestamp_new())
return content
def timestamp_new(self) -> str:
"""返回整数字符串时间戳"""
return str(int(datetime.now().timestamp()))
def handle_API(self, phone):
def handle_API(self, phone=None):
"""
:param API: one API basemodel
:return: API basemodel
"""
if self.method != "POST":
self.method = "GET"
self.data = self.replace_data(self.data, phone)
self.url = self.replace_data(self.url, phone)
if isinstance(self.data, str):
self.data = json.loads(self.data)
if isinstance(self.header, str):
self.header = json.loads(self.header)
self.data = self.replace_data(self.data, phone)
self.url = self.replace_data(self.url, phone)
return self
@@ -77,6 +81,7 @@ def test_resq(api: API, phone) -> httpx.Response:
:return: httpx 请求对象.
"""
api = api.handle_API(phone)
print(api.dict())
with httpx.Client(headers=default_header, timeout=8) as client:
if not isinstance(api.data, dict):
client.request(method=api.method, headers=api.header,