Manage JoinQuant following in scheduler script

Author: sinall

README.rst

@@ -8,11 +8,24 @@ ShiPanE-Python-SDK
 | 详情见：http://www.iguuu.com/e
 | 交流QQ群：11527956 |实盘易-股票自动交易|
 
-自动新股申购
-------------
+功能介绍
+--------
 
-一. Windows
-~~~~~~~~~~~~
+- 简单的实盘易 HTTP API 封装，见 shipane_sdk/client.py
+- 多账号自动新股申购（自动打新）
+- 聚宽集成
+
+定时任务调度
+--------------
+
+- 多账号自动新股申购（自动打新）
+- 聚宽跟单（抓取方式）
+
+安装
+~~~~
+
+Windows
+^^^^^^^
 
 - 安装 Python 3.5（建议安装 Anaconda3）
 - cmd 中运行：pip install --no-binary shipane_sdk shipane_sdk
@@ -21,15 +34,25 @@ ShiPanE-Python-SDK
 - 找到 python 安装目录，例如：C:\\Program Files\\Anaconda3
 - cmd 下执行（具体路径自行修改）：python "C:\\Program Files\\Anaconda3\\Scripts\\shipane-scheduler.py"
 
-二. Mac/Linux
-~~~~~~~~~~~~~
+Mac/Linux
+^^^^^^^^^
 
 - 安装 Python 3.5
 - terminal 中运行：pip install --no-binary shipane_sdk shipane_sdk
 - terminal 中运行：cp -n ~/.shipane_sdk/config/scheduler-example.ini ~/.shipane_sdk/config/scheduler.ini
 - 修改 ~/.shipane_sdk/config/scheduler.ini
 - terminal 中运行：shipane-scheduler.py
 
+升级
+~~~~~
+
+pip install --upgrade --no-deps --no-binary shipane_sdk shipane_sdk
+
+配置
+~~~~~
+
+定时任务默认禁用；如需启动，请设置 enabled=true
+
 聚宽集成
 --------
 
@@ -48,11 +71,9 @@ ShiPanE-Python-SDK
 步骤
 ^^^^
 
--  将 shipane\_sdk/client.py 上传至聚宽“投资研究”根目录，并重命名为
-   shipane\_sdk.py。
+-  将 shipane\_sdk/client.py 上传至聚宽“投资研究”根目录，并重命名为 shipane\_sdk.py。
 -  将 shipane\_sdk/joinquant/executor.py 追加到 shpane\_sdk.py 中。
--  用法请参考 examples/joinquant/simple\_strategy.py (注意将其中的
-   xxx.xxx.xxx.xxx 替换为实际 IP)。
+-  用法请参考 examples/joinquant/simple\_strategy.py (注意将其中的 xxx.xxx.xxx.xxx 替换为实际 IP)。
 
 二. 抓取方式
 ~~~~~~~~~~~~
@@ -68,11 +89,7 @@ ShiPanE-Python-SDK
 步骤
 ^^^^
 
--  git clone 或下载项目到本地。
--  安装必要的依赖 "pip install requests"。
--  参考 examples/joinquant/config/config.ini.template 创建
-   examples/joinquant/config/config.ini，并完善配置。
--  命令行运行 "python ./examples/joinquant/simple\_runner.py"。
+见 `定时任务调度 <#定时任务调度>`__
 
 其他语言 SDK
 ------------

config/scheduler-example.ini

@@ -1,3 +1,17 @@
+; schedule 参数设置，类似于 cron 表达式
+;
+; 格式为：[秒(0-59)] [分(0-59)] [时(0-23)] [星期几(0-6 或英文缩写)] [星期(1-53)] [日(1-31)] [月(1-12)] [年(四位数字)]
+; （星期几英文缩写：mon,tue,wed,thu,fri,sat,sun）
+;
+; 字段支持表达式：
+; - * 为任意单位时间触发
+; - */a 为每 a 个单位时间触发
+; - a-b 为 a 到 b 的区间触发
+; - a-b/c 为 a 到 b 的区间每 c 个单位时间触发
+;
+; 详见：https://apscheduler.readthedocs.io/en/v2.1.2/cronschedule.html
+;
+
 ;
 ; 实盘易配置
 ;
@@ -21,23 +35,25 @@ client2=title:guotai
 ; 是否启用？
 enabled=false
 
-; 触发时间设置，类似于 cron 表达式
-;
-; 格式为：[秒(0-59)] [分(0-59)] [时(0-23)] [星期几(0-6 或英文缩写)] [星期(1-53)] [日(1-31)] [月(1-12)] [年(四位数字)]
-; （星期几英文缩写：mon,tue,wed,thu,fri,sat,sun）
-;
-; 字段支持表达式：
-; - * 为任意单位时间触发
-; - */a 为每 a 个单位时间触发
-; - a-b 为 a 到 b 的区间触发
-; - a-b/c 为 a 到 b 的区间每 c 个单位时间触发
-;
-; 详见：https://apscheduler.readthedocs.io/en/v2.1.2/cronschedule.html
-;
 ; 默认设置为：星期一至星期五中午十二点
