gjdatac-api 0.1.7__tar.gz → 0.1.9__tar.gz

{gjdatac_api-0.1.7 → gjdatac_api-0.1.9}/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.1
 Name: gjdatac_api
-Version: 0.1.7
+Version: 0.1.9
 Summary: 数据接口打包
 Author: yin
 Author-email: 2018209921@qq.com
 Requires-Python: >3.6
 Description-Content-Type: text/markdown
 
-### 新增主力合约接口
+### 合并接口、修改日期入参格式

{gjdatac_api-0.1.7 → gjdatac_api-0.1.9}/gjdatac_api/data_api_class.py

@@ -15,7 +15,8 @@ from utils.dict_data import (get_price_frequency_dict, get_price_data_type_dict,
                              future_margin_ratio_return_fields, future_margin_ratio_return_fields_rename,
                              future_limit_position_return_fields, future_limit_position_return_fields_rename,
                              main_contract_return_fields, main_contract_return_fields_rename,
-                             tick_main_contract_return_fields, tick_main_contract_return_fields_rename
+                             tick_main_contract_return_fields, tick_main_contract_return_fields_rename,
+                             gtja_mink_main_contract_return_fields, gtja_mink_main_contract_return_fields_rename
                              )
 from utils.dolphindb_info import *
 from utils.instrument_class import Instrument
@@ -44,7 +45,6 @@ class GJDataC:
         # 创建向量，包含所有可用节点的地址和端口
         self.sites = dolphindb_service
 
-
     def connect_db(self, table_name, db_path):
         """
         此方法用于根据不同的数据表及数据库获取数据，只返回Table对象，对应的数据处理在对应的数据接口处理
@@ -60,7 +60,7 @@ class GJDataC:
 
         # 创建连接；开启高可用，并指定sites为所有可用节点的ip:port
         success = db_session.connect(self.host, self.port, self.user_id, self.password, highAvailability=True,
-                  highAvailabilitySites=self.sites)
+                                     highAvailabilitySites=self.sites)
         if not success:
             raise Exception("please check your database connection information")
 
@@ -1767,7 +1767,7 @@ class GJDataC:
         return {
             '1d': (qc_future_maincontract_dayk, dayk_future_maincontract_db),
             '1min': (qc_future_maincontract_minutek, minutek_future_maincontract_db),
-            '1min_gtja': (qc_future_maincontract_minutek, minutek_future_maincontract_db)
+            '1min_gtja': (qc_future_maincontract_gtja_minutek, minutek_future_maincontract_gtja_db)
         }.get(frequency, (None, None))
 
     @staticmethod
@@ -1938,9 +1938,12 @@ class GJDataC:
         """
         new_df = data.copy()
 
-        if frequency in ['1min', '1d', '1min_gtja']:
+        if frequency in ['1min', '1d']:
             return_fields = main_contract_return_fields
             return_fields_rename = main_contract_return_fields_rename
+        elif frequency in ['1min_gtja']:
+            return_fields = gtja_mink_main_contract_return_fields
+            return_fields_rename = gtja_mink_main_contract_return_fields_rename
         else:
             return_fields = tick_main_contract_return_fields
             return_fields_rename = tick_main_contract_return_fields_rename
@@ -1951,10 +1954,12 @@ class GJDataC:
         # 将实际数据字段转换成需求文档中指定的字段
         new_df.rename(columns=return_fields_rename, inplace=True)
 
-        if frequency in ['1min', '1d', '1min_gtja']:
+        if frequency in ['1min', '1d']:
             new_df['none_close'] = new_df['close']
             new_df['none_prev_close'] = new_df['prev_close']
             new_df['ex_factor'] = ex_factor
+        elif frequency in ['1min_gtja']:
+            new_df['ex_factor'] = ex_factor
         else:
             new_df['total_turnover'] = new_df.groupby(['order_book_ids', 'trading_day'])['turnover'].transform('sum')
 
