gjdatac-api 0.2.7__tar.gz → 0.2.9__tar.gz

{gjdatac_api-0.2.7 → gjdatac_api-0.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gjdatac_api
-Version: 0.2.7
+Version: 0.2.9
 Summary: 数据接口打包
 Author: yin
 Author-email: 2018209921@qq.com

{gjdatac_api-0.2.7 → gjdatac_api-0.2.9}/gjdatac_api/data_api_class.py

@@ -24,7 +24,7 @@ from .instrument_class import Instrument
 
 class GJDataC:
 
-    def __init__(self, host, port, user_id, password, dolphindb_service):
+    def __init__(self, host, port, user_id, password, dolphindb_service=None):
         """
         用于定义一些不同接口所需的变量以及一些初始化操作
         :param host: 数据库地址
@@ -1135,7 +1135,7 @@ class GJDataC:
             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, is_include_start, is_include_end)
-        elif trade_time and date_count and date_type:  # 查询前/后 N个交易日/时间数据
+        elif trade_time and date_count is not None 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")
@@ -3845,8 +3845,7 @@ class GJDataC:
 
         return trading_dates
 
-    @staticmethod
-    def get_trading_days(trading_dates, trade_time, date_count):
+    def get_trading_days(self, trading_dates, trade_time, date_count):
         """
         查询前后 N 个交易日
         :param trading_dates: 交易日历数据
@@ -3862,14 +3861,17 @@ class GJDataC:
             # 向前查询
             filtered_dates = trading_dates[trading_dates['trading_day'] < trade_time]
             result = filtered_dates['trading_day'].sort_values(ascending=False).head(abs(date_count))
-        else:
+        elif date_count > 0:
             # 向后查询
             filtered_dates = trading_dates[trading_dates['trading_day'] > trade_time]
             result = filtered_dates['trading_day'].sort_values().head(date_count)
+        else:
+            filtered_dates = trading_dates[trading_dates['trading_day'] == trade_time]
+            result = filtered_dates['trading_day']
 
         # 将结果转换为指定格式的字符串列表
         fina_result = sorted(result.dt.strftime('%Y.%m.%d').tolist())
-        return fina_result[0] if len(fina_result) <= 1 else fina_result
+        return fina_result if len(fina_result) <= 1 else fina_result
 
     @staticmethod
     def get_trading_minutes(trading_dates, trade_time, date_count):
@@ -3886,10 +3888,13 @@ class GJDataC:
         # 筛选 trade_flag 列等于 T 的数据（是交易日的数据）
         trading_dates = trading_dates[trading_dates['trade_flag'] == 'T']
 
+        # 转换字符串为 datetime
+        trade_time = datetime.datetime.strptime(trade_time, '%Y.%m.%d %H:%M:%S')
+
         # 生成时间点列表
         time_points = pd.date_range(start=trade_time - datetime.timedelta(minutes=abs(date_count)),
                                     end=trade_time + datetime.timedelta(minutes=abs(date_count)),
-                                    freq='T')  # 'T' 表示分钟频率
+                                    freq='min')  # 'T' 表示分钟频率
 
         # 将时间点转换为指定格式的字符串
         time_points_str = time_points.strftime('%Y.%m.%d %H:%M:%S')
@@ -3905,11 +3910,14 @@ class GJDataC:
         if date_count < 0:
             # 向前查询
             result = [p for p in valid_time_points if pd.to_datetime(p) < trade_time][-abs(date_count):]
-        else:
+        elif date_count > 0:
             # 向后查询
             result = [p for p in valid_time_points if pd.to_datetime(p) > trade_time][:date_count]
+        else:
+            result = [p for p in valid_time_points if pd.to_datetime(p) == trade_time]
 
-        return sorted(result) if len(result) > 1 else result[0]
+        # return sorted(result) if len(result) > 1 else result[0]
+        return sorted(result) if len(result) > 1 else result
 
     """ 交易日判断 """
 

{gjdatac_api-0.2.7 → gjdatac_api-0.2.9}/gjdatac_api.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gjdatac-api
-Version: 0.2.7
+Version: 0.2.9
 Summary: 数据接口打包
 Author: yin
 Author-email: 2018209921@qq.com

{gjdatac_api-0.2.7 → gjdatac_api-0.2.9}/setup.py

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