gjdatac-api 0.1.4__tar.gz → 0.1.6__tar.gz

{gjdatac_api-0.1.4 → gjdatac_api-0.1.6}/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.1
 Name: gjdatac_api
-Version: 0.1.4
-Summary: vwap,twap接口数据，接入tick_l2数据，新增实时订阅接口
+Version: 0.1.6
+Summary: 数据接口打包
 Author: yin
 Author-email: 2018209921@qq.com
 Requires-Python: >3.6
 Description-Content-Type: text/markdown
 
-将原有接口全部更新，新增后续新需求
+### 新增主力合约接口

{gjdatac_api-0.1.4 → gjdatac_api-0.1.6}/gjdatac_api/data_api_class.py

@@ -2,6 +2,7 @@
 
 import re
 import datetime
+import operator
 import asyncio
 
 import holidays
@@ -10,10 +11,11 @@ import pandas as pd
 import dolphindb as ddb
 
 from .dict_data import (get_price_frequency_dict, get_price_data_type_dict, get_instruments_type_dict,
-                       future_delivery_fee_return_fields, future_delivery_fee_return_fields_rename,
-                       future_margin_ratio_return_fields, future_margin_ratio_return_fields_rename,
-                       future_limit_position_return_fields, future_limit_position_return_fields_rename)
+                             future_delivery_fee_return_fields, future_delivery_fee_return_fields_rename,
+                             future_margin_ratio_return_fields, future_margin_ratio_return_fields_rename,
+                             future_limit_position_return_fields, future_limit_position_return_fields_rename)
 from .dolphindb_info import *
+from .instrument_class import Instrument
 
 
 class GJDataC:
@@ -952,23 +954,37 @@ class GJDataC:
 
     """   get_trading_dates(交易日历)  """
 
-    def get_trading_dates(self, date=datetime.date.today(), n=None, start_date=None, end_date=None):
+    def get_trading_dates(self, date=datetime.date.today(), n=None, start_date=None, end_date=None,
+                          is_include_start=True, is_include_end=True,
+                          trade_time=None, date_count=None, date_type=None):
         """
+
         获取交易日历接口
         :param date: str--选填, 日期
         :param n: str--必填，根据不同值获取对应的交易日历
         :param start_date: str（datetime.date, datetime.datetime）--选填, 开始日期
         :param end_date: str（datetime.date, datetime.datetime）--选填, 结束日期
+        :param is_include_start 是否包含开始日期
+        :param is_include_end 是否包含结束日期
         若填写【date、n】为入参，则无法填写【start_date、end_date】，反之依然
         :return: data:交易日历
+
+        当入参为trade_time, date_count, date_type时，功能如下：
+        查询前后 N 个交易日或时间点
+        :param trade_time: 当前时间点，格式为 'YYYY.MM.DD hh:mm:ss'
+        :param date_count: 查询的交易日或时间数量，负数表示向前查询，正数表示向后查询
+        :param date_type: 数据类型，'1d' 表示日频率，'1m' 表示分钟频率
+        :return: 查询到的交易日或时间点列表
         """
 
         """ 校验参数 """
-        data = self.get_trading_dates_validate_date_n_start_end(date, n, start_date, end_date)
+        data = self.get_trading_dates_validate_date_n_start_end(date, n, start_date, end_date, is_include_start,
+                                                                is_include_end, trade_time, date_count, date_type)
 
         return data
 
-    def get_trading_dates_validate_date_n_start_end(self, date, n, start_date, end_date):
+    def get_trading_dates_validate_date_n_start_end(self, date, n, start_date, end_date, is_include_start,
+                                                    is_include_end, trade_time, date_count, date_type):
         """
         校验date、n和start_date、end_date有效性
 
@@ -976,6 +992,11 @@ class GJDataC:
         :param n: str--根据不同值获取对应的交易日历
         :param start_date: str（datetime.date, datetime.datetime）--开始日期
         :param end_date: str（datetime.date, datetime.datetime）--结束日期
+        :param is_include_start 是否包含开始日期
+        :param is_include_end 是否包含结束日期
+        :param trade_time: 当前时间点，格式为 'YYYY.MM.DD hh:mm:ss'
+        :param date_count: 查询的交易日或时间数量，负数表示向前查询，正数表示向后查询
+        :param date_type: 数据类型，'1d' 表示日频率，'1m' 表示分钟频率
         :return: 如果校验通过，就返回对应的数据，否则抛出异常
         """
 