@@ -1971,7 +1976,7 @@ class GJDataC:
         :return: 计算后的数据
         """
 
-        if not adjust_type or frequency in ['tick_l1', 'tick_l2']:
+        if not adjust_type or frequency in ['tick_l1', 'tick_l2', '1min_gtja']:
             return data
 
         adjust_fields = ['open', 'close', 'high', 'low', 'limit_up', 'limit_down', 'settlement', 'prev_settlement',
@@ -2911,20 +2916,26 @@ class GJDataC:
         except Exception as e:
             raise e
 
-    """ get_history_bar 查询合约指定日期前的行情数据（2024-12-24提出） """
+    """ 
+    get_history_bars 此接口和下面的接口合并为一个，根据传入的参数进行判断，调用不同的方法
+    查询合约指定日期前的行情数据（2024-12-24提出） 
+    """
 
-    def get_history_bar(self, datetime_str=None, instrument_id=None, bar_count=None,
-                        frequency=None, fields=None):
+    def get_history_bars(self, datetime_str=None, instrument_id=None, bar_count=None,
+                        frequency=None, fields=None, start_date=None, end_date=None):
         """
+        （此接口根据接收的参数调用不同的功能函数）
         查询合约指定日期时间前的行情数据
         需要程序自行判断合约代码是期货还是期权（根据期货/期权合约信息表中的 mk_instrument_id 和 instrument_id 进行筛选）
         mk_instrument_id: 米筐合约代码， instrument_id: CTP合约代码
         Args:
-            datetime_str: str, 查询的日期时间，格式为'YYYY-MM-DD HH:MM:SS'
+            datetime_str: str, 查询的日期时间，格式为'YYYY.MM.DD'
             instrument_id: str or List[str], 合约代码，可以是单个合约或合约列表
             bar_count: int, 获取的历史K线数量
             frequency: str, 历史行情频率，可选值：'1d'或'1m'
             fields: str, 返回的数据字段，如'close_price'表示收盘价
+            start_date: str, 开始日期，格式为'YYYY.MM.DD' （此接口根据接收的参数调用不同的功能函数）
+            end_date: str, 结束日期，格式为'YYYY.MM.DD' （此接口根据接收的参数调用不同的功能函数）
         Returns:
             如果instrument_id是字符串，返回一个浮点数列表 [619.70, 620.20, 618.40, 622.00, 610.10]
             如果instrument_id是列表，返回一个浮点数列表的列表 [[619.70, 620.20], [618.40, 622.00]]
@@ -2932,21 +2943,25 @@ class GJDataC:
             ValueError: 参数验证失败时抛出
         """
 
-        # 参数验证
-        self._get_history_bar_validate_params(datetime_str, instrument_id, bar_count, frequency, fields)
+        ''' 根据参数的不同，调用不同的函数 '''
+        if datetime_str and bar_count:
+            # 参数验证
+            self._get_history_bar_validate_params(datetime_str, instrument_id, bar_count, frequency, fields)
 
-        # 转换日期格式
-        query_time = self._get_history_bar_parse_datetime(datetime_str)
+            # 转换日期格式
+            # query_time = self._get_history_bar_parse_datetime(datetime_str)
 
-        # 执行查询并筛选数据
-        result, bar_db_session = self._get_history_bar_execute_query(frequency, instrument_id, query_time,
-                                                                     bar_count)
-        # 此处在所有的查询筛选都在数据库层面进行完后，再关闭连接
-        # 如果将dolphindb的Table对象转换成DataFrame时，如果数据量较大，那么需要耗费的时间较长，所以尽可能将所有的筛选在数据库层面完成
-        bar_db_session.close()
+            # 执行查询并筛选数据
+            result, bar_db_session = self._get_history_bar_execute_query(frequency, instrument_id, datetime_str,
+                                                                         bar_count)
+            # 此处在所有的查询筛选都在数据库层面进行完后，再关闭连接
+            # 如果将dolphindb的Table对象转换成DataFrame时，如果数据量较大，那么需要耗费的时间较长，所以尽可能将所有的筛选在数据库层面完成
+            bar_db_session.close()
 
-        # 处理结果
-        return self._get_history_bar_process_result(result, instrument_id, fields, bar_count)
+            # 处理结果
+            return self._get_history_bar_process_result(result, instrument_id, fields, bar_count)
+        else:
+            return self.get_history_bar(instrument_id, start_date, end_date, frequency, fields)
 
     def _get_history_bar_validate_params(self, datetime_str, instrument_id, bar_count,
                                          frequency, fields):
@@ -2964,8 +2979,8 @@ class GJDataC:
         # 验证时间格式
         if not datetime_str or not isinstance(datetime_str, str):
             raise ValueError("datetime_str must be a non-empty string")
-        if not self._get_history_bar_is_convertible_datetime(datetime_str):
-            raise ValueError("datetime_str must be in the format 'YYYY-MM-DD HH:MM:SS'")
+        # if not self._get_history_bar_is_convertible_datetime(datetime_str):
+        #     raise ValueError("datetime_str must be in the format 'YYYY-MM-DD HH:MM:SS'")
 
         # 验证合约代码
         self.general_validate_field_str_or_list(instrument_id, "instrument_id")
@@ -3061,6 +3076,7 @@ class GJDataC:
         """
         contract_type = self.filter_contract_type(instrument_id)
 
