From bd353f8d84947e7608a8f71bf57a25c219b4af3a Mon Sep 17 00:00:00 2001
From: AdminWhaleFall <whalefall9420@outlook.com>
Date: Wed, 27 Apr 2022 01:36:04 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9E=20fix:=20=E7=B4=A7=E6=80=A5?=
 =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E4=B8=80=E4=B8=AA=E4=B8=A5=E9=87=8Dbug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 flask_app/model.py   | 6 +++++-
 flask_app/utils.py   | 1 -
 smsboom.py           | 2 --
 smsboom_pyinstall.py | 1 -
 utils.py             | 7 +++++--
 5 files changed, 10 insertions(+), 7 deletions(-)

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)