tushare 1.4.5__py3-none-any.whl → 1.4.6__py3-none-any.whl

tushare/subs/model/tick.py

@@ -1,128 +1,128 @@
-from datetime import datetime
-from typing import Optional
-
-from pydantic import BaseModel
-
-
-class TsTick(BaseModel):
-    ts_code: str
-    name: Optional[str]
-    trade_time: Optional[datetime]
-    pre_close_price: Optional[float]
-    last_price: Optional[float]
-    open_price: Optional[float]
-    high_price: Optional[float]
-    low_price: Optional[float]
-    close_price: Optional[float]
-    volume: Optional[int]           # 成交量
-    amount: Optional[int]           # 成交金额
-    count: Optional[int]            # 交易数量
-    ask_price1: Optional[float]     # 申卖价, 委托档位卖1档价格
-    ask_volume1: Optional[int]      # 申卖量
-    bid_price1: Optional[float]     # 申买价, 委托档位买1档价格
-    bid_volume1: Optional[int]      # 申买量
-    ask_price2: Optional[float]
-    ask_volume2: Optional[int]
-    bid_price2: Optional[float]
-    bid_volume2: Optional[int]
-    ask_price3: Optional[float]
-    ask_volume3: Optional[int]
-    bid_price3: Optional[float]
-    bid_volume3: Optional[int]
-    ask_price4: Optional[float]
-    ask_volume4: Optional[int]
-    bid_price4: Optional[float]
-    bid_volume4: Optional[int]
-    ask_price5: Optional[float]
-    ask_volume5: Optional[int]
-    bid_price5: Optional[float]
-    bid_volume5: Optional[int]
-    ask_price6: Optional[float]
-    ask_volume6: Optional[int]
-    bid_price6: Optional[float]
-    bid_volume6: Optional[int]
-    ask_price7: Optional[float]
-    ask_volume7: Optional[int]
-    bid_price7: Optional[float]
-    bid_volume7: Optional[int]
-    ask_price8: Optional[float]
-    ask_volume8: Optional[int]
-    bid_price8: Optional[float]
-    bid_volume8: Optional[int]
-    ask_price9: Optional[float]
-    ask_volume9: Optional[int]
-    bid_price9: Optional[float]
-    bid_volume9: Optional[int]
-    ask_price10: Optional[float]
-    ask_volume10: Optional[int]
-    bid_price10: Optional[float]
-    bid_volume10: Optional[int]
-
-
-class TsTickIdx(BaseModel):
-    ts_code: str
-    name: Optional[str]
-    trade_time: Optional[datetime]
-    last_price: Optional[float]                 # last_price    单位元
-    pre_close_price: Optional[float]            # pre_close_price
-    high_price: Optional[float]                 # high_price
-    open_price: Optional[float]                 # open_price
-    low_price: Optional[float]                  # low_price
-    close_price: Optional[float]                # close_price
-    volume: Optional[int]                       # volume, 成交总量
-    amount: Optional[float]                     # amount, 成交总金额
-
-
-class TsTickOpt(BaseModel):
-    ts_code: str
-    instrument_id: str
-    trade_time: Optional[datetime]
-    pre_price: Optional[float]                  # 单位元
-    price: Optional[float]
-    open: Optional[float]
-    high: Optional[float]
-    low: Optional[float]
-    close: Optional[float]
-    open_int: Optional[int]
-    vol: Optional[float]
-    amount: Optional[float]
-    num: Optional[int]
-    ask_price1: Optional[float]
-    ask_volume1: Optional[int]
-    bid_price1: Optional[float]
-    bid_volume1: Optional[int]
-    pre_delta: Optional[float]              # 昨虚实度，暂未提供
-    dif_price1: Optional[float]
-    dif_price2: Optional[float]
-    high_limit_price: Optional[float]
-    low_limit_price: Optional[float]
-    refer_price: Optional[float]            # 参考价，港股使用
-
-
-class TsTickFuture(BaseModel):
-    ts_code: str
-    instrument_id: Optional[str]
-    trade_time: Optional[datetime]
-    pre_price: Optional[float]
-    price: Optional[float]
-    open: Optional[float]
-    high: Optional[float]
-    low: Optional[float]
-    close: Optional[float]
-    open_int: Optional[int]
-    vol: Optional[int]
-    amount: Optional[int]                  # 单数量
-    num: Optional[int]
-    ask_price1: Optional[float]
-    ask_volume1: Optional[int]
-    bid_price1: Optional[float]
-    bid_volume1: Optional[int]
-    pre_delta: Optional[float]
-    curr_delta: Optional[float]
-    dif_price1: Optional[float]
-    dif_price2: Optional[float]
-    high_limit_price: Optional[float]
-    low_limit_price: Optional[float]
-    refer_price: Optional[float]
-    pre_settle_price: Optional[float]
-    settle_price: Optional[float]
+from datetime import datetime
+from typing import Optional
+
+from pydantic import BaseModel
+
+
+class TsTick(BaseModel):
+    ts_code: str
+    name: Optional[str]
+    trade_time: Optional[datetime]
+    pre_close_price: Optional[float]
+    last_price: Optional[float]
+    open_price: Optional[float]
+    high_price: Optional[float]
+    low_price: Optional[float]
+    close_price: Optional[float]
+    volume: Optional[int]           # 成交量
+    amount: Optional[int]           # 成交金额
+    count: Optional[int]            # 交易数量
+    ask_price1: Optional[float]     # 申卖价, 委托档位卖1档价格
+    ask_volume1: Optional[int]      # 申卖量
+    bid_price1: Optional[float]     # 申买价, 委托档位买1档价格
+    bid_volume1: Optional[int]      # 申买量
+    ask_price2: Optional[float]
+    ask_volume2: Optional[int]
+    bid_price2: Optional[float]
+    bid_volume2: Optional[int]
+    ask_price3: Optional[float]
+    ask_volume3: Optional[int]
+    bid_price3: Optional[float]
+    bid_volume3: Optional[int]
+    ask_price4: Optional[float]
+    ask_volume4: Optional[int]
+    bid_price4: Optional[float]
+    bid_volume4: Optional[int]
+    ask_price5: Optional[float]
+    ask_volume5: Optional[int]
+    bid_price5: Optional[float]
+    bid_volume5: Optional[int]
+    ask_price6: Optional[float]
+    ask_volume6: Optional[int]
+    bid_price6: Optional[float]
+    bid_volume6: Optional[int]
+    ask_price7: Optional[float]
+    ask_volume7: Optional[int]
+    bid_price7: Optional[float]
+    bid_volume7: Optional[int]
+    ask_price8: Optional[float]
+    ask_volume8: Optional[int]
+    bid_price8: Optional[float]
+    bid_volume8: Optional[int]
+    ask_price9: Optional[float]
+    ask_volume9: Optional[int]
+    bid_price9: Optional[float]
+    bid_volume9: Optional[int]
+    ask_price10: Optional[float]
+    ask_volume10: Optional[int]
+    bid_price10: Optional[float]
+    bid_volume10: Optional[int]
+
+
+class TsTickIdx(BaseModel):
+    ts_code: str
+    name: Optional[str]
+    trade_time: Optional[datetime]
+    last_price: Optional[float]                 # last_price
+    pre_close_price: Optional[float]            # pre_close_price
+    high_price: Optional[float]                 # high_price
+    open_price: Optional[float]                 # open_price
+    low_price: Optional[float]                  # low_price
+    close_price: Optional[float]                # close_price
+    volume: Optional[int]                       # volume, 成交总量
+    amount: Optional[float]                     # amount, 成交总金额
+
+
+class TsTickOpt(BaseModel):
+    ts_code: str
+    instrument_id: str
+    trade_time: Optional[datetime]
+    pre_price: Optional[float]
+    price: Optional[float]
+    open: Optional[float]
+    high: Optional[float]
+    low: Optional[float]
+    close: Optional[float]
+    open_int: Optional[int]
+    vol: Optional[float]
+    amount: Optional[float]
+    num: Optional[int]
+    ask_price1: Optional[float]
+    ask_volume1: Optional[int]
+    bid_price1: Optional[float]
+    bid_volume1: Optional[int]
+    pre_delta: Optional[float]              # 昨虚实度，暂未提供
+    dif_price1: Optional[float]
+    dif_price2: Optional[float]
+    high_limit_price: Optional[float]
+    low_limit_price: Optional[float]
+    refer_price: Optional[float]            # 参考价，港股使用
+
+
+class TsTickFuture(BaseModel):
+    ts_code: str
+    instrument_id: Optional[str]
+    trade_time: Optional[datetime]
+    pre_price: Optional[float]
+    price: Optional[float]
+    open: Optional[float]
+    high: Optional[float]
+    low: Optional[float]
+    close: Optional[float]
+    open_int: Optional[int]
+    vol: Optional[int]
+    amount: Optional[int]                  # 单数量
+    num: Optional[int]
+    ask_price1: Optional[float]
+    ask_volume1: Optional[int]
+    bid_price1: Optional[float]
+    bid_volume1: Optional[int]
+    pre_delta: Optional[float]
+    curr_delta: Optional[float]
+    dif_price1: Optional[float]
+    dif_price2: Optional[float]
+    high_limit_price: Optional[float]
+    low_limit_price: Optional[float]
+    refer_price: Optional[float]
+    pre_settle_price: Optional[float]
+    settle_price: Optional[float]

