def use(broker, debug=False, **kwargs):
"""用于生成特定的券商对象
:param broker:券商名支持 ['yh_client', '银河客户端'] ['ht_client', '华泰客户端']
:param debug: 控制 debug 日志的显示, 默认为 True
:param initial_assets: [雪球参数] 控制雪球初始资金，默认为一百万
:return the class of trader
Usage::
>>> import easytrader
>>> user = easytrader.use('xq')
>>> user.prepare('xq.json')
"""
if debug:
logger.setLevel(logging.DEBUG)
if broker.lower() in ["xq", "雪球"]:
return XueQiuTrader(**kwargs)
if broker.lower() in ["yh_client", "银河客户端"]:
from .yh_clienttrader import YHClientTrader
return YHClientTrader()
if broker.lower() in ["ht_client", "华泰客户端"]:
from .ht_clienttrader import HTClientTrader
return HTClientTrader()
if broker.lower() in ["wk_client", "五矿客户端"]:
from easytrader.wk_clienttrader import WKClientTrader
return WKClientTrader()
if broker.lower() in ["htzq_client", "海通证券客户端"]:
from easytrader.htzq_clienttrader import HTZQClientTrader
return HTZQClientTrader()
if broker.lower() in ["gj_client", "国金客户端"]:
from .gj_clienttrader import GJClientTrader
return GJClientTrader()
if broker.lower() in ["gf_client", "广发客户端"]:
from .gf_clienttrader import GFClientTrader
return GFClientTrader()
if broker.lower() in ["universal_client", "通用同花顺客户端"]:
from easytrader.universal_clienttrader import UniversalClientTrader
return UniversalClientTrader()
if broker.lower() in ["ths", "同花顺客户端"]:
from .clienttrader import ClientTrader
return ClientTrader()
raise NotImplementedError
def test_prepare_account(self):
user = XueQiuTrader()
params_without_cookies = dict(
portfolio_code="ZH123456", portfolio_market="cn"
)
with self.assertRaises(TypeError):
user._prepare_account(**params_without_cookies)
params_without_cookies.update(cookies="123")
user._prepare_account(**params_without_cookies)
self.assertEqual(params_without_cookies, user.account_config)