ezKit 1.10.4__py3-none-any.whl → 1.10.5__py3-none-any.whl

ezKit/database.py

@@ -6,8 +6,10 @@
 # PostgreSQL 14 Data Types
 #     https://www.postgresql.org/docs/14/datatype.html
 import csv
+import json
 from typing import Any
 
+import pandas as pd
 from loguru import logger
 from sqlalchemy import CursorResult, Index, create_engine, text
 from sqlalchemy.orm import DeclarativeBase
@@ -30,10 +32,14 @@ class Database():
         else:
             pass
 
+    # ----------------------------------------------------------------------------------------------
+
     def initializer(self):
         """ensure the parent proc's database connections are not touched in the new connection pool"""
         self.engine.dispose(close=False)
 
+    # ----------------------------------------------------------------------------------------------
+
     def connect_test(self) -> bool:
         info = "Database connect test"
         try:
@@ -46,6 +52,8 @@ class Database():
             logger.exception(e)
             return False
 
+    # ----------------------------------------------------------------------------------------------
+
     def metadata_init(self, base: DeclarativeBase, **kwargs) -> bool:
         # https://stackoverflow.com/questions/19175311/how-to-create-only-one-table-with-sqlalchemy
         info = "Database init table"
@@ -60,6 +68,8 @@ class Database():
             logger.exception(e)
             return False
 
+    # ----------------------------------------------------------------------------------------------
+
     def create_index(self, index_name, table_field) -> bool:
         # 创建索引
         #   https://stackoverflow.com/a/41254430
@@ -82,6 +92,8 @@ class Database():
             logger.error(e)
             return False
 
+    # ----------------------------------------------------------------------------------------------
+
     # 私有函数, 保存 execute 的结果到 CSV 文件
     def _result_save(self, file, data) -> bool:
         try:
@@ -93,6 +105,8 @@ class Database():
             logger.exception(e)
             return False
 
+    # ----------------------------------------------------------------------------------------------
+
     def execute(
         self,
         sql: str | None = None,
@@ -204,3 +218,46 @@ class Database():
             logger.error(f'{info} [failure]')
             logger.exception(e)
             return False
+
+    # ----------------------------------------------------------------------------------------------
+
+    def read_data_with_pandas(self, result_type: str = "df", **kwargs) -> pd.DataFrame | dict | list | None:
+        """读取表中所有数据"""
+
+        # 使用 pd.read_sql_table 的参数
+        # read_data_with_pandas(result_type="df", table_name="ashare")
+
+        info = f"读取 {kwargs.get('table_name', None)} 表中所有数据"
+
+        try:
+
+            logger.info(f"{info} ......")
+
+            # 从 kwargs 中删除 con 键
+            kwargs.pop('con', None)
+
+            # 读取数据
+            data: pd.DataFrame = pd.read_sql_table(con=self.engine, **kwargs)
+
+            if data.empty:
+                logger.error(f"{info} [失败]")
+                return None
+
+            logger.success(f"{info} [成功]")
+
+            if utils.isTrue(result_type, str) and result_type == "json":
+                return json.loads(data.to_json(orient='records'))
+
+            if utils.isTrue(result_type, str) and result_type == "dict":
+                return data.to_dict()
+
+            if utils.isTrue(result_type, str) and result_type == "list":
+                # https://stackoverflow.com/a/26716774
+                return data.to_dict('list')
+
+            return data
+
+        except Exception as e:
+            logger.error(f"{info} [失败]")
+            logger.exception(e)
+            return None

{ezKit-1.10.4.dist-info → ezKit-1.10.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ezKit
-Version: 1.10.4
+Version: 1.10.5
 Summary: Easy Kit
 Author: septvean
 Author-email: septvean@gmail.com

{ezKit-1.10.4.dist-info → ezKit-1.10.5.dist-info}/RECORD

@@ -2,7 +2,7 @@ ezKit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ezKit/bottle.py,sha256=usKK1wVaZw4_D-4VwMYmOIc8jtz4TrpM30nck59HMFw,180178
 ezKit/bottle_extensions.py,sha256=3reEQVZuHklXTl6r7F8kiBFFPb0RaAGc3mYJJnrMDjQ,1129
 ezKit/cipher.py,sha256=0T_StbjiNI4zgrjVgcfU-ffKgu1waBA9UDudAnqFcNM,2896
-ezKit/database.py,sha256=Rc4RgjHOOtf5dMLvMkK1beRfbIai5E1x4HTsDwKsA-Q,6822
+ezKit/database.py,sha256=b_4gR_bvyC3IhP6TbmeKdVrif1DNG_NqO3YIyy-XqQE,8891
 ezKit/http.py,sha256=i3Kn5AMAMicDMcDjxKKZU7zqEKTU88Ec9_LwCuBJy-0,1801
 ezKit/mongo.py,sha256=dOm_1wXEPp_e8Ml5Qq78M7FDNrQUAZaThzVIiiLJJwk,2393
 ezKit/qywx.py,sha256=X_H4fzP-iEqeDEbumr7D1bXi6dxczaxfO8iyutzy02s,7171
@@ -11,8 +11,8 @@ ezKit/sendemail.py,sha256=tRXCsJm_RfTJ9xEWe_lTQ5kOs2JxHGPXvq0oWA7prq0,7263
 ezKit/token.py,sha256=HKREyZj_T2S8-aFoFIrBXTaCKExQq4zE66OHXhGHqQg,1750
 ezKit/utils.py,sha256=ILLaptYUSalERbCOhmoq4w0ZNZgj9yurpHEZwjRMZ8w,42387
 ezKit/xftp.py,sha256=XyIdr_2rxRVLqPofG6fIYWhAMVsFwTyp46dg5P9FLW4,7774
-ezKit-1.10.4.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-ezKit-1.10.4.dist-info/METADATA,sha256=uH8X29vT0V5J3nwzW7pwMvgor45UyDqw_1iGCwU_9A4,191
-ezKit-1.10.4.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-ezKit-1.10.4.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
-ezKit-1.10.4.dist-info/RECORD,,
+ezKit-1.10.5.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+ezKit-1.10.5.dist-info/METADATA,sha256=onqUqJ4_5WVpBoLpSOis-Z6uYbmM3GCMPPrAfTvi_yw,191
+ezKit-1.10.5.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+ezKit-1.10.5.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
+ezKit-1.10.5.dist-info/RECORD,,