+
         if contract_type == "future":
             return {
                 "1d": (history_future_day_db_table_name, history_future_day_db_path),
@@ -3089,7 +3105,8 @@ class GJDataC:
         """
 
         # 将日期转换成 DolphinDB 识别的日期格式，用于筛选数据
-        dolphin_date = self._get_history_bar_datetime_to_dolphin_time(datetime_str)
+        # dolphin_date = self._get_history_bar_datetime_to_dolphin_time(datetime_str)
+        dolphin_date = datetime_str
 
         # 使用 DolphinDB 的 where 子句筛选数据(这里因为实际需求的原因需要对每个instrument_id进行筛选，所以需要使用 contextby)
         # 根据bar_count 分别对不同的instrument_id 进行筛选
@@ -3157,7 +3174,7 @@ class GJDataC:
 
     """ get_history_bars 查询行情指定范围日期的行情数据"""
 
-    def get_history_bars(self, instrument_id, start_date, end_date, frequency, fields=None):
+    def get_history_bar(self, instrument_id, start_date, end_date, frequency, fields=None):
         """
         Args:
             instrument_id: str or List[str], 合约代码，可以是单个合约或合约列表

{gjdatac_api-0.1.7 → gjdatac_api-0.1.9}/gjdatac_api/dolphindb_info.py

@@ -101,7 +101,21 @@ get_warehouse_stocks_future_db_table_name = "qc_warehouse_stocks"  # 表名（
 
 # 主力合约
 main_contract_db_path = "dfs://dayk"  # 数据库路径
-main_contract_db_table_name = "qc_future_maincontract"  # 表名
+main_contract_db_table_name = "qc_gtja_future_maincontract"  # 表名
+
+# 主/次力合约
+dayk_future_maincontract_db = "dfs://dayk_future_maincontract"
+minutek_future_maincontract_db = "dfs://minutek_future_maincontract"
+minutek_future_maincontract_gtja_db = "dfs://minutek"
+
+qc_future_maincontract_dayk = "qc_future_maincontract_dayk"
+qc_future_maincontract_minutek = "qc_future_maincontract_mink"
+qc_future_maincontract_gtja_minutek = "qc_gtja_future_mink"
+
+
+tick_main_contract_db_path = "dfs://tick"  # 数据库路径
+tick_main_contract_db_table_name = "qc_future_tick"  # 表名
+
 
 # 期货交割手续费(目前需求中所需的字段在交易参数表中都存在，所以直接使用交易参数表)
 delivery_fee_db_path = "dfs://basicinfo"  # 数据库路径
@@ -132,5 +146,6 @@ inventory_db_table_name = "qc_product_inventory"  # 表名
 subscribe_min_db_table_name = "QcMinKlineTable"  # 表名
 
 
-''' 仿真 '''
+
+''' 仿真服务器 '''
 dolphindb_service = ["10.74.56.101:8902", "10.74.56.102:8902", "10.74.56.103:8902"]

{gjdatac_api-0.1.7 → gjdatac_api-0.1.9}/gjdatac_api.egg-info/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.1
 Name: gjdatac-api
-Version: 0.1.7
+Version: 0.1.9
 Summary: 数据接口打包
 Author: yin
 Author-email: 2018209921@qq.com
 Requires-Python: >3.6
 Description-Content-Type: text/markdown
 
-### 新增主力合约接口
+### 合并接口、修改日期入参格式

{gjdatac_api-0.1.7 → gjdatac_api-0.1.9}/setup.py

@@ -5,7 +5,7 @@ with open('description_info.md', 'r', encoding='utf-8') as f:
 
 setup(
     name='gjdatac_api',  # 你的包名
-    version='0.1.7',  # 版本号
+    version='0.1.9',  # 版本号
     description='数据接口打包',  # 包的简要描述
     long_description=long_description,  # 包的详细描述
     long_description_content_type='text/markdown',  # 描述文件的类型