ezKit 1.12.2__tar.gz → 1.12.4__tar.gz

{ezkit-1.12.2/ezKit.egg-info → ezkit-1.12.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ezKit
-Version: 1.12.2
+Version: 1.12.4
 Summary: Easy Kit
 Author: septvean
 Author-email: septvean@gmail.com

{ezkit-1.12.2 → ezkit-1.12.4}/ezKit/database.py

@@ -8,15 +8,17 @@
 #     https://www.postgresql.org/docs/14/datatype.html
 import csv
 import json
-from typing import Any
+from typing import Any, Type
 
 import pandas as pd
 from loguru import logger
 from sqlalchemy import CursorResult, Index, create_engine, text
-from sqlalchemy.orm import DeclarativeBase
+from sqlalchemy.orm import DeclarativeBase, Session, declarative_base
 
 from . import utils
 
+Base = declarative_base()
+
 
 class Database:
     """Database"""
@@ -71,6 +73,17 @@ class Database:
 
     # ----------------------------------------------------------------------------------------------
 
+    def initialization_table(self, table: str):
+        """初始化表"""
+
+        # 初始化所有表
+        #   db.metadata_init(Base)
+        # 初始化指定表
+        #   database.metadata_init(Base, tables=[Base.metadata.tables['ashare']])
+        self.metadata_init(Base, tables=[Base.metadata.tables[table]])
+
+    # ----------------------------------------------------------------------------------------------
+
     def create_index(self, index_name, table_field) -> bool:
         # 创建索引
         #   https://stackoverflow.com/a/41254430
@@ -346,3 +359,36 @@ class Database:
             logger.error(f"{info} [failed]")
             logger.exception(e)
             return data
+
+    # ----------------------------------------------------------------------------------------------
+
+    def create_with_dict_or_list(self, TS: Type, data: dict | list):
+        """将 dict 或 list 类型的数据添加到表中. TS: Table Schema"""
+
+        info: str = "create data"
+
+        try:
+
+            logger.info(f"{info} ......")
+
+            with Session(self.engine) as session:
+
+                if utils.isTrue(data, dict):
+                    logger.info(f"{info} [data type is dict]")
+                    session.add(TS(**data))  # type: ignore
+
+                if utils.isTrue(data, list):
+                    logger.info(f"{info} [data type is list]")
+                    for item in data:
+                        session.add(TS(**item))
+
+                session.commit()
+
+            logger.success(f"{info} [success]")
+
+            return True
+
+        except Exception as e:
+            logger.error(f"{info} [failed]")
+            logger.exception(e)
+            return False

{ezkit-1.12.2 → ezkit-1.12.4/ezKit.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ezKit
-Version: 1.12.2
+Version: 1.12.4
 Summary: Easy Kit
 Author: septvean
 Author-email: septvean@gmail.com

{ezkit-1.12.2 → ezkit-1.12.4}/setup.py

@@ -4,7 +4,7 @@ from setuptools import find_packages, setup
 
 setup(
     name="ezKit",
-    version="1.12.2",
+    version="1.12.4",
     author="septvean",
     author_email="septvean@gmail.com",
     description="Easy Kit",