知微
fa45d8aa5f
- health_checklist.json: 192.168.1.122→node122
- ocr_client.py: docstring IP→node122
- docs/market-data-requirements.md: IP→node122
- 所有API调用通过ProxyHandler({})绕过系统代理
Privoxy对node122:18003返回500，直连正常
75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
#!/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
|