ezKit 1.10.3__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/utils.py

@@ -13,7 +13,7 @@ from multiprocessing import Pool
 from multiprocessing.pool import ThreadPool
 from pathlib import Path
 from shutil import rmtree
-from typing import Any, Callable, List, Union
+from typing import Any, Callable, List, Optional, Union
 from urllib.parse import ParseResult, urlparse
 from uuid import uuid4
 
@@ -1001,46 +1001,45 @@ def datetime_object(
 # --------------------------------------------------------------------------------------------------
 
 
-# run_cmd = bash('echo ok', universal_newlines=True, stdout=PIPE)
-#
-# if run_cmd != None:
-#     returncode = run_cmd.returncode
-#     outputs = run_cmd.stdout.splitlines()
-#     print(returncode, type(returncode))
-#     print(outputs, type(outputs))
-#
-# # echo 'echo ok' > /tmp/ok.sh
-# run_script = bash('/tmp/ok.sh', file=True, universal_newlines=True, stdout=PIPE)
-#
-# if run_script != None:
-#     returncode = run_script.returncode
-#     outputs = run_script.stdout.splitlines()
-#     print(returncode, type(returncode))
-#     print(outputs, type(outputs))
-
-
 def shell(
     command: str,
     isfile: bool = False,
-    sh_shell: str = '/bin/bash',
-    sh_option: str | None = None,
+    sh_shell: str = "/bin/bash",
+    sh_option: Optional[str] = None,
     **kwargs
-) -> subprocess.CompletedProcess | None:
-    """run shell command or script"""
+) -> Optional[subprocess.CompletedProcess]:
+    """执行 Shell 命令 或 脚本"""
+
+    # :param command: 需要执行的命令
+    # :param isfile: 是否将命令视为文件路径
+    # :param sh_shell: 使用的 Shell 程序路径
+    # :param sh_option: Shell 执行选项，例如 '-c'
+    # :param kwargs: 其他传递给 subprocess.run 的参数
+    # :return: 返回 subprocess.CompletedProcess 对象，失败时返回 None
+
     try:
-        match True:
-            case True if not check_file_type(sh_shell, 'file'):
-                return None
-            case True if isTrue(sh_shell, str) and isTrue(command, str):
-                if isfile is True:
-                    if sh_option is None:
-                        return subprocess.run([sh_shell, command], **kwargs, check=False)
-                    return subprocess.run([sh_shell, sh_option, command], **kwargs, check=False)
-                if sh_option is None:
-                    sh_option = '-c'
-                return subprocess.run([sh_shell, sh_option, command], **kwargs, check=False)
-            case _:
-                return None
+
+        # 校验 Shell 程序路径
+        if not check_file_type(sh_shell, "file"):
+            logger.error(f"Invalid shell path: {sh_shell}")
+            return None
+
+        # 校验 command 和 sh_shell 的类型
+        if not (isTrue(sh_shell, str) and isTrue(command, str)):
+            logger.error("Invalid shell or command input.")
+            return None
+
+        # 构造命令
+        if isfile:
+            args = [sh_shell, command] if sh_option is None else [sh_shell, sh_option, command]
+        else:
+            sh_option = sh_option or "-c"
+            args = [sh_shell, sh_option, command]
+
+        logger.info(f"Executing command: {args}")
+
+        return subprocess.run(args, **kwargs, check=False)
+
     except Exception as e:
         logger.exception(e)
         return None

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

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

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

@@ -2,17 +2,17 @@ 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
 ezKit/redis.py,sha256=g2_V4jvq0djRc20jLZkgeAeF_bYrq-Rbl_kHcCUPZcA,1965
 ezKit/sendemail.py,sha256=tRXCsJm_RfTJ9xEWe_lTQ5kOs2JxHGPXvq0oWA7prq0,7263
 ezKit/token.py,sha256=HKREyZj_T2S8-aFoFIrBXTaCKExQq4zE66OHXhGHqQg,1750
-ezKit/utils.py,sha256=k3hSnOwNSyyRDVwfEzQUXQh_oJJ51KOT-PvLFPYOtOE,42517
+ezKit/utils.py,sha256=ILLaptYUSalERbCOhmoq4w0ZNZgj9yurpHEZwjRMZ8w,42387
 ezKit/xftp.py,sha256=XyIdr_2rxRVLqPofG6fIYWhAMVsFwTyp46dg5P9FLW4,7774
-ezKit-1.10.3.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-ezKit-1.10.3.dist-info/METADATA,sha256=19kgwO-MdMKnsJvQ55DaYMC07qdEmyoJ29cj0qaIF9A,191
-ezKit-1.10.3.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-ezKit-1.10.3.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
-ezKit-1.10.3.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,,