wei-data-shu 0.4.0__py3-none-any.whl

wei_data_shu/SQLManager.py

@@ -0,0 +1,297 @@
+# !/usr/bin/python
+# -*- coding:utf-8 -*-
+"""
+████─█──█─████─███─█──█─███─███
+█──█─█──█─█──█─█───██─█──█──█──
+████─████─█──█─███─█─██──█──███
+█────█──█─█──█─█───█──█──█──█──
+█────█──█─████─███─█──█─███─███
+╔╗╔╗╔╗╔═══╗╔══╗╔╗──╔══╗╔══╗╔══╗╔═══╗╔══╗
+║║║║║║║╔══╝╚╗╔╝║║──╚╗╔╝║╔╗║║╔╗║╚═╗─║╚╗╔╝
+║║║║║║║╚══╗─║║─║║───║║─║╚╝║║║║║─╔╝╔╝─║║─
+║║║║║║║╔══╝─║║─║║───║║─║╔╗║║║║║╔╝╔╝──║║─
+║╚╝╚╝║║╚══╗╔╝╚╗║╚═╗╔╝╚╗║║║║║╚╝║║─╚═╗╔╝╚╗
+╚═╝╚═╝╚═══╝╚══╝╚══╝╚══╝╚╝╚╝╚══╝╚═══╝╚══╝
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+佛祖保佑       永不宕机     永无BUG
+
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+@project:home
+@author:Phoenix,weiliaozi
+@file:pywork
+@ide:PyCharm
+@date:2023/12/3
+@time:17:33
+@month:十二月
+@email:thisluckyboy@126.com
+"""
+import mysql.connector
+from collections import deque
+# import openpyxl
+# import pandas as pd
+from .timingTool import fn_timer
+from typing import List, Optional, Union, Any
+# import pymysql
+
+# class Database:
+#     def __init__(self, host: str, port: int, user: str, password: str, db: str, charset: str = 'utf8'):
+#         """
+#         Database class for managing MySQL connections and operations.
+#         """
+#         self.connection_state = False
+#         self.connection = None
+#         self.host = host
+#         self.port = port
+#         self.user = user
+#         self.password = password
+#         self.db = db
+#         self.charset = charset
+
+#     def _connect(self) -> None:
+#         """
+#         Establish a connection to the database if not already connected.
+#         """
+#         if not self.connection_state:
+#             try:
+#                 self.connection = pymysql.connect(
+#                     host=self.host,
+#                     port=self.port,
+#                     user=self.user,
+#                     password=self.password,
+#                     db=self.db,
+#                     charset=self.charset,
+#                     cursorclass=pymysql.cursors.DictCursor
+#                 )
+#                 self.connection_state = True
+#             except Exception as e:
+#                 raise ConnectionError(f"Database connection failed: {e}")
+
+#     def _close(self) -> None:
+#         """
+#         Close the database connection if open.
+#         """
+#         if self.connection_state and self.connection:
+#             try:
+#                 self.connection.close()
+#                 self.connection_state = False
+#             except Exception as e:
+#                 raise RuntimeError(f"Error closing the connection: {e}")
+
+#     def _execute(
+#         self, 
+#         sql: str, 
+#         params: Optional[Union[List[tuple], tuple]] = None, 
+#         fetch_all: bool = True, 
+#         as_df: bool = False, 
+#         procedure: bool = False
+#     ) -> Optional[Union[List[dict], pd.DataFrame]]:
+#         """
+#         Execute a SQL query or stored procedure.
+#         """
+#         self._connect()  # Ensure connection is open
+#         try:
+#             with self.connection.cursor() as cursor:
+#                 if procedure:
+#                     cursor.callproc(sql, params)
+#                 elif params:
+#                     cursor.executemany(sql, params) if isinstance(params, list) else cursor.execute(sql, params)
+#                 else:
+#                     cursor.execute(sql)
+
+#                 if fetch_all:
+#                     results = cursor.fetchall()
+#                     return pd.DataFrame(results) if as_df else results
+
+#                 self.connection.commit()
+#         except Exception as e:
+#             self.connection.rollback()
+#             raise RuntimeError(f"Error executing query: {e}")
+#         finally:
+#             self._close()
+
+#     def fetch_all(self, sql: str) -> List[dict]:
+#         """
+#         Fetch all results from a query.
+#         """
+#         return self._execute(sql, fetch_all=True, as_df=False)
+
+#     def fetch_as_df(self, sql: str) -> pd.DataFrame:
+#         """
+#         Fetch results as a pandas DataFrame.
+#         """
+#         return self._execute(sql, fetch_all=True, as_df=True)
+
+#     def execute_write(self, sql: str, params: Optional[Union[List[tuple], tuple]] = None) -> None:
+#         """
+#         Execute a write (INSERT, UPDATE, DELETE) operation.
+#         """
+#         self._execute(sql, params=params, fetch_all=False)
+
+#     def execute_many(self, sql: str, params: List[tuple]) -> None:
+#         """
+#         Execute many write operations (batch insert/update/delete).
+#         """
+#         self._execute(sql, params=params, fetch_all=False)
+
+#     def call_procedure(self, procedure_name: str, params: Optional[tuple] = None) -> None:
+#         """
+#         Call a stored procedure.
+#         """
+#         self._execute(procedure_name, params=params, fetch_all=False, procedure=True)
+
+#     def __call__(
+#         self, 
+#         sql: str, 
+#         params: Optional[Union[List[tuple], tuple]] = None, 
+#         operation_mode: str = "r", 
+#         as_df: bool = False
+#     ) -> Optional[Union[List[dict], pd.DataFrame, None]]:
+#         """
+#         A unified method for executing different types of operations.
+        
+#         Modes:
+#             - 'r': Read query, fetch results.
+#             - 'w': Write operation.
+#             - 'm': Write many (batch operations).
+#             - 'p': Call a stored procedure.
+#         """
+#         if operation_mode == "r":
+#             return self.fetch_as_df(sql) if as_df else self.fetch_all(sql)
+#         elif operation_mode == "w":
+#             self.execute_write(sql, params=params)
+#         elif operation_mode == "m":
+#             if not isinstance(params, list):
+#                 raise ValueError("For batch operations, 'params' must be a list of tuples.")
+#             self.execute_many(sql, params=params)
+#         elif operation_mode == "p":
+#             self.call_procedure(sql, params=params)
+#         else:
+#             raise ValueError(f"Invalid operation mode: {operation_mode}")
+
+# Example Usage:
+# db = Database(host='localhost', port=3306, user='root', password='password', db='testdb')
+# db('SELECT * FROM users', operation_mode='r')
+# db('INSERT INTO users (name, age) VALUES (%s, %s)', params=[('Alice', 30), ('Bob', 25)], operation_mode='m')
+
+class MySQLDatabase:
+    def __init__(self, config):
+        self.config = config
+        self.connection = None
+        self.connect()
+
+    def connect(self):
+        try:
+            self.connection = mysql.connector.connect(**self.config)
+            print("Connected to MySQL database")
+        except mysql.connector.Error as err:
+            print(f"Error: {err}")
+
+    def close(self):
+        if self.connection:
+            self.connection.close()
+            print("MySQL connection closed")
+
+    def execute_query(self, query, params=None):
+        cursor = self.connection.cursor()
+        try:
+            if params:
+                if isinstance(params, list):
+                    cursor.executemany(query, params)
+                else:
+                    cursor.execute(query, params)
+            else:
+                cursor.execute(query)
+            self.connection.commit()
+            print("Query executed successfully")
+        except mysql.connector.Error as err:
+            print(f"Error: {err}")
+        finally:
+            cursor.close()
+
+    def execute_many(self, query, params_list):
+        cursor = self.connection.cursor()
+        try:
+            cursor.executemany(query, params_list)
+            self.connection.commit()
+            print("Batch query executed successfully")
+        except mysql.connector.Error as err:
+            print(f"Error: {err}")
+        finally:
+            cursor.close()
+
+    def fetch_query(self, query, params=None,dictionary=False):
+        cursor = self.connection.cursor(dictionary=dictionary)
+        try:
+            if params:
+                cursor.execute(query, params)
+            else:
+                cursor.execute(query)
+            result = cursor.fetchall()
+            return result
+        except mysql.connector.Error as err:
+            print(f"Error: {err}")
+        finally:
+            cursor.close()            
+            
+    def call_procedure(self, proc_name, params=None):
+        """
+        调用存储过程的方法
+        :param proc_name: 存储过程名称
+        :param params: 存储过程参数,可以是单个值或元组
+        :return: 如果存储过程有返回结果则返回结果集,否则返回None
+        """
+        cursor = self.connection.cursor(dictionary=True)
+        try:
+            if params:
+                cursor.callproc(proc_name, params if isinstance(params, (list, tuple)) else (params,))
+            else:
+                cursor.callproc(proc_name)
+            
+            # 获取存储过程的所有结果集
+            results = []
+            for result in cursor.stored_results():
+                results.extend(result.fetchall())
+                
+            self.connection.commit()
+            return results if results else None
+            
+        except mysql.connector.Error as err:
+            print(f"存储过程调用错误: {err}")
+            self.connection.rollback()
+            return None
+        finally:
+            cursor.close()
+
+    def run_ai_chatbot(self, chat_history_size=5, system_msg="System: You are a helpful AI assistant."):
+        try:
+            from mysql.ai.genai import MyLLM
+        except Exception as e:
+            print(f"AI模块加载失败: {e}")
+            return
+        my_llm = MyLLM(self.connection)
+        chat_history = deque(maxlen=chat_history_size)
+        while True:
+            user_input = input("\nUser: ")
+            if user_input.lower() in ("exit", "quit"):
+                break
+            history = [system_msg] + list(chat_history) + [f"User: {user_input}"]
+            prompt = "\n".join(history)
+            try:
+                response = my_llm.invoke(prompt)
+            except Exception as err:
+                print(f"AI调用错误: {err}")
+                continue
+            print(f"Bot: {response}")
+            chat_history.append(f"User: {user_input}")
+            chat_history.append(f"Bot: {response}")
+
+# from wei_office_simptool import SQLManager
+# cfg = {
+#     'user': 'root',
+#     'password': '你的密码',
+#     'host': '127.0.0.1',
+#     'database': 'mlcorpus'
+# }
+# db = SQLManager.MySQLDatabase(cfg)
+# db.run_ai_chatbot(chat_history_size=5, system_msg="System: You are a helpful AI assistant.")

