添加sup库查询

Author: zhangyongsheng

CHANGELOG.md

@@ -1,3 +1,6 @@
+# 2019-06-05
+- 添加sup库
+
 # 2019-05-10
 - 更新python2环境下run_query不支持中文
 

jqdatasdk/__init__.py

@@ -24,7 +24,7 @@ def logout():
     JQDataClient.instance().logout()
 
 
-__version__ = "1.6.5"
+__version__ = "1.6.6"
 
 
 __all__ = ["auth", "auth_by_token", "logout", "alpha101", "alpha191", "technical_analysis", "__version__"]

jqdatasdk/table.py

@@ -15,6 +15,7 @@
     "opt",
     "ssymmetry",
     "bond",
+    "sup",
     "DBTable",
 ]
 
@@ -27,6 +28,7 @@ def __init__(self, db, disable_join=False):
         self.__table_names = []
         self.db_name = db
 
+    @assert_auth
     def __load_table_names(self):
         self.__table_names = JQDataClient.instance().get_table_orm(db=self.db_name)
         for name in self.__table_names:
@@ -86,3 +88,4 @@ def __getattribute__(self, key):
 opt = DBTable("opt")
 ssymmetry = DBTable("ssymmetry")
 bond = DBTable("bond")
+sup = DBTable("sup")

tests/test_table.py

@@ -35,3 +35,17 @@ def test_bond():
                                     '04001,"bond_form":"\\u8bb0\\u8d26\\u5f0f","list_date":1389225600000,"delist_Date'
                                     '":1420416000000,"interest_begin_date":1389139200000,"maturity_date":1420675200000,'
                                     '"interest_date":null,"last_cash_date":1420675200000,"cash_comment":null}')
+
+
+def test_sup():
+    df = sup.run_query(query(sup.STK_FINANCE_SUPPLEMENT).limit(10))
+    assert len(df) == 10
+    assert df.iloc[0].to_json() ==  (
+                                '{"id":1,"company_id":300000062,"company_name":"\\u5e7f\\u4e1c\\u5fb7\\u7f8e\\u7cbe'
+                                '\\u7ec6\\u5316\\u5de5\\u96c6\\u56e2\\u80a1\\u4efd\\u6709\\u9650\\u516c\\u53f8",'
+                                '"code":"002054.XSHE","a_code":"002054","b_code":null,"h_code":null,'
+                                '"pub_date":1151020800000,"start_date":1041379200000,"end_date":1072828800000,'
+                                '"report_date":1072828800000,"report_type":0,"source_id":321001,"source":"\\u62db'
+                                '\\u52df\\u8bf4\\u660e\\u4e66","bill_and_account_receivable":null,'
+                                '"bill_and_account_payable":null,"rd_expenses":null}'
+    )