cppackage 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.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.
CPpackage/db/__init__.py CHANGED
@@ -1,2 +1,2 @@
1
- from .sql_model import sel_data, update_data, updata, update_datas, del_data, deldata, deldata_time, update_test
1
+ from .sql_model import sel_data, update_datas, del_data
2
2
  from .config import set_db_config, get_db_config
CPpackage/db/sql_model.py CHANGED
@@ -32,7 +32,7 @@ def _get_connection(database=None, port=None):
32
32
  )
33
33
 
34
34
 
35
- # ===================== 基础操作 =====================
35
+ # ===================== 查询操作 =====================
36
36
 
37
37
  def sel_data(sql, params=None, port=None, database=None):
38
38
 
@@ -45,9 +45,16 @@ def sel_data(sql, params=None, port=None, database=None):
45
45
  except pymysql.MySQLError as e:
46
46
  print("查询失败:", e)
47
47
  return None
48
+
49
+ # ===================== 增量更新 =====================
50
+ def incremental_update(sql, values, port=None, database=None):
51
+ with _get_connection(database=database, port=port) as conn:
52
+ with conn.cursor() as cursor:
53
+ cursor.executemany(sql, values)
54
+ conn.commit()
48
55
 
49
56
 
50
-
57
+ # ===================== 删除操作 =====================
51
58
  def del_data(sql, params=None, port=None, database=None):
52
59
 
53
60
  try:
@@ -134,7 +141,6 @@ def update_datas(df, table_name, database):
134
141
 
135
142
  try:
136
143
  # 表结构校验
137
- check_and_sync_columns(df, table_name, database)
138
144
  conn = _get_connection(database)
139
145
  cursor = conn.cursor()
140
146
  cols = list(df.columns)