wei_data_shu/__init__.py

@@ -0,0 +1,101 @@
+"""Public package exports for `wei-data-shu`."""
+
+from importlib import import_module
+from importlib.metadata import PackageNotFoundError, version
+
+try:
+    __version__ = version("wei-data-shu")
+except PackageNotFoundError:
+    __version__ = "0+unknown"
+
+__all__ = [
+    "SQLManager",
+    "baseColor",
+    "chartsManager",
+    "excelManager",
+    "fileManager",
+    "mailManager",
+    "ollamaManager",
+    "stringManager",
+    "textManager",
+    "timingTool",
+    "database",
+    "excel",
+    "mail",
+    "text",
+    "MySQLDatabase",
+    "mav_colors",
+    "TrendPredictor",
+    "MultipleTrendPredictor",
+    "TextAnalysis",
+    "create_workbook",
+    "ExcelManager",
+    "ExcelHandler",
+    "OpenExcel",
+    "ExcelOperation",
+    "quick_excel",
+    "read_excel_quick",
+    "FileManagement",
+    "DailyEmailReport",
+    "ChatBot",
+    "StringBaba",
+    "DateFormat",
+    "decrypt",
+    "eFormat",
+    "textCombing",
+    "fn_timer",
+    "__version__",
+]
+
+_EXPORTS = {
+    "SQLManager": ("wei_data_shu.SQLManager", None),
+    "baseColor": ("wei_data_shu.baseColor", None),
+    "chartsManager": ("wei_data_shu.chartsManager", None),
+    "excelManager": ("wei_data_shu.excelManager", None),
+    "fileManager": ("wei_data_shu.fileManager", None),
+    "mailManager": ("wei_data_shu.mailManager", None),
+    "ollamaManager": ("wei_data_shu.ollamaManager", None),
+    "stringManager": ("wei_data_shu.stringManager", None),
+    "textManager": ("wei_data_shu.textManager", None),
+    "timingTool": ("wei_data_shu.timingTool", None),
+    "database": ("wei_data_shu.database", None),
+    "excel": ("wei_data_shu.excel", None),
+    "mail": ("wei_data_shu.mail", None),
+    "text": ("wei_data_shu.text", None),
+    "MySQLDatabase": ("wei_data_shu.database", "MySQLDatabase"),
+    "mav_colors": ("wei_data_shu.baseColor", "mav_colors"),
+    "TrendPredictor": ("wei_data_shu.text.forecast", "TrendPredictor"),
+    "MultipleTrendPredictor": ("wei_data_shu.text.forecast", "MultipleTrendPredictor"),
+    "TextAnalysis": ("wei_data_shu.text.analysis", "TextAnalysis"),
+    "create_workbook": ("wei_data_shu.excel", "create_workbook"),
+    "ExcelManager": ("wei_data_shu.excel", "ExcelManager"),
+    "ExcelHandler": ("wei_data_shu.excel", "ExcelHandler"),
+    "OpenExcel": ("wei_data_shu.excel", "OpenExcel"),
+    "ExcelOperation": ("wei_data_shu.excel", "ExcelOperation"),
+    "quick_excel": ("wei_data_shu.excel", "quick_excel"),
+    "read_excel_quick": ("wei_data_shu.excel", "read_excel_quick"),
+    "FileManagement": ("wei_data_shu.fileManager", "FileManagement"),
+    "DailyEmailReport": ("wei_data_shu.mail", "DailyEmailReport"),
+    "ChatBot": ("wei_data_shu.ollamaManager", "ChatBot"),
+    "StringBaba": ("wei_data_shu.text", "StringBaba"),
+    "DateFormat": ("wei_data_shu.text", "DateFormat"),
+    "decrypt": ("wei_data_shu.text", "decrypt"),
+    "eFormat": ("wei_data_shu.text", "eFormat"),
+    "textCombing": ("wei_data_shu.text", "textCombing"),
+    "fn_timer": ("wei_data_shu.timingTool", "fn_timer"),
+}
+
+
+def __getattr__(name: str):
+    target = _EXPORTS.get(name)
+    if target is None:
+        raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+    module_name, attr_name = target
+    module = import_module(module_name)
+    if attr_name is None:
+        return module
+    return getattr(module, attr_name)
+
+
+def __dir__():
+    return sorted(set(globals().keys()) | set(__all__))

