feat: 增加http请求的ua头 (#1)

Co-authored-by: veehou <veehou@tencent.com>
tencent-connect · Jan 13, 2022 · 49f31a3 · 49f31a3
1 parent 9522ded
commit 49f31a3
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 6 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -1,6 +1,6 @@
 repos:
   - repo: https://github.com/ambv/black
-    rev: stable
+    rev: 21.12b0
     hooks:
       - id: black
   - repo: https://github.com/pre-commit/pre-commit-hooks

diff --git a/examples/config.yaml b/examples/config.yaml
@@ -0,0 +1,4 @@
+# test yaml 用于设置test相关的参数，开源版本需要去掉参数
+token:
+  appid: "101981207"
+  token: "Ugqr4v71Xwd4uacrYjjHeYPVOxWSQKCy"
diff --git a/qqbot/core/network/http.py b/qqbot/core/network/http.py
@@ -68,7 +68,10 @@ def __init__(self, time_out, token, type):
         self.scheme = type
 
     def get(self, api_url, params=None):
-        headers = {"Authorization": self.scheme + " " + self.token}
+        headers = {
+            "Authorization": self.scheme + " " + self.token,
+            "User-Agent": "botpy",
+        }
         logger.debug("http get headers: %s, api_url: %s" % (headers, api_url))
         response = requests.get(
             url=api_url, params=params, timeout=self.timeout, headers=headers
@@ -77,7 +80,10 @@ def get(self, api_url, params=None):
         return response
 
     def post(self, api_url, request=None, params=None):
-        headers = {"Authorization": self.scheme + " " + self.token}
+        headers = {
+            "Authorization": self.scheme + " " + self.token,
+            "User-Agent": "botpy",
+        }
         logger.debug(
             "http post headers: %s, api_url: %s, request: %s"
             % (headers, api_url, request)
@@ -93,7 +99,10 @@ def post(self, api_url, request=None, params=None):
         return response
 
     def delete(self, api_url, request=None, params=None):
-        headers = {"Authorization": self.scheme + " " + self.token}
+        headers = {
+            "Authorization": self.scheme + " " + self.token,
+            "User-Agent": "botpy",
+        }
         logger.debug("http delete headers: %s, api_url: %s" % (headers, api_url))
         response = requests.delete(
             url=api_url,
@@ -106,7 +115,10 @@ def delete(self, api_url, request=None, params=None):
         return response
 
     def put(self, api_url, request=None, params=None):
-        headers = {"Authorization": self.scheme + " " + self.token}
+        headers = {
+            "Authorization": self.scheme + " " + self.token,
+            "User-Agent": "botpy",
+        }
         logger.debug(
             "http put headers: %s, api_url: %s, request: %s"
             % (headers, api_url, request)
@@ -122,7 +134,10 @@ def put(self, api_url, request=None, params=None):
         return response
 
     def patch(self, api_url, request=None, params=None):
-        headers = {"Authorization": self.scheme + " " + self.token}
+        headers = {
+            "Authorization": self.scheme + " " + self.token,
+            "User-Agent": "botpy",
+        }
         logger.debug(
             "http patch headers: %s, api_url: %s, request: %s"
             % (headers, api_url, request)