datahub_binary 1.9.2__cp312-cp312-win_amd64.whl → 1.9.4__cp312-cp312-win_amd64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
datahub/dbo/database.pyi CHANGED
@@ -20,6 +20,22 @@ class Database:
20
20
  :param data: 要插入的数据列表
21
21
  :return: 影响的行数
22
22
  """
23
+ def insert_ignore(self, table_name: str, data: dict, keys: list) -> bool:
24
+ """
25
+ 插入忽略,不存在就插入,存在就忽略(单条)
26
+ :param table_name: 表名
27
+ :param data: 插入数据
28
+ :param keys: 关键字,关键字相同则认为数据相同
29
+ :return: 是否成功
30
+ """
31
+ def insert_many_ignore(self, table_name: str, data: list[dict], keys: list) -> bool:
32
+ """
33
+ 批量插入忽略,不存在就插入,存在就忽略
34
+ :param table_name: 表名
35
+ :param data: 插入数据 [{row1:v1, row2:v2...}...]
36
+ :param keys: 关键字,关键字相同则认为数据相同
37
+ :return: 是否成功
38
+ """
23
39
  def upsert(self, table_name: str, data: dict, keys: list) -> bool:
24
40
  """
25
41
  插入更新,不存在就插入,存在就更新
@@ -63,7 +79,7 @@ class Database:
63
79
  '''
64
80
  查询数据
65
81
  :param table_name: 表名
66
- :param filters: "字段名"=值
82
+ :param filters: "字段名"=值(支持单值或列表)
67
83
  :return: 查询结果列表
68
84
  '''
69
85
  def query(self, sql: str, return_format: Literal['dataframe', 'records'] = 'records') -> pl.DataFrame | list[dict] | None:
@@ -77,7 +93,7 @@ class Database:
77
93
  @contextmanager
78
94
  def get_session(self) -> Generator[Session, Any, None]:
79
95
  """获取一个数据库会话,支持用户手动管理 session 生命周期"""
80
- def query_with_session(self, sql: str, session: Session, return_format: Literal['dataframe', 'records'] = 'dataframe'):
96
+ def query_with_session(self, sql: str, session: Session, return_format: Literal['dataframe', 'records'] = 'records'):
81
97
  """
82
98
  执行原生SQL查询
83
99
 
datahub/dbo/pg.pyi CHANGED
@@ -9,6 +9,7 @@ from typing import Any, Sequence
9
9
  class Postgres(Database):
10
10
  db_url: Incomplete
11
11
  def __init__(self, setting: PostgresSetting) -> None: ...
12
+ def query(self, sql, return_format: str = 'dataframe'): ...
12
13
  def upsert_many(self, table_name: str, data: list[dict[str, Any]], keys: list[str]) -> int:
13
14
  """
14
15
  根据给定的keys, 批量插入更新数据
@@ -35,7 +36,7 @@ class Postgres(Database):
35
36
  :param end_date: 截止日期最新的, 不传则为当前日期
36
37
  :return:
37
38
  """
38
- def get_sbl_list(self, brokers: Sequence[str] = (), sbl_ids: Sequence[str] = (), start_date: date | None = None, end_date: date | None = None) -> pl.DataFrame:
39
+ def get_sbl_list(self, brokers: Sequence[str] = (), sbl_ids: Sequence[str] = (), start_date: date | None = None, end_date: date | None = None, version: int = 1) -> pl.DataFrame:
39
40
  """
40
41
  截止到end_date最新的券单, 一日可能有多批
41
42
 
@@ -43,5 +44,6 @@ class Postgres(Database):
43
44
  :param sbl_ids: 券池id,默认全部
44
45
  :param start_date: 开始时间, 默认为 2000-01-01
45
46
  :param end_date: 截止日期最新的, 默认为当前日期
47
+ :param version: 当日第几批券池,默认第1批
46
48
  :return:
47
49
  """
datahub/dbo/sr.pyi CHANGED
@@ -18,6 +18,7 @@ class StarRocks(Database):
18
18
  sftp_client: Incomplete
19
19
  force_query: Incomplete
20
20
  def __init__(self, setting: StarRocksSetting) -> None: ...
21
+ def query(self, sql, return_format: str = 'dataframe'): ...
21
22
  def query_with_cache(self, sql: str, return_format: Literal['dataframe', 'records'] = 'dataframe') -> pl.DataFrame | list[dict] | None:
22
23
  """
23
24
  使用查询缓存, 用于下载大量数据
Binary file
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: datahub_binary
3
- Version: 1.9.2
3
+ Version: 1.9.4
4
4
  Summary: A comprehensive Python library for data processing, integration, and management.
5
5
  Requires-Python: <3.13,>=3.10
6
6
  Description-Content-Type: text/markdown
@@ -1,14 +1,14 @@
1
- datahub.cp312-win_amd64.pyd,sha256=-vggRQU9JvZgGMCQp9ofLEcXNv9ayp0LSsfJy7eMJv8,1893376
1
+ datahub.cp312-win_amd64.pyd,sha256=T8Ut9F3uEZpJfo5gViK7Ky6IcVbAd8gW9daCPWiWOt4,1923584
2
2
  datahub.pyi,sha256=biltT_CgztnMyL9mbp332i5vT7LUMws0bN6fdQ3WYH4,1555
3
3
  datahub/__init__.pyi,sha256=hX8D9Qd6uPjZiUBd6rIq7Llkt_H4l_RvRXsws3sdYbI,630
4
4
  datahub/datacache.pyi,sha256=Z2b3ktYv5t2y9Z1yXn0axbdoV06-GhjJhEIl_Ms-FjU,5044
5
5
  datahub/datahub.pyi,sha256=NKfhw00iPMsp0Z_q4usO_D7xB6-Hr50wRxpe3KGYH7E,51370
6
6
  datahub/setting.pyi,sha256=u6gYjjF6r1thSpp2XNMvg_1JUcqB64dRTkNoHlOoKUQ,889
7
7
  datahub/ats/client_sdk.pyi,sha256=RScBpyY2ML84RBrYt60RnHtodOwWOe5-LbnJ3nFFBGs,8710
8
- datahub/dbo/database.pyi,sha256=A6Yk-0Z_fSy7F7YeBowpebGvq0RfGK7-lSVQ54USnjs,3355
9
- datahub/dbo/pg.pyi,sha256=fz2vWRF7vd5ANW5dVQAf7NfTzzjFN5aRKSw32nPRN4M,1931
8
+ datahub/dbo/database.pyi,sha256=zGncpT29ygIdMu3i8SzApdx2X08GBBspHor_Wy7PzjU,4100
9
+ datahub/dbo/pg.pyi,sha256=diMoeSPBqOswXYvVjtpfF5XklWqa_f20vQ1TPiHZ7WA,2077
10
10
  datahub/dbo/redis_stream.pyi,sha256=geVI1BoTHRlxodS44MquM-FY_aJt27eDqmAKf2a4PyM,1938
11
- datahub/dbo/sr.pyi,sha256=WWslzbQ11sasBqORbba3U5IZsAYWCiG4jeZK_f5-FZg,28579
11
+ datahub/dbo/sr.pyi,sha256=ZKcUUxGP-s4xGr_zF1qgVlRZcL4fkyROAOiJHuX-Fwk,28644
12
12
  datahub/protos/client_msg.pyi,sha256=ftmkInjCGv1DEt477f5dLmRuFdVzrJ40jbOFGr3ZrZU,4063
13
13
  datahub/protos/client_pb2.pyi,sha256=DCbERSx2ZKHyGn_0ClE_lBScMWi6RBqjCExpIuIq8ww,229745
14
14
  datahub/protos/common_pb2.pyi,sha256=0xUOqWzJ8m_pOQhFEosdwljqZWRaVYz0-H_qmqAOmHs,33627
@@ -25,7 +25,7 @@ datahub/utils/sftp.pyi,sha256=w8S-cxJWMeWJ6--RjT95qVgng_IhwPImiyoqdCpOCQI,2107
25
25
  datahub/utils/monitor/__init__.pyi,sha256=EOwgf8CbJ4g3iUAaFoR6O-mYemJ9xjP42zlBj2KeA9c,76
26
26
  datahub/utils/monitor/base.pyi,sha256=n1dKYK73JJ_7QPkCWFc-6Aj_a16j2z_VBaQ9zkN9A3E,310
27
27
  datahub/utils/monitor/feishu.pyi,sha256=GgvJXYeX3cPOQjqpV0GJUr_Ri1_cZe2-viRBkpe6O_g,402
28
- datahub_binary-1.9.2.dist-info/METADATA,sha256=sNpDjr_oy6TEg4XR4W1GJzT2Sf0f8MihlVKoDLpcy5g,7903
29
- datahub_binary-1.9.2.dist-info/WHEEL,sha256=mktjIo72eqyqXY-lxdbITmUE47ZU2O9WYzyv0PnFKrU,96
30
- datahub_binary-1.9.2.dist-info/top_level.txt,sha256=J0rzqYfZghMgpObWaPGhQtV88I-KXrgArDvi_ddf6ZA,8
31
- datahub_binary-1.9.2.dist-info/RECORD,,
28
+ datahub_binary-1.9.4.dist-info/METADATA,sha256=WhZ8OR0YNi_Mxrxth7-omQuKEboU846UkU99qBTqK_c,7903
29
+ datahub_binary-1.9.4.dist-info/WHEEL,sha256=mktjIo72eqyqXY-lxdbITmUE47ZU2O9WYzyv0PnFKrU,96
30
+ datahub_binary-1.9.4.dist-info/top_level.txt,sha256=J0rzqYfZghMgpObWaPGhQtV88I-KXrgArDvi_ddf6ZA,8
31
+ datahub_binary-1.9.4.dist-info/RECORD,,