wei_data_shu/baseColor.py

@@ -0,0 +1,41 @@
+mav_colors = [
+    '#60ACFC',  # 靛蓝
+    '#32D3EB',  # 天蓝
+    '#5BC49F',  # 薄荷绿
+    '#FEB64D',  # 橙黄
+    '#FF7C7C',  # 桃红
+    '#9287E7',  # 淡紫
+    '#FFDD55',  # 浅黄
+    '#FFAA85',  # 浅橙
+    '#A8E6CF',  # 浅绿
+    '#DCE775',  # 柠檬黄
+    '#FF8A65',  # 珊瑚橙
+    '#9575CD',  # 紫罗兰
+    '#81C784',  # 草绿色
+    '#4DD0E1',  # 青蓝
+    '#BA68C8',  # 浅紫
+    '#7986CB',  # 灰蓝
+    '#4FC3F7',  # 宝石蓝
+    '#F06292',  # 樱花粉
+    '#AED581',  # 青柠绿
+    '#FFD54F',  # 金黄
+    '#FFAB91',  # 桃橙
+    '#FBC02D',  # 金黄
+    '#8D6E63',  # 栗色
+    '#BDBDBD',  # 灰色
+    '#FFCDD2',  # 淡粉
+    '#C5E1A5',  # 浅绿
+    '#80DEEA',  # 浅青
+    '#CE93D8',  # 薰衣草紫
+    '#F48FB1',  # 粉红
+    '#B39DDB',  # 丁香紫
+    '#FF7043',  # 鲑鱼橙
+    '#D4E157',  # 柠檬绿
+    '#FFEB3B',  # 向日葵黄
+    '#9575CD',  # 淡紫
+    '#7986CB',  # 薰衣草蓝
+    '#E57373',  # 玫瑰红
+    '#FFF176',  # 浅黄
+    '#FFB74D',  # 杏橙
+    '#A1887F'   # 浅褐
+]