@@ -986,7 +1007,9 @@ class GJDataC:
         elif date and n:  # 根据date和n获取交易日历
             return self.get_trading_dates_by_date_n(date, n)
         elif start_date and end_date:  # 根据start_date和end_date获取交易日历
-            return self.get_trading_dates_by_start_end(start_date, end_date)
+            return self.get_trading_dates_by_start_end(start_date, end_date, is_include_start, is_include_end)
+        elif trade_time and date_count and date_type:  # 查询前/后 N个交易日/时间数据
+            return self.get_trading_date(trade_time, date_count, date_type)
         else:
             raise ValueError("parameter error: please pass date、n or start_date、end_date")
 
@@ -1010,17 +1033,23 @@ class GJDataC:
 
         return self.get_dates_in_n(date, n)
 
-    def get_trading_dates_by_start_end(self, start_date, end_date):
+    def get_trading_dates_by_start_end(self, start_date, end_date, is_include_start, is_include_end):
         """
         根据start_date和end_date获取交易日历
         :param start_date: str（datetime.date, datetime.datetime）--选填, 开始日期
         :param end_date: str（datetime.date, datetime.datetime）--选填, 结束日期
+        :param is_include_start 是否包含开始日期
+        :param is_include_end 是否包含结束日期
 
         :return: data:根据start_date和end_date筛选后的数据
         """
 
         trading_params_data = self.get_trading_dates_get_data()
 
+        # 动态生成比较运算符
+        start_operator = operator.ge if is_include_start else operator.gt
+        end_operator = operator.le if is_include_end else operator.lt
+
         # 校验日期类型
         if self.check_date_type(start_date, end_date):
             # 如果类型校验通过，则将start_date和end_date处理成对应的时间数据类型
@@ -1029,14 +1058,22 @@ class GJDataC:
             if isinstance(end_date, str):
                 end_date = self._get_trading_dates_by_start_end(end_date)
             # 筛选 trading_day_str 列在 start_date 和 end_date 之间的数据
+            # filtered_df = trading_params_data.loc[
+            #     (trading_params_data['trading_day_str'] >= start_date) & (
+            #             trading_params_data['trading_day_str'] <= end_date)]
+
+            # 过滤数据
             filtered_df = trading_params_data.loc[
-                (trading_params_data['trading_day_str'] >= start_date) & (
-                        trading_params_data['trading_day_str'] <= end_date)]
+                start_operator(trading_params_data['trading_day_str'], start_date) &
+                end_operator(trading_params_data['trading_day_str'], end_date)
+                ]
 
             # 筛选 trade_flag 列等于 T 的数据（是交易日的数据）
             final_df = filtered_df.loc[filtered_df['trade_flag'] == 'T']
-            data = sorted(final_df["trading_day_str"].tolist())
-            return data
+            date_list = sorted(final_df["trading_day_str"].tolist())
+            formatted_date_list = [d.strftime('%Y.%m.%d') for d in date_list]
+
+            return formatted_date_list
         else:
             raise ValueError(
                 "start_date or end_date type error, please input str, datetime.date or datetime.datetime type")
@@ -1049,6 +1086,10 @@ class GJDataC:
         :param time_data: 时间数据
         :return: 转换后的时间数据类型
         """
+
+        if isinstance(time_data, str):
+            time_data = time_data.replace('.', '-')
+
         try:
             time_data = datetime.datetime.strptime(time_data, "%Y-%m-%d").date()
             return time_data
@@ -1068,6 +1109,12 @@ class GJDataC:
 
         :return: 如果类型正确，返回 True。否则返回 False，并打印错误信息
         """
+
+        if isinstance(_start_date, str):
+            _start_date = _start_date.replace('.', '-')
+        if isinstance(_start_date, str):
+            _end_date = _end_date.replace('.', '-')
+
         if isinstance(_start_date, (str, datetime.date, datetime.datetime)) and \
                 isinstance(_end_date, (str, datetime.date, datetime.datetime)):
             if isinstance(_start_date, str) and not self.general_validate_date(_start_date):
@@ -3152,7 +3199,7 @@ class GJDataC:
         db_session.close()
 
         ''' 处理数据 '''