tushare/subs/tgw_subs/convert.py

@@ -62,14 +62,14 @@ def convert_tick_index(tgw_data: dict) -> TsTickIdx:
         ts_code=get_ts_code(tgw_data.get("market_type"), tgw_data.get("security_code")),
         name=None,
         trade_time=datetime.strptime(str(tgw_data.get('orig_time'))[:-3], '%Y%m%d%H%M%S'),
-        last_price=tgw_data.get('last_index')/1000000,
-        pre_close_price=tgw_data.get('pre_close_index')/1000000,
-        open_price=tgw_data.get('open_index')/1000000,
-        high_price=tgw_data.get('high_index')/1000000,
-        low_price=tgw_data.get('low_index')/1000000,
-        close_price=tgw_data.get('close_index')/1000000,
-        volume=tgw_data.get('total_volume_trade')/100,
-        amount=tgw_data.get('total_value_trade')/100000,
+        last_price=tgw_data.get('last_index'),
+        pre_close_price=tgw_data.get('pre_close_index'),
+        open_price=tgw_data.get('open_index'),
+        high_price=tgw_data.get('high_index'),
+        low_price=tgw_data.get('low_index'),
+        close_price=tgw_data.get('close_index'),
+        volume=tgw_data.get('total_volume_trade'),
+        amount=tgw_data.get('total_value_trade'),
     )
     return item
 
@@ -138,22 +138,22 @@ def convert_tick_stock(tgw_data: dict) -> TsTick:
     """
     extra = {}
     for i in range(1, 11):
-        extra[f'ask_price{i}'] = tgw_data.get(f'offer_price{i}')/1000000
-        extra[f'ask_volume{i}'] = tgw_data.get(f'offer_volume{i}')/100
-        extra[f'bid_price{i}'] = tgw_data.get(f'bid_price{i}')/1000000
-        extra[f'bid_volume{i}'] = tgw_data.get(f'bid_volume{i}')/100
+        extra[f'ask_price{i}'] = tgw_data.get(f'offer_price{i}')
+        extra[f'ask_volume{i}'] = tgw_data.get(f'offer_volume{i}')
+        extra[f'bid_price{i}'] = tgw_data.get(f'bid_price{i}')
+        extra[f'bid_volume{i}'] = tgw_data.get(f'bid_volume{i}')
     item = TsTick(
         ts_code=get_ts_code(tgw_data.get("market_type"), tgw_data.get("security_code")),
         name=None,
         trade_time=datetime.strptime(str(tgw_data.get('orig_time'))[:-3], '%Y%m%d%H%M%S'),
-        pre_close_price=tgw_data.get("pre_close_price")/1000000,
-        last_price=tgw_data.get("last_price")/1000000,
-        open_price=tgw_data.get("open_price")/1000000,
-        high_price=tgw_data.get("high_price")/1000000,
-        low_price=tgw_data.get("low_price")/1000000,
-        close_price=tgw_data.get("close_price")/1000000,
-        volume=tgw_data.get("total_volume_trade")/100,
-        amount=tgw_data.get("total_value_trade")/1000000,
+        pre_close_price=tgw_data.get("pre_close_price"),
+        last_price=tgw_data.get("last_price"),
+        open_price=tgw_data.get("open_price"),
+        high_price=tgw_data.get("high_price"),
+        low_price=tgw_data.get("low_price"),
+        close_price=tgw_data.get("close_price"),
+        volume=tgw_data.get("total_volume_trade"),
+        amount=tgw_data.get("total_value_trade"),
         count=tgw_data.get("num_trades"),
         **extra
     )

tushare/subs/ts_subs.py

@@ -0,0 +1,192 @@
+# -*- coding:utf-8 -*- 
+'''
+@group:waditu
+@author: DY
+'''
+import _thread as thread
+import fnmatch
+import json
+
+import logging, sys
+import re
+import ssl
+import threading
+import time
+from collections import defaultdict
+from functools import wraps
+from multiprocessing.context import Process
+
+import websocket
+from websocket import WebSocketConnectionClosedException
+
+logging.basicConfig(stream=sys.stdout, level=logging.ERROR, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+class Subs(object):
+
+    def __init__(self, token='', callback_mode='multi-thread', debug=False):
+        self.url = 'wss://ws.tushare.pro/listening'
+        self.token = token
+        self.debug = debug
+
+        self.callback_mode = callback_mode
+        self.topics = defaultdict(lambda: list())
+        self.callback_funcs = defaultdict(lambda: list())
+
+        self.websocket = None
+
+    def threading_keepalive_ping(self):
+        def ping():
+            time.sleep(30)
+            req_data = {
+                "action": "ping"
+            }
+            self.websocket.send(json.dumps(req_data))
+            logger.debug('send ping message')
+        threading.Thread(target=ping).start()
+
+    def on_open(self, *args, **kwargs):
+        req_data = {
+            "action": "listening",
+            "token": self.token,
+            "data": self.topics
+        }
+        self.websocket.send(json.dumps(req_data))
+        logger.info('application starting...')
+        self.threading_keepalive_ping()
+
+    def on_message(self, message, *args, **kwargs):
+        logger.debug(message)
+        if isinstance(message, (str, bytes, bytearray)):
+            resp_data = json.loads(message)
+            if not resp_data.get('status'):
+                logger.error(resp_data.get('message'))
+                return
+        else:
+            logger.info(message)
+            return
+        data = resp_data.get('data', {})
+        if not data or not isinstance(data, dict):
+            return
+
+        topic = data.get('topic')
+        code = data.get('code')
+        record = data.get('record')
+        if not topic or not code or not record:
+            logger.warning('get invalid response-data(%s)' % resp_data)
+            return
+
+        self._do_callback_function(topic, code, record)
+
+    def on_error(self, error, *args, **kwargs):
+        if self.debug:
+            logging.error(str(error), exc_info=True)
+
+    def on_close(self, *args, **kwargs):
+        logger.error('close')
+        _type, _value, _traceback = sys.exc_info()
+        if _type in [WebSocketConnectionClosedException, ConnectionRefusedError]:
+            time.sleep(1)
+            self.run()
+
+    def run(self):
+        if not self.topics:
+            logger.error('no data.')
+            return
+
+        self.websocket = websocket.WebSocketApp(
+            self.url,
+            on_message=self.on_message,
+            on_error=self.on_error,
+            on_close=self.on_close,
+            on_open=self.on_open
+        )
+        if self.url.startswith('wss:'):
+            self.websocket.run_forever(sslopt={"cert_reqs": ssl.CERT_NONE})
+        else:
+            self.websocket.run_forever()
+
+    def register(self, topic, codes):
+        codes = set(codes)
+
+        def decorator(func):
+            func.codes = set()
+            func.pcodes = set()
+            for code in codes:
+                if not re.match(r'[\d\w\.\*]+', code):
+                    logger.error('error code')
+                    exit(1)
+
+                if '*' in code:
+                    for code1 in func.pcodes:
+                        if fnmatch.fnmatch(code1, code) or fnmatch.fnmatch(code, code1):
+                            logger.error('duplicate code')
+                            exit(1)
+                    for code1 in func.codes:
+                        if fnmatch.fnmatch(code1, code):
+                            logger.error('duplicate code')
+                            exit(1)
+                    func.pcodes.add(code)
+                else:
+                    for code1 in func.pcodes:
+                        if fnmatch.fnmatch(code, code1):
+                            logger.error('duplicate code')
+                            exit(1)
+                    func.codes.add(code)
+
+            self.topics[topic] += codes
+            self.callback_funcs[topic].append(func)
+
+            @wraps(func)
+            def inner(*args, **kwargs):
+                """ should receive a message-value parameter """
+                return func(*args, **kwargs)
+
+            return inner
+
+        return decorator
+
+    def _do_callback_function(self, topic, code, value):
+        for func in self.callback_funcs[topic]:
+            checked = False
+            if code in func.codes:
+                checked = True
+            else:
+                for pcode in func.pcodes:
+                    if fnmatch.fnmatch(code, pcode):
+                        checked = True
+                        break
+
+            if not checked:
+                continue
+
+            if self.callback_mode == 'single-thread':
+                func(value)
+            elif self.callback_mode == 'multi-thread':
+                thread.start_new_thread(func, (value,))
+            elif self.callback_mode == 'multi-process':
+                p = Process(target=func, args=(value,))
+                p.start()
+
+
+def test():
+    app = Subs(token='')
+
+    #  code 可以包含 * （通配符）
+    @app.register(topic='HQ_STK_TICK', codes=["000001.SZ"])
+    def print_message(record):
+        """
+        订阅主题topic，并指定codes列表，在接收到topic的推送消息时，符合code条件，就会执行回调
+        :param record:
+        :return:
+        """
+        print('用户定义业务代码输出 print_message(%s)' % str(record))
+
+    app.run()
+
+
+if __name__ == '__main__':
+    test()