Merge pull request yutiansut#1072 from ZerounNet/master

为QACMD添加etf_list支持
zsluedem · Mar 31, 2019 · 758a45c · 758a45c
2 parents 2b0d7a4 + 5e03329
commit 758a45c
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/QUANTAXIS/QACmd/__init__.py b/QUANTAXIS/QACmd/__init__.py
@@ -47,6 +47,7 @@
     QA_SU_save_index_min,
     QA_SU_save_future_list,
     QA_SU_save_index_list,
+    QA_SU_save_etf_list,
     QA_SU_save_etf_day,
     QA_SU_save_etf_min,
     QA_SU_save_financialfiles,
@@ -242,6 +243,7 @@ def print_save_usage(self):
             命令格式：save option_commodity_day : 保存商品期权日线数据（不包括已经过期摘牌的数据） \n\
             命令格式：save option_commodity_min : 保存商品期权分钟线数据（不包括已经过期摘牌的数据） \n\
             命令格式: save index_list : 保存指数列表 \n\
+            命令格式: save etf_list : 保存etf列表 \n\
             命令格式: save future_list : 保存期货列表 \n\
             ----------------------------------------------------------\n\
             if you just want to save daily data just\n\
@@ -275,6 +277,7 @@ def do_save(self, arg):
                 # QA_SU_save_stock_min('tdx')
                 QA_SU_save_index_day('tdx')
                 # QA_SU_save_index_min('tdx')
+                QA_SU_save_etf_list('tdx')
                 # QA_SU_save_etf_day('tdx')
                 # QA_SU_save_etf_min('tdx')
                 QA_SU_save_index_list('tdx')
@@ -298,6 +301,7 @@ def do_save(self, arg):
                 # QA_SU_save_stock_min('tdx')
                 QA_SU_save_index_day('tdx')
                 # QA_SU_save_index_min('tdx')
+                QA_SU_save_etf_list('tdx')
                 QA_SU_save_etf_day('tdx')
                 # QA_SU_save_etf_min('tdx')
                 QA_SU_save_index_list('tdx')
@@ -320,6 +324,7 @@ def do_save(self, arg):
                 QA_SU_save_stock_min('tdx')
                 # QA_SU_save_index_day('tdx')
                 QA_SU_save_index_min('tdx')
+                QA_SU_save_etf_list('tdx')
                 # QA_SU_save_etf_day('tdx')
                 QA_SU_save_etf_min('tdx')
                 QA_SU_save_stock_list('tdx')
@@ -339,6 +344,7 @@ def do_save(self, arg):
                 QA_SU_save_stock_min('tdx')
                 QA_SU_save_index_day('tdx')
                 QA_SU_save_index_min('tdx')
+                QA_SU_save_etf_list('tdx')
                 QA_SU_save_etf_day('tdx')
                 QA_SU_save_etf_min('tdx')
                 QA_SU_save_stock_list('tdx')
@@ -397,6 +403,7 @@ def do_save(self, arg):
                         save stock_list :对应输入命令 save stock_list
                         save stock_block:对应输入命令 save stock_block
                         save stock_info :对应输入命令 save stock_info
+                        save etf_list :对应输入命令 save etf_list
                         save index_list :对应输入命令 save index_list
                         save future_list :对应输入命令 save future_list
                         save future_day  : 对应输入命令  save future_day

diff --git a/QUANTAXIS/QASU/main.py b/QUANTAXIS/QASU/main.py
@@ -92,6 +92,20 @@ def QA_SU_save_index_list(engine, client=DATABASE):
     engine.QA_SU_save_index_list(client=client)
 
 
+def QA_SU_save_etf_list(engine, client=DATABASE):
+    """save etf_list
+
+    Arguments:
+        engine {[type]} -- [description]
+
+    Keyword Arguments:
+        client {[type]} -- [description] (default: {DATABASE})
+    """
+
+    engine = select_save_engine(engine)
+    engine.QA_SU_save_etf_list(client=client)
+
+
 def QA_SU_save_future_list(engine, client=DATABASE):
     """save future_list