-;
 schedule=0 0 12 mon-fri * * * *
 
 ; 需要自动新股申购的交易客户端列表，以,（半角逗号）分割
 ; clients=client1,client2
 clients=client1
+
+[JoinQuant]
+username=
+password=
+
+; 模拟交易 URL 中 backtestId= 后面的那串字符
+; 例如，模拟交易 URL 为：https://www.joinquant.com/algorithm/live/index?backtestId=c215e5e57b30a65df4139bfff8c90e99
+; 则此处填写：c215e5e57b30a65df4139bfff8c90e99
+backtest_id=
+
+; 是否启用？
+enabled=false
+
+;
+; 默认设置为：星期一至星期五 9:00 到 15:00 每分钟的第 30 秒
+schedule=30 */1 9-15 mon-fri * * * *

examples/joinquant/config/config.ini.template

@@ -0,0 +1,67 @@
+# -*- coding: utf-8 -*-
+
+import logging
+from datetime import datetime
+
+from shipane_sdk.joinquant.client import JoinQuantClient
+from shipane_sdk.joinquant.transaction import JoinQuantTransaction
+from shipane_sdk.market_utils import MarketUtils
+
+
+class JoinQuantFollowingJob(object):
+    def __init__(self, config, client):
+        self._log = logging.getLogger()
+        self._config = config
+        self._client = client
+        self._jq_client = JoinQuantClient(username=self._config.get('JoinQuant', 'username'),
+                                          password=self._config.get('JoinQuant', 'password'),
+                                          backtest_id=self._config.get('JoinQuant', 'backtest_id'))
+        self._jq_client.login()
+        self._start_datatime = datetime.now()
+        self._processed_transactions = []
+
+    def __call__(self):
+        if MarketUtils.is_closed():
+            self._log.warning("********** 休市期间不跟单 **********")
+            if self._processed_transactions:
+                del self._processed_transactions[:]
+            return
+
+        self._log.info("********** 开始跟单 **********")
+        try:
+            transaction_detail = self._jq_client.query()
+            raw_transactions = transaction_detail['data']['transaction']
+            self._log.info("获取到 {} 条委托".format(len(raw_transactions)))
+
+            transactions = []
+            for raw_transaction in raw_transactions:
+                transaction = JoinQuantTransaction(raw_transaction).normalize()
+                if self._is_expired(transaction):
+                    continue
+
+                transactions.append(transaction)
+            self._log.info("获取到 {} 条有效委托".format(len(transactions)))
+
+            for tx in transactions:
+                self._processed_transactions.append(tx)
+                self._log.info("开始以 {}元 {} {}股 {}".format(tx.price, tx.get_cn_type(), tx.amount, tx.symbol))
+                response = self._shipane_client.execute(None,
+                                                        action=tx.type,
+                                                        symbol=tx.symbol,
+                                                        type='LIMIT',
+                                                        price=tx.price,
+                                                        amount=tx.amount)
+                if response is not None:
+                    self._log.info(u'实盘易回复：\nstatus_code: %d\ntext: %s', response.status_code, response.text)
+                else:
+                    self._log.error('实盘易未回复')
+        except Exception as e:
+            self._log.exception("跟单异常")
+        self._log.info("********** 结束跟单 **********\n")
+
+    def _is_expired(self, transaction):
+        if transaction.completed_at < self._start_datatime:
+            return True
+        if transaction in self._processed_transactions:
+            return True
+        return False

examples/joinquant/simple_runner.py

@@ -32,7 +32,7 @@ def login(self):
             'ajax': 1
         })
         self._session.headers.update({
-            'cookie': response.headers['set-Cookie']
+            'cookie': response.headers['Set-Cookie']
         })
 
     def query(self):

shipane_sdk/jobs/joinquant_following.py

@@ -11,6 +11,7 @@
 
 from shipane_sdk import Client
 from shipane_sdk.ap import APCronParser
+from shipane_sdk.jobs.joinquant_following import JoinQuantFollowingJob
 from shipane_sdk.jobs.new_stock_purchase import NewStockPurchaseJob
 
 if six.PY2:
@@ -34,13 +35,22 @@ def __init__(self):
                               key=self._config.get('ShiPanE', 'key'))
 
         self._new_stock_purchase_job = NewStockPurchaseJob(self._config, self._client)
+        self._joinquant_following_job = JoinQuantFollowingJob(self._config, self._client)
 
     def start(self):
         scheduler = BackgroundScheduler()
 
         if self._config.getboolean('NewStocks', 'enabled'):
             scheduler.add_job(self._new_stock_purchase_job,
                               APCronParser.parse(self._config.get('NewStocks', 'schedule')))
+        else:
+            self._log.warning('New stock purchase job is not enabled')
+
+        if self._config.getboolean('JoinQuant', 'enabled'):
+            scheduler.add_job(self._joinquant_following_job,
+                              APCronParser.parse(self._config.get('JoinQuant', 'schedule')))
+        else:
+            self._log.warning('JoinQuant following job is not enabled')
 
         scheduler.start()
         print('Press Ctrl+{0} to exit'.format('Break' if os.name == 'nt' else 'C'))