统一请求API

music_dl/api.py

@@ -0,0 +1,33 @@
+#!/usr/bin/env python
+# -*- coding:utf-8 -*-
+"""
+@author: HJK
+@file: api.py
+@time: 2019-06-11
+"""
+
+import requests
+from . import config
+from .exceptions import RequestError, ResponseError, DataError
+
+
+class MusicApi:
+    # class property
+    # 子类修改时使用deepcopy
+    session = requests.Session()
+    session.headers.update(config.get("fake_headers"))
+    if config.get("proxies"):
+        session.proxies.update(config.get("proxies"))
+    session.headers.update({"referer": "http://www.google.com/"})
+
+    @classmethod
+    def request(cls, url, method="POST", data=None):
+        if method == "GET":
+            resp = cls.session.get(url, params=data, timeout=7)
+        else:
+            resp = cls.session.post(url, data=data, timeout=7)
+        if resp.status_code != requests.codes.ok:
+            raise RequestError(resp.text)
+        if not resp.text:
+            raise ResponseError("No response data.")
+        return resp.json()