-        return self.get_instrument_handle_data(data)
+        return self.get_instrument_handle_data(data, instrument_id)
 
     def get_instrument_validate_params(self, instrument_id):
         """
@@ -3227,10 +3274,11 @@ class GJDataC:
 
         return query_data
 
-    def get_instrument_handle_data(self, data):
+    def get_instrument_handle_data(self, data, instrument_id):
         """
         处理数据
         :param data: 期货/期权数据
+        :param instrument_id: 合约代码
         :return: 处理后的最终数据
         """
 
@@ -3249,7 +3297,7 @@ class GJDataC:
         result_fields = ["instrument_id", "exchange_id", "instrument_name", "product_id", "product_type",
                          "create_date", "open_date", "expire_date", "basis_price", "volume_multiple",
                          "underlying_instrument_id", "underlying_multiple", "option_type", "strike_price",
-                         "price_tick", "最小下单数量(暂无此字段)"]
+                         "price_tick", "limit_price_min_vol"]
 
         result = pd.DataFrame()
 
@@ -3265,9 +3313,46 @@ class GJDataC:
                 # 如果字段在两个数据源中都不存在，填充为空值
                 result[field] = None  # 或者使用 pd.NA、空字符串等
 
+        # 将所有最小下单量为空的填充为0
+        result['limit_price_min_vol'] = result['limit_price_min_vol'].astype(float).fillna(0)
+
+        # 将dataframe处理成 class 对象
+        result = self.get_instrument_convert_to_instrument(result, instrument_id)
+
         return result
 
+    @staticmethod
+    def get_instrument_convert_to_instrument(df, instrument_id):
+        """
+        将 DataFrame 转换成 Instrument 对象或对象列表
+        :param 用户传入的合约代码
+        :param 需要转化的结果 dataframe
+        """
+
+        if isinstance(instrument_id, list) and len(instrument_id) > 1:
+            # 如果 instrument_id 是列表且长度大于1
+            matching_records = df[df['instrument_id'].isin(instrument_id)]
+            if matching_records.empty:
+                return None  # 没有匹配结果
+            else:
+                # 去重后返回结果
+                unique_records = matching_records.drop_duplicates(subset=['instrument_id'])
+                return [Instrument(**row.to_dict()) for _, row in unique_records.iterrows()]
+        else:
+            # 如果 instrument_id 是单个字符串
+            matching_records = df[df['instrument_id'] == instrument_id] if isinstance(instrument_id, str) else df[
+                df['instrument_id'].isin(instrument_id)]
+            unique_records = matching_records.drop_duplicates(subset=['instrument_id'])
+            if matching_records.empty:
+                return None  # 没有匹配结果
+            elif len(unique_records) == 1:
+                return Instrument(**matching_records.iloc[0].to_dict())
+            else:
+                # 返回匹配结果
+                return [Instrument(**row.to_dict()) for _, row in unique_records.iterrows()]
+
     ''' 查询交易日列表 '''
+
     def get_trading__dates(self, start_date, end_date, is_include_start=True, is_include_end=True):
         """
         查询交易日列表
@@ -3324,6 +3409,7 @@ class GJDataC:
         return sorted(result['trading_day'].dt.strftime('%Y.%m.%d').tolist())
 
     """ 查询前/后 N个交易日/时间 """
+
     def get_trading_date(self, trade_time, date_count, date_type):
         """
         查询前后 N 个交易日或时间点
@@ -3391,7 +3477,8 @@ class GJDataC:
             result = filtered_dates['trading_day'].sort_values().head(date_count)
 
         # 将结果转换为指定格式的字符串列表
-        return sorted(result.dt.strftime('%Y.%m.%d').tolist())
+        fina_result = sorted(result.dt.strftime('%Y.%m.%d').tolist())
+        return fina_result[0] if len(fina_result) <= 1 else fina_result
 
     @staticmethod
     def get_trading_minutes(trading_dates, trade_time, date_count):
@@ -3431,10 +3518,10 @@ class GJDataC:
             # 向后查询
             result = [p for p in valid_time_points if pd.to_datetime(p) > trade_time][:date_count]
 
-        return sorted(result)
-
+        return sorted(result) if len(result) > 1 else result[0]
 
     """ 交易日判断 """
+
     def is_trading_day(self, date):
         """
         判断给定日期是否为交易日
@@ -3452,6 +3539,7 @@ class GJDataC:
         return False if filtered_data.empty else True
 
     """ 节假日判断 """
