Arthur-Wu commit

commons/SignatureYM.py

@@ -0,0 +1,32 @@
+# !/usr/bin/python
+# -*- coding: utf-8 -*-
+from authlib.integrations.requests_client import OAuth2Session
+
+
+class SignatureYM():
+
+    def __init__(self):
+        self.client_id = "95579765e4fe91af9962"
+        self.client_secret = "3dcb964ede5d455f1d54623dad5e496bff468f81"
+        self.redirect_uri = 'https://auth.yimaogo.com/api/callback'
+        self.authorization_base_url = 'https://auth.yimaogo.com/api/login/oauth/authorize'
+        self.token_url = 'https://auth.yimaogo.com/api/login/oauth/access_token'
+
+    def __tokens(self):
+        oauth = OAuth2Session(self.client_id, redirect_uri=self.redirect_uri)
+        authorization_url, state = oauth.create_authorization_url(self.authorization_base_url)
+        token_dict = oauth.fetch_token(self.token_url, authorization_response={}, client_secret=self.client_secret)
+        return token_dict["access_token"]
+
+    def return_headers(self):
+        tokens = self.__tokens()
+        headers = {
+            'Authorization': tokens,
+            'User-Agent': 'Apifox/1.0.0 (https://apifox.com)',
+            'Content-Type': 'application/json',
+            'Accept': '*/*',
+            'Host': 'api.test.yimaogo.com',
+            'Connection': 'keep-alive'
+        }
+        return headers
+