wei_data_shu/chartsManager.py

@@ -0,0 +1,53 @@
+"""Compatibility layer for legacy `chartsManager` imports.
+
+Deprecated:
+- wei_data_shu.chartsManager.TrendPredictor -> wei_data_shu.text.forecast.TrendPredictor
+- wei_data_shu.chartsManager.MultipleTrendPredictor -> wei_data_shu.text.forecast.MultipleTrendPredictor
+- wei_data_shu.chartsManager.TextAnalysis -> wei_data_shu.text.analysis.TextAnalysis
+"""
+
+import warnings
+
+from .text.analysis import TextAnalysis as _TextAnalysis
+from .text.forecast import MultipleTrendPredictor as _MultipleTrendPredictor
+from .text.forecast import TrendPredictor as _TrendPredictor
+
+warnings.warn(
+    "`wei_data_shu.chartsManager` 已弃用，将在后续版本移除。"
+    "请改用 `wei_data_shu.text.analysis` 或 `wei_data_shu.text.forecast`.",
+    FutureWarning,
+    stacklevel=2,
+)
+
+
+class TrendPredictor(_TrendPredictor):
+    def __init__(self, *args, **kwargs):
+        warnings.warn(
+            "`chartsManager.TrendPredictor` 已弃用，请改用 `wei_data_shu.text.forecast.TrendPredictor`.",
+            FutureWarning,
+            stacklevel=2,
+        )
+        super().__init__(*args, **kwargs)
+
+
+class MultipleTrendPredictor(_MultipleTrendPredictor):
+    def __init__(self, *args, **kwargs):
+        warnings.warn(
+            "`chartsManager.MultipleTrendPredictor` 已弃用，请改用 `wei_data_shu.text.forecast.MultipleTrendPredictor`.",
+            FutureWarning,
+            stacklevel=2,
+        )
+        super().__init__(*args, **kwargs)
+
+
+class TextAnalysis(_TextAnalysis):
+    def __init__(self, *args, **kwargs):
+        warnings.warn(
+            "`chartsManager.TextAnalysis` 已弃用，请改用 `wei_data_shu.text.analysis.TextAnalysis`.",
+            FutureWarning,
+            stacklevel=2,
+        )
+        super().__init__(*args, **kwargs)
+
+
+__all__ = ["TrendPredictor", "MultipleTrendPredictor", "TextAnalysis"]

wei_data_shu/database/__init__.py

@@ -0,0 +1,5 @@
+"""Database domain exports."""
+
+from ..SQLManager import MySQLDatabase
+
+__all__ = ["MySQLDatabase"]

wei_data_shu/docManager.py

@@ -0,0 +1,2 @@
+from .fileManager import FileManagement
+from .excelManager import ExcelHandler, OpenExcel, ExcelOperation, eExcel

wei_data_shu/excel/__init__.py

@@ -0,0 +1,13 @@
+"""Excel domain exports."""
+
+from ..excelManager import ExcelHandler, ExcelManager, ExcelOperation, OpenExcel, create_workbook, quick_excel, read_excel_quick
+
+__all__ = [
+    "create_workbook",
+    "ExcelManager",
+    "ExcelHandler",
+    "OpenExcel",
+    "ExcelOperation",
+    "quick_excel",
+    "read_excel_quick",
+]