diff --git a/flask_app/model.py b/flask_app/model.py index 0fba7f4..c8a425d 100644 --- a/flask_app/model.py +++ b/flask_app/model.py @@ -58,8 +58,12 @@ class API(BaseModel): :param API: one API basemodel :return: API basemodel """ - if isinstance(self.header, str) and self.header: + if self.header == "": + self.header = {} + self.header['Referer'] = self.url # 增加 Referer + else: self.header = self.replace_data(self.header, phone) + self.data = self.replace_data(self.data, phone) self.url = self.replace_data(self.url, phone) return self diff --git a/flask_app/utils.py b/flask_app/utils.py index d70936e..ffe3691 100644 --- a/flask_app/utils.py +++ b/flask_app/utils.py @@ -11,7 +11,6 @@ def test_resq(api: API, phone) -> httpx.Response: :return: httpx 请求对象. """ api = api.handle_API(phone) - default_header['Referer'] = api.url with httpx.Client(headers=default_header, timeout=8) as client: if not isinstance(api.data, dict): print("data") diff --git a/smsboom.py b/smsboom.py index 63c57a2..4f27ddc 100644 --- a/smsboom.py +++ b/smsboom.py @@ -88,8 +88,6 @@ def req(api: Union[API, str], phone: tuple): else: phone_lst = [phone] - # 自动添加 Referer - default_header['Referer'] = api.url with httpx.Client(headers=default_header, verify=False) as client: for ph in phone_lst: try: diff --git a/smsboom_pyinstall.py b/smsboom_pyinstall.py index 93a8d4a..d1dbcb1 100644 --- a/smsboom_pyinstall.py +++ b/smsboom_pyinstall.py @@ -89,7 +89,6 @@ def req(api: Union[API, str], phone: tuple): phone_lst = [_ for _ in phone] else: phone_lst = [phone] - default_header['Referer'] = api.url with httpx.Client(headers=default_header, verify=False) as client: for ph in phone_lst: try: diff --git a/utils.py b/utils.py index 39e9136..5e1633b 100644 --- a/utils.py +++ b/utils.py @@ -101,8 +101,11 @@ class API(BaseModel): :return: API basemodel """ # 如果传入的 header 是字符串,就转为字典. - if (isinstance(self.header, str) and self.header): - self.header = json.loads(self.header.replace("'", '"')) + if self.header == "": + self.header = {} + self.header['Referer'] = self.url # 增加 Referer + else: + self.header = self.replace_data(self.header, phone) self.data = self.replace_data(self.data, phone) self.url = self.replace_data(self.url, phone)