stock-tracker

2026-07-04 00:17:11 +08:00
commit 6087341a48
6463 changed files with 1929869 additions and 0 deletions
@@ -0,0 +1,74 @@
+#!/usr/bin/env python
+# -*- coding:utf-8 -*-
+"""
+Date: 2019/11/10 22:52
+Desc: 数据接口源代码
+"""
+
+from functools import partial
+from urllib import parse
+
+import pandas as pd
+import requests
+
+
+class DataApi:
+    __token = ""
+    __http_url = "https://api.qhkch.com"
+
+    def __init__(self, token, timeout=10):
+        """
+        初始化函数
+        :param token: API接口TOKEN，用于用户认证
+        :type token: str
+        :param timeout: 超时设置
+        :type timeout: int
+        """
+        self.__token = token
+        self.__timeout = timeout
+
+    def query(self, api_name, fields="", **kwargs):
+        """
+        :param api_name: 需要调取的接口
+        :type api_name: str
+        :param fields: 想要获取的字段
+        :type fields: str
+        :param kwargs: 指定需要输入的参数
+        :type kwargs: 键值对
+        :return: 指定的数据
+        :rtype: dict or pandas.DataFrame
+        """
+        headers = {
+            "X-Token": self.__token,
+        }
+        url = parse.urljoin(self.__http_url, "/".join([api_name, *kwargs.values()]))
+        res = requests.get(url, headers=headers, timeout=self.__timeout)
+        if res.status_code != 200:
+            raise Exception("连接异常, 请检查您的Token是否过期和输入的参数是否正确")
+        data_json = res.json()
+        if fields == "":
+            try:
+                return pd.DataFrame(data_json)
+            except ValueError:
+                result_df = pd.DataFrame.from_dict(
+                    data_json, orient="index", columns=[api_name]
+                )
+                return result_df
+        else:  # 此处增加处理
+            if api_name == "variety_all_positions":
+                big_df = pd.DataFrame()
+                for item in data_json[fields].keys():
+                    temp_df = pd.DataFrame(data_json[fields][item])
+                    temp_df["code"] = item
+                    big_df = big_df.append(temp_df, ignore_index=True)
+                big_df.reset_index(inplace=True, drop=True)
+                return big_df
+            else:
+                return pd.DataFrame(data_json[fields])
+
+    def __getattr__(self, name):
+        return partial(self.query, name)
+
+
+if __name__ == "__main__":
+    pass
@@ -0,0 +1,9 @@
+#!/usr/bin/env python
+# -*- coding:utf-8 -*-
+"""
+Date: 2020/12/24 16:28
+Desc: API常量文件
+"""
+
+TOKEN_F_P = "tk.csv"
+TOKEN_ERR_MSG = "请设置 AKShare pro 的 token 凭证码，如果没有权限，请访问 https://qhkch.com/ 注册申请"
@@ -0,0 +1,30 @@
+#!/usr/bin/env python
+# -*- coding:utf-8 -*-
+"""
+Date: 2019/11/10 22:52
+Desc: 数据接口初始化
+"""
+
+from akshare.pro import client
+from akshare.utils import token_process
+
+
+def pro_api(token=""):
+    """
+    初始化 pro API,第一次可以通过ak.set_token('your token')来记录自己的token凭证，临时token可以通过本参数传入
+    """
+    if token == "" or token is None:
+        token = token_process.get_token()
+    if token is not None and token != "":
+        pro = client.DataApi(token)
+        return pro
+    else:
+        raise Exception("api init error.")
+
+
+if __name__ == "__main__":
+    pro_test = pro_api()
+    variety_all_df = pro_test.variety_all()
+    print(variety_all_df)
+    variety_no_futures_df = pro_test.variety_no_futures(symbol="RB", date="2018-08-08")
+    print(variety_no_futures_df)