Skip to content

Commit

Permalink
feat(test): use new feature
Browse files Browse the repository at this point in the history
  • Loading branch information
shidenggui committed Mar 5, 2016
1 parent fa35ad1 commit 3a2fe5b
Show file tree
Hide file tree
Showing 4 changed files with 39 additions and 7 deletions.
8 changes: 4 additions & 4 deletions easyquant/log_handler/default_handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,18 +10,18 @@
class DefaultLogHandler(object):
"""默认的 Log 类"""

def __init__(self, name='default', logtype='stream', filepath='default.log', loglevel='DEBUG'):
def __init__(self, name='default', log_type='stdout', filepath='default.log', loglevel='DEBUG'):
"""Log对象
:param name: log 名字
:param :logtype: 'stream' 输出到屏幕, 'file' 输出到指定文件
:param :logtype: 'stdout' 输出到屏幕, 'file' 输出到指定文件
:param :filename: log 文件名
:param :loglevel: 设定log等级 ['CRITICAL', 'ERROR', 'WARNING', 'NOTICE', 'INFO', 'DEBUG', 'TRACE', 'NOTSET']
:return log handler object
"""
self.log = Logger(name)
if logtype == 'stream':
if log_type == 'stdout':
StreamHandler(sys.stdout, level=loglevel).push_application()
if logtype == 'file':
if log_type == 'file':
if os.path.isdir(filepath) and not os.path.exists(filepath):
os.makedirs(os.path.dirname(filepath))
FileHandler(filepath, level=loglevel).push_application()
Expand Down
28 changes: 25 additions & 3 deletions test.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,17 @@
import easyquant
from easyquant import DefaultQuotationEngine, DefaultLogHandler

print('easyquant 测试 DEMO')
print('请输入你使用的券商:')
choose = input('1: 华泰 2: 佣金宝 \n:')
broker = 'ht' if choose == '1' else 'yjb'
choose = input('1: 华泰 2: 佣金宝 3: 银河 4: 雪球模拟组合\n:')

broker = 'ht'
if choose == '2':
broker = 'yjb'
elif choose == '3':
broker = 'yh'
elif choose == '4':
broker = 'xq'


def get_broker_need_data(choose_broker):
Expand All @@ -12,6 +20,20 @@ def get_broker_need_data(choose_broker):
return '%s.json' % choose_broker
return need_data

m = easyquant.MainEngine(broker, get_broker_need_data(broker))

need_data = get_broker_need_data(broker)

push_interval = int(input('请输入行情推送间隔(s):'))
DefaultQuotationEngine.PushInterval = push_interval

log_type_choose = input('请输入 log 记录方式: 1: 显示在屏幕 2: 记录在指定文件')
log_type = 'stdout' if log_type_choose == '1' else 'file'

if log_type == 'file':
log_filepath = input('请输入 log 文件记录路径')

log_handler = DefaultLogHandler(name='测试1', log_type=log_type, filepath=log_filepath)

m = easyquant.MainEngine(broker, need_data, quotation_engines=[DefaultQuotationEngine], log_handler=log_handler)
m.load_strategy()
m.start()
5 changes: 5 additions & 0 deletions xq.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
{
"account": "手机号",
"password": "加密后的密码",
"portfolio_code": "组合代码(例:ZH818559)"
}
5 changes: 5 additions & 0 deletions yh.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
{
"inputaccount": "客户号",
"trdpwd": "加密后的密码"
}

0 comments on commit 3a2fe5b

Please sign in to comment.