@@ -0,0 +1,105 @@
1
+ import pymysql
2
+ import time
3
+ from pymysql import Error
4
+ from .config import get_db_config
5
+
6
+ def _get_connection(database=None, port=None):
7
+ cfg = get_db_config()
8
+ db = database if database is not None else cfg.get('database')
9
+ prt = port if port is not None else cfg.get('port', 3306)
10
+ return pymysql.connect(
11
+ host=cfg.get('host'),
12
+ user=cfg.get('user'),
13
+ password=cfg.get('password'),
14
+ database=db,
15
+ port=prt,
16
+ )
17
+
18
+ def sel_data(sql, port=None, database=None):
19
+ try:
20
+ conn = _get_connection(database=database, port=port)
21
+ cursor = conn.cursor()
22
+ cursor.execute(sql)
23
+ result = cursor.fetchall()
24
+ cursor.close()
25
+ conn.close()
26
+ return result
27
+ except pymysql.MySQLError as e:
28
+ print(f"An error occurred: {e}")
29
+
30
+ def update_data(sql, values, port=None, database=None):
31
+ conn = _get_connection(database=database, port=port)
32
+ cursor = conn.cursor()
33
+ cursor.execute(sql, values)
34
+ conn.commit()
35
+ cursor.close()
36
+ conn.close()
37
+
38
+ def updata(data, databases, table):
39
+ s = ""
40
+ st1 = ""
41
+ for i in data.columns.values:
42
+ s = s + "`" + i + "`,"
43
+ st1 = st1 + "%s,"
44
+ s = s[:-1]
45
+ st1 = st1[:-1]
46
+ all_data = []
47
+ for _, k in data.iterrows():
48
+ all_data.append(tuple(k.to_list()))
49
+ sql = 'INSERT INTO %s (%s) VALUES (%s)' % (table, s, st1)
50
+ update_datas(sql, tuple(all_data), database=databases)
51
+ time.sleep(2)
52
+
53
+ def deldata(all_del_list, id_name, table_name, database):
54
+ str11 = ""
55
+ for te in all_del_list:
56
+ if "str" in str(type(te)):
57
+ str11 = str11 + "'" + str(te) + "',"
58
+ else:
59
+ str11 = str11 + str(te) + ","
60
+ str11 = str11[:-1]
61
+ delsql = 'DELETE from `%s` where %s in (%s)' % (table_name, id_name, str11)
62
+ del_data(delsql, database=database)
63
+
64
+ def deldata_time(id_name, table_name, id, database, time_name, start_time, end_time):
65
+ delsql = 'DELETE from `%s` where %s = %s and `%s`>= "%s" and `%s`<= "%s"' % (
66
+ table_name, id_name, id, time_name, start_time, time_name, end_time
67
+ )
68
+ del_data(delsql, database=database)
69
+
70
+ def update_datas(sql, values, port=None, database=None):
71
+ with _get_connection(database=database, port=port) as conn:
72
+ with conn.cursor() as cursor:
73
+ cursor.executemany(sql, values)
74
+ conn.commit()
75
+
76
+ def del_data(sql, port=None, database=None):
77
+ try:
78
+ conn = _get_connection(database=database, port=port)
79
+ cursor = conn.cursor()
80
+ cursor.execute(sql)
81
+ conn.commit()
82
+ cursor.close()
83
+ conn.close()
84
+ except pymysql.MySQLError as e:
85
+ print(f"An error occurred: {e}")
86
+
87
+ def update_test(df, table_name, databases):
88
+ try:
89
+ conn = _get_connection(database=databases)
90
+ cursor = conn.cursor()
91
+ columns = ', '.join(df.columns)
92
+ values_template = ', '.join([f'({", ".join(["%s"] * len(df.columns))})'] * len(df))
93
+ update_clause = ', '.join([f'{col} = VALUES({col})' for col in df.columns if col != 'id'])
94
+ sql = f"INSERT INTO {table_name} ({columns}) VALUES {values_template} " \
95
+ f"ON DUPLICATE KEY UPDATE {update_clause}"
96
+ values = [tuple(row) for row in df.values]
97
+ flat_values = [item for sublist in values for item in sublist]
98
+ cursor.execute(sql, flat_values)
99
+ conn.commit()
100
+ print(f"成功插入/更新 {cursor.rowcount} 条记录")
101
+ except Error as e:
102
+ print(f"数据库错误: {e}")
103
+ finally:
104
+ if conn:
105
+ conn.close()
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: cppackage
3
- Version: 0.2.1
3
+ Version: 0.2.2
4
4
  Summary: 超品集团自用的Python包
5
5
  Home-page: https://github.com/example/CPpackage
6
6
  Author: team-数智组
@@ -40,7 +40,7 @@ CPpackage是超品集团内部使用的Python包,核心功能集中在MySQL数
40
40
  ## 安装
41
41
 
42
42
  ```bash
43
- pip install CPpackage
43
+ pip install cppackage
44
44
  ```
45
45
 
46
46
  ## 使用方法
@@ -48,13 +48,14 @@ pip install CPpackage
48
48
  ```python
49
49
  # 导入包
50
50
  from CPpackage.db.config import set_db_config
51
- from CPpackage.db.sql_model import sel_data, del_data , get_table_columns,add_columns,check_and_sync_columns,update_datas
51
+ from CPpackage.db.sql_model import sel_data, del_data , incremental_update, get_table_columns,add_columns,check_and_sync_columns,update_datas
52
52
 
53
53
  # 配置数据库连接(也可通过环境变量CPPACKAGE_DB_*配置)
54
54
  set_db_config(host='your_host', user='your_user', password='your_password', database='your_database', port=3306)
55
55
 
56
- # 更新数据函数,在完成数据库信息配置后,df为pd.dataframe格式的数据
57
56
  # 更新过程会先经过check_and_sync_columns函数完成表格校验
57
+ check_and_sync_columns(df, table_name, database, max_new=5)
58
+ # 更新数据函数,在完成数据库信息配置后,df为pd.dataframe格式的数据
58
59
  update_datas(df, table_name, database)
59
60
 
60
61
  ```
