知微
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，直连正常
70 lines
1.7 KiB
Python
70 lines
1.7 KiB
Python
# -*- coding:utf-8 -*-
|
|
|
|
"""
|
|
Created on 2015/08/24
|
|
@author: Jimmy Liu
|
|
@group : waditu
|
|
@contact: jimmysoa@sina.cn
|
|
"""
|
|
|
|
import pandas as pd
|
|
import os
|
|
from tushare.stock import cons as ct
|
|
|
|
BK = 'bk'
|
|
|
|
def set_token(token):
|
|
df = pd.DataFrame([token], columns=['token'])
|
|
user_home = os.path.expanduser('~')
|
|
fp = os.path.join(user_home, ct.TOKEN_F_P)
|
|
df.to_csv(fp, index=False)
|
|
|
|
|
|
def get_token():
|
|
# 通过环境变量TUSHARE_TOKEN或者TS_TOKEN，获取token
|
|
token = os.getenv('TUSHARE_TOKEN') or os.getenv('TS_TOKEN')
|
|
if token:
|
|
return token
|
|
|
|
# 通过加目录中的文件 tk.csv 来获取token
|
|
user_home = os.path.expanduser('~')
|
|
fp = os.path.join(user_home, ct.TOKEN_F_P)
|
|
if os.path.exists(fp):
|
|
df = pd.read_csv(fp)
|
|
token = str(df.loc[0]['token'])
|
|
if token:
|
|
return token
|
|
|
|
print(ct.TOKEN_ERR_MSG)
|
|
return None
|
|
|
|
|
|
def set_broker(broker='', user='', passwd=''):
|
|
df = pd.DataFrame([[broker, user, passwd]],
|
|
columns=['broker', 'user', 'passwd'],
|
|
dtype=object)
|
|
if os.path.exists(BK):
|
|
all = pd.read_csv(BK, dtype=object)
|
|
if (all[all.broker == broker]['user']).any():
|
|
all = all[all.broker != broker]
|
|
all = all.append(df, ignore_index=True)
|
|
all.to_csv(BK, index=False)
|
|
else:
|
|
df.to_csv(BK, index=False)
|
|
|
|
|
|
def get_broker(broker=''):
|
|
if os.path.exists(BK):
|
|
df = pd.read_csv(BK, dtype=object)
|
|
if broker == '':
|
|
return df
|
|
else:
|
|
return df[df.broker == broker]
|
|
else:
|
|
return None
|
|
|
|
|
|
def remove_broker():
|
|
os.remove(BK)
|
|
|