+
     @staticmethod
     def is_holiday(date):
         """
@@ -3465,8 +3553,8 @@ class GJDataC:
         # 判断是否是节假日
         return calendar.is_holiday(date)
 
-
     """ 周末判断 """
+
     @staticmethod
     def is_weekend(date):
         """

gjdatac_api-0.1.6/gjdatac_api/instrument_class.py

@@ -0,0 +1,34 @@
+class Instrument:
+    def __init__(self, instrument_id, exchange_id, instrument_name, product_id,
+       product_type, create_date, open_date, expire_date,
+       basis_price, volume_multiple, underlying_instrument_id,
+       underlying_multiple, option_type, strike_price, price_tick,
+       limit_price_min_vol):
+        self.instrument_id = instrument_id
+        self.exchange_id = exchange_id
+        self.instrument_name = instrument_name
+        self.product_id = product_id
+        self.product_type = product_type
+        self.create_date = create_date
+        self.open_date = open_date
+        self.expire_date = expire_date
+        self.basis_price = basis_price
+        self.volume_multiple = volume_multiple
+        self.underlying_instrument_id = underlying_instrument_id
+        self.underlying_multiple = underlying_multiple
+        self.option_type = option_type
+        self.strike_price = strike_price
+        self.price_tick = price_tick
+        self.limit_price_min_vol = limit_price_min_vol
+
+    def __repr__(self):
+        return (
+            f"Instrument(instrument_id={self.instrument_id}, exchange_id={self.exchange_id}, "
+            f"instrument_name={self.instrument_name}, product_id={self.product_id}, "
+            f"product_type={self.product_type}, create_date={self.create_date}, "
+            f"open_date={self.open_date}, expire_date={self.expire_date}, "
+            f"basis_price={self.basis_price}, volume_multiple={self.volume_multiple}, "
+            f"underlying_instrument_id={self.underlying_instrument_id}, underlying_multiple={self.underlying_multiple}, "
+            f"option_type={self.option_type}, strike_price={self.strike_price}, "
+            f"price_tick={self.price_tick}, limit_price_min_vol={self.limit_price_min_vol})"
+        )

{gjdatac_api-0.1.4 → gjdatac_api-0.1.6}/gjdatac_api.egg-info/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.1
 Name: gjdatac-api
-Version: 0.1.4
-Summary: vwap,twap接口数据，接入tick_l2数据，新增实时订阅接口
+Version: 0.1.6
+Summary: 数据接口打包
 Author: yin
 Author-email: 2018209921@qq.com
 Requires-Python: >3.6
 Description-Content-Type: text/markdown
 
-将原有接口全部更新，新增后续新需求
+### 新增主力合约接口

{gjdatac_api-0.1.4 → gjdatac_api-0.1.6}/gjdatac_api.egg-info/SOURCES.txt

@@ -3,6 +3,7 @@ gjdatac_api/__init__.py
 gjdatac_api/data_api_class.py
 gjdatac_api/dict_data.py
 gjdatac_api/dolphindb_info.py
+gjdatac_api/instrument_class.py
 gjdatac_api.egg-info/PKG-INFO
 gjdatac_api.egg-info/SOURCES.txt
 gjdatac_api.egg-info/dependency_links.txt

{gjdatac_api-0.1.4 → gjdatac_api-0.1.6}/setup.py

@@ -5,12 +5,12 @@ with open('description_info.md', 'r', encoding='utf-8') as f:
 
 setup(
     name='gjdatac_api',  # 你的包名
-    version='0.1.4',  # 版本号
-    description='vwap,twap接口数据，接入tick_l2数据，新增实时订阅接口',  # 包的简要描述
+    version='0.1.6',  # 版本号
+    description='数据接口打包',  # 包的简要描述
     long_description=long_description,  # 包的详细描述
     long_description_content_type='text/markdown',  # 描述文件的类型
     include_package_data=True,  # 包含包数据
-    package_data={'gjdatac_api': ['*.py', 'dict_data.py', 'dolphindb_info.py']},  # 指定数据文件
+    package_data={'gjdatac_api': ['*.py', 'dict_data.py', 'dolphindb_info.py', 'instrument_class.py']},  # 指定数据文件
     author='yin',  # 作者姓名
     author_email='2018209921@qq.com',  # 作者邮箱
     packages=find_packages(),  # 自动查找包目录