update index_stock

rmfish · Nov 27, 2023 · 1cc099a · 1cc099a
1 parent e8dff2d
commit 1cc099a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 7 deletions.
diff --git a/tutake/api/query_client.py b/tutake/api/query_client.py
@@ -1,11 +1,9 @@
 from functools import partial
 
-import tushare
 from pandas import DataFrame
 
 from tutake.api.ts.tushare_api import TushareAPI
 from tutake.api.xq.xueqiu_api import XueQiuAPI
-from tutake.remote.client import TutakeClient
 
 
 class TushareQuery(object):
@@ -22,6 +20,7 @@ def __init__(self, config):
     def _init_remote_client(self):
         remote_address = self.config.get_remote_address()
         if remote_address is not None:
+            from tutake.remote.client import TutakeClient
             self._remote_client = TutakeClient(remote_address)
 
     def query(self, api_name, fields='', **kwargs) -> DataFrame:
@@ -48,6 +47,7 @@ def apis(self):
 
     def pro_bar(self, ts_code='', start_date='', end_date='', freq='D', asset='E', exchange='', adj=None,
                 ma=[], factors=None, adjfactor=False, offset=None, limit=None, contract_type=''):
+        import tushare
         return tushare.pro_bar(ts_code, self, start_date, end_date, freq, asset, exchange, adj, ma, factors, adjfactor,
                                offset,
                                limit, contract_type)

diff --git a/tutake/api/ts/index_stock.py b/tutake/api/ts/index_stock.py
@@ -894,6 +894,5 @@ def default_order_by(self) -> str:
     config = TutakeConfig(project_root())
 
     api = IndexStock(config)
-    # print(api.fetch_and_append(index_code='000300.SH'))
-    # print(api.process())  # 同步增量数据
-    print(api.index_stock())
+    print(api.process())  # 同步增量数据
+    print(api.index_stock(con_code='T00018.SH'))
diff --git a/tutake/api/ts/stock_basic.py b/tutake/api/ts/stock_basic.py
@@ -246,5 +246,5 @@ def fetch_save(offset_val=0):
     print(pro.stock_basic())
 
     api = StockBasic(config)
-    print(api.process())    # 同步增量数据
-    print(api.stock_basic())    # 数据查询接口
+    # print(api.process())    # 同步增量数据
+    print(api.stock_basic(ts_code='T00018.SH'))    # 数据查询接口