@@ -0,0 +1,13 @@
1
+ CPpackage/__init__.py,sha256=KpVaXkIzkNDl-dfSX8Rr7z9XFlipxx1nZ_GzS4Qls4I,173
2
+ CPpackage/core/__init__.py,sha256=92mF0310uQ5ujlgd-LCMs5kZVzfGRdcl67GHIRWhHcA,504
3
+ CPpackage/db/__init__.py,sha256=WhrLqsjq97KKHF7YV-HcZf9pRtHuqKhRjXDNPCBVbjY,107
4
+ CPpackage/db/config.py,sha256=wygya2246ph0LY90xikQBVoRxS8vlI-RpWWNGOiUZ5s,723
5
+ CPpackage/db/sql_model.py,sha256=UbTSzpp9WiKy9_ub52yhCDV46elJuFlKILzTgc-gs48,4733
6
+ CPpackage/db/sql_modelcopy.py,sha256=tLE88fXE-EZavsqIqH7ZvkXy5-7ZA-cQCMPMQmRG0Mg,3564
7
+ CPpackage/utils/__init__.py,sha256=sAespT0aoLr3O0WF27DwHLYF60VZgPA1tiT0JJ8s8zI,264
8
+ cppackage-0.2.2.dist-info/licenses/LICENSE,sha256=6Pj597LnvlRtq3Vgjkfp6zgBeBh7wl8jKE-nCFg4vZI,1065
9
+ cppackage-0.2.2.dist-info/METADATA,sha256=Mqi578TfPTeayB-37A8hSo6DNmrHBab_lgXqy_QH7fQ,2510
10
+ cppackage-0.2.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
11
+ cppackage-0.2.2.dist-info/entry_points.txt,sha256=SAzF2klkNyeNCqXuiNkDirN999l7v8CGCrHR0StD_oo,50
12
+ cppackage-0.2.2.dist-info/top_level.txt,sha256=FQrfWDbJistWQIrHW_Aoxoy_UWSzjOByWVcVS5ig9Tk,10
13
+ cppackage-0.2.2.dist-info/RECORD,,
@@ -1,12 +0,0 @@
1
- CPpackage/__init__.py,sha256=KpVaXkIzkNDl-dfSX8Rr7z9XFlipxx1nZ_GzS4Qls4I,173
2
- CPpackage/core/__init__.py,sha256=92mF0310uQ5ujlgd-LCMs5kZVzfGRdcl67GHIRWhHcA,504
3
- CPpackage/db/__init__.py,sha256=Uw09n27c0k2lM61oYw0GMpnbIafPKiY9EAMudxBn7IY,164
4
- CPpackage/db/config.py,sha256=wygya2246ph0LY90xikQBVoRxS8vlI-RpWWNGOiUZ5s,723
5
- CPpackage/db/sql_model.py,sha256=qj8NQctWkIJbdnLYDfg2mpOKl-STlz1lbaeiV9D-JY0,4433
6
- CPpackage/utils/__init__.py,sha256=sAespT0aoLr3O0WF27DwHLYF60VZgPA1tiT0JJ8s8zI,264
7
- cppackage-0.2.1.dist-info/licenses/LICENSE,sha256=6Pj597LnvlRtq3Vgjkfp6zgBeBh7wl8jKE-nCFg4vZI,1065
8
- cppackage-0.2.1.dist-info/METADATA,sha256=UBo6vBIsitJAUXFX4j3RSnF44QlnIof9LnZ0frzQ8eQ,2429
9
- cppackage-0.2.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
10
- cppackage-0.2.1.dist-info/entry_points.txt,sha256=SAzF2klkNyeNCqXuiNkDirN999l7v8CGCrHR0StD_oo,50
11
- cppackage-0.2.1.dist-info/top_level.txt,sha256=FQrfWDbJistWQIrHW_Aoxoy_UWSzjOByWVcVS5ig9Tk,10
12
- cppackage-0.2.1.dist-info/RECORD,,