terryutils 1.0.4__py3-none-any.whl → 1.0.6__py3-none-any.whl

terryutils/__init__.py

@@ -0,0 +1,3 @@
+from .mysql_util import MysqlUtil
+from .log import setup_logger
+from .date_util import  DateUtils

terryutils/date_util.py

@@ -0,0 +1,228 @@
+from datetime import datetime, date, timedelta
+import calendar
+import re
+
+class DateUtils:
+    """企业级 Python 日期工具类"""
+
+    COMMON_FORMATS = [
+        "%Y-%m-%d %H:%M:%S",
+        "%Y-%m-%d %H:%M",
+        "%Y-%m-%d",
+        "%d/%m/%Y",
+        "%d-%m-%Y",
+        "%Y/%m/%d",
+    ]
+
+    # -------------------- 基础获取 --------------------
+    @staticmethod
+    def now(fmt="%Y-%m-%d %H:%M:%S"):
+        """当前时间"""
+        return datetime.now().strftime(fmt)
+
+    @staticmethod
+    def today(fmt="%Y-%m-%d"):
+        """今天"""
+        return date.today().strftime(fmt)
+
+    @staticmethod
+    def datetime_now():
+        """返回datetime对象"""
+        return datetime.now()
+
+    @staticmethod
+    def date_today():
+        """返回date对象"""
+        return date.today()
+
+    # -------------------- 字符串与datetime互转 --------------------
+    @staticmethod
+    def parse(date_str, fmt=None):
+        """尝试多种格式解析字符串"""
+        if isinstance(date_str, (int, float)):
+            return datetime.fromtimestamp(date_str)
+        if isinstance(date_str, datetime):
+            return date_str
+        if fmt:
+            return datetime.strptime(date_str, fmt)
+        for f in DateUtils.COMMON_FORMATS:
+            try:
+                return datetime.strptime(date_str, f)
+            except:
+                continue
+        raise ValueError(f"无法解析日期: {date_str}")
+
+    @staticmethod
+    def format(dt, fmt="%Y-%m-%d %H:%M:%S"):
+        if isinstance(dt, str):
+            dt = DateUtils.parse(dt)
+        return dt.strftime(fmt)
+
+    @staticmethod
+    def to_timestamp(dt):
+        dt = DateUtils.parse(dt) if isinstance(dt, (str, int, float)) else dt
+        return int(dt.timestamp())
+
+    @staticmethod
+    def from_timestamp(ts):
+        return datetime.fromtimestamp(ts)
+
+    # -------------------- 日期加减 --------------------
+    @staticmethod
+    def add_days(dt, days):
+        dt = DateUtils.parse(dt)
+        return dt + timedelta(days=days)
+
+    @staticmethod
+    def add_seconds(dt, seconds, ft):
+        dt = DateUtils.parse(dt, ft)
+        return dt + timedelta(seconds=seconds)
+
+    @staticmethod
+    def add_months(dt, months):
+        dt = DateUtils.parse(dt)
+        month = dt.month - 1 + months
+        year = dt.year + month // 12
+        month = month % 12 + 1
+        day = min(dt.day, calendar.monthrange(year, month)[1])
+        return dt.replace(year=year, month=month, day=day)
+
+    @staticmethod
+    def add_years(dt, years):
+        dt = DateUtils.parse(dt)
+        try:
+            return dt.replace(year=dt.year + years)
+        except ValueError:
+            # 闰年2月29日处理
+            return dt.replace(month=2, day=28, year=dt.year + years)
+
+    # -------------------- 周/月/季度起止 --------------------
+    @staticmethod
+    def start_of_day(dt=None):
+        dt = DateUtils.parse(dt) if dt else datetime.now()
+        return datetime(dt.year, dt.month, dt.day, 0, 0, 0)
+
+    @staticmethod
+    def end_of_day(dt=None):
+        dt = DateUtils.parse(dt) if dt else datetime.now()
+        return datetime(dt.year, dt.month, dt.day, 23, 59, 59)
+
+    @staticmethod
+    def start_of_week(dt=None):
+        dt = DateUtils.parse(dt) if dt else datetime.now()
+        start = dt - timedelta(days=dt.weekday())
+        return datetime(start.year, start.month, start.day)
+
+    @staticmethod
+    def end_of_week(dt=None):
+        dt = DateUtils.parse(dt) if dt else datetime.now()
+        end = dt + timedelta(days=6 - dt.weekday())
+        return datetime(end.year, end.month, end.day, 23, 59, 59)
+
+    @staticmethod
+    def start_of_month(dt=None):
+        dt = DateUtils.parse(dt) if dt else datetime.now()
+        return datetime(dt.year, dt.month, 1)
+
+    @staticmethod
+    def end_of_month(dt=None):
+        dt = DateUtils.parse(dt) if dt else datetime.now()
+        last_day = calendar.monthrange(dt.year, dt.month)[1]
+        return datetime(dt.year, dt.month, last_day, 23, 59, 59)
+
+    @staticmethod
+    def start_of_quarter(dt=None):
+        dt = DateUtils.parse(dt) if dt else datetime.now()
+        month = ((dt.month - 1) // 3) * 3 + 1
+        return datetime(dt.year, month, 1)
+
+    @staticmethod
+    def end_of_quarter(dt=None):
+        dt = DateUtils.parse(dt) if dt else datetime.now()
+        month = ((dt.month - 1) // 3 + 1) * 3
+        last_day = calendar.monthrange(dt.year, month)[1]
+        return datetime(dt.year, month, last_day, 23, 59, 59)
+
+    # -------------------- 日期判断 --------------------
+    @staticmethod
+    def is_leap_year(year):
+        return calendar.isleap(year)
+
+    @staticmethod
+    def weekday(dt=None):
+        dt = DateUtils.parse(dt) if dt else datetime.now()
+        return dt.weekday()  # 0=周一
+
+    @staticmethod
+    def is_weekend(dt=None):
+        return DateUtils.weekday(dt) >= 5
+
+    @staticmethod
+    def days_between(start, end):
+        start = DateUtils.parse(start)
+        end = DateUtils.parse(end)
+        return (end - start).days
+
+    # -------------------- 日期区间 --------------------
+    @staticmethod
+    def date_range(start, end, step=1):
+        start = DateUtils.parse(start)
+        end = DateUtils.parse(end)
+        days = []
+        delta = timedelta(days=step)
+        current = start
+        while current <= end:
+            days.append(current)
+            current += delta
+        return days
+
+    # -------------------- 自然语言解析 --------------------
+    @staticmethod
+    def smart_parse(text):
+        """支持 '今天','明天','昨天','上周一','下个月15号'"""
+        today = datetime.now()
+        text = text.strip()
+        if text == "今天":
+            return today
+        elif text == "明天":
+            return today + timedelta(days=1)
+        elif text == "昨天":
+            return today - timedelta(days=1)
+        m = re.match(r"上周([一二三四五六日])", text)
+        if m:
+            weekday_map = {"一":0,"二":1,"三":2,"四":3,"五":4,"六":5,"日":6}
+            target_weekday = weekday_map[m.group(1)]
+            delta_days = today.weekday() - target_weekday + 7
+            return today - timedelta(days=delta_days)
+        # TODO: 可继续扩展自然语言解析
+        return DateUtils.parse(text)
+
+    @staticmethod
+    def add_days_str(date_str, days, fmt="%Y-%m-%d"):
+        """直接对字符串日期加减天数，返回字符串"""
+        dt = datetime.strptime(date_str, fmt)
+        dt += timedelta(days=days)
+        return dt.strftime(fmt)
+
+    @staticmethod
+    def format_day(date_str, fmt="%Y-%m-%d"):
+        """直接对字符串日期加减天数，返回字符串"""
+        dt = datetime.strptime(date_str, fmt)
+        return dt.strftime("%Y年%m月%d日")
+
+    @staticmethod
+    def to_timestamp(dt):
+        dt = DateUtils.parse(dt) if isinstance(dt, (str, int, float)) else dt
+        return int(dt.timestamp())
+# ------------------------ 测试 ------------------------
+if __name__ == "__main__":
+    print("现在时间:", DateUtils.now())
+    print("今天:", DateUtils.today())
+    print("月初:", DateUtils.start_of_month())
+    print("月末:", DateUtils.end_of_month())
+    print("季度开始:", DateUtils.start_of_quarter())
+    print("季度结束:", DateUtils.end_of_quarter())
+    print("加3天:", DateUtils.add_days(datetime.now(), 3))
+    print("是否周末:", DateUtils.is_weekend())
+    print("自然语言解析 明天:", DateUtils.smart_parse("明天"))
+    print("日期区间:", [d.strftime("%Y-%m-%d") for d in DateUtils.date_range("2025-12-01", "2025-12-05")])

terryutils/log.py

@@ -0,0 +1,32 @@
+import logging
+import os
+from logging.handlers import TimedRotatingFileHandler
+
+# 配置日志
+def setup_logger(app_name="MyApp", log_dir="./logs"):
+    os.makedirs(log_dir, exist_ok=True)
+    log_filename = os.path.join(log_dir, f'{app_name}.log')  # 固定名字，滚动文件自动加日期
+    logger = logging.getLogger(app_name)
+    logger.setLevel(logging.DEBUG)
+
+    # 每天生成一个新文件，保留30天
+    file_handler = TimedRotatingFileHandler(
+        log_filename, when='midnight', interval=1, backupCount=30, encoding='utf-8'
+    )
+    file_handler.setLevel(logging.DEBUG)
+
+    console_handler = logging.StreamHandler()
+    console_handler.setLevel(logging.INFO)
+
+    formatter = logging.Formatter(
+        '%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+        datefmt='%Y-%m-%d %H:%M:%S'
+    )
+    file_handler.setFormatter(formatter)
+    console_handler.setFormatter(formatter)
+
+    logger.handlers = []
+    logger.addHandler(file_handler)
+    logger.addHandler(console_handler)
+
+    return logger

terryutils/mysql_util.py

@@ -0,0 +1,108 @@
+import pymysql
+import traceback
+
+import pymysql
+from pymysql.cursors import DictCursor
+
+class MysqlUtil:
+    def __init__(self, host, user, password, database, port=3306, dict_result=False, debug=True):
+        """
+        初始化数据库连接
+        :param dict_result: 是否返回 dict
+        :param debug: 是否打印 SQL 调试信息
+        """
+        self.conn_params = {
+            "host": host,
+            "user": user,
+            "password": password,
+            "database": database,
+            "port": port,
+            "charset": "utf8mb4",
+            "autocommit": False,  # 手动事务管理
+        }
+        if dict_result:
+            self.conn_params["cursorclass"] = DictCursor
+
+        self.conn = None
+        self.debug = debug
+        self._connect()
+
+    def _connect(self):
+        """建立连接"""
+        self.conn = pymysql.connect(**self.conn_params)
+
+    def _ensure_connection(self):
+        """确保连接可用，失效时自动重连"""
+        try:
+            self.conn.ping(reconnect=True)
+        except Exception:
+            self._connect()
+
+    def _execute(self, func, sql=None, params=None):
+        """
+        通用执行器，保证 commit/rollback
+        """
+        self._ensure_connection()
+        try:
+            if self.debug and sql:
+                # print(f"[DEBUG SQL] {sql} | params={params}")
+                pass
+            result = func()
+            self.conn.commit()
+            return result
+        except Exception:
+            traceback.print_exc()
+            self.conn.rollback()
+            raise
+
+    def execute(self, sql, params=None):
+        """执行单条增删改"""
+        def run():
+            with self.conn.cursor() as cursor:
+                return cursor.execute(sql, params or ())
+        return self._execute(run, sql, params)
+
+    def execute_by_sqls(self, sqls):
+        """执行多条 SQL"""
+        def run():
+            with self.conn.cursor() as cursor:
+                for sql in sqls:
+                    if self.debug:
+                        print(f"[DEBUG SQL] {sql}")
+                    cursor.execute(sql)
+        return self._execute(run)
+
+    def executemany(self, sql, param_list):
+        """批量执行"""
+        def run():
+            with self.conn.cursor() as cursor:
+                return cursor.executemany(sql, param_list)
+        return self._execute(run, sql, param_list)
+
+    def fetchone(self, sql, params=None):
+        """查询一条"""
+        def run():
+            with self.conn.cursor() as cursor:
+                cursor.execute(sql, params or ())
+                return cursor.fetchone()
+        return self._execute(run, sql, params)
+
+    def fetchall(self, sql, params=None):
+        """查询多条"""
+        def run():
+            with self.conn.cursor() as cursor:
+                cursor.execute(sql, params or ())
+                return cursor.fetchall()
+        return self._execute(run, sql, params)
+
+    def close(self):
+        """关闭连接"""
+        if self.conn:
+            self.conn.close()
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()
+

{terryutils-1.0.4.dist-info → terryutils-1.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: terryutils
-Version: 1.0.4
+Version: 1.0.6
 Summary: My personal python toolkit
 Author-email: Terry Zhu <596480606@qq.com>
 License: MIT

terryutils-1.0.6.dist-info/RECORD

@@ -0,0 +1,8 @@
+terryutils/__init__.py,sha256=rDDyXE01Cmz90yMbiDhWa9AOdsB7JX-KqSmYBYt9PAY,99
+terryutils/date_util.py,sha256=KdsTOd8EjKu7p9gbfBjm0Yd06zokM8WDAD2NfnogPu0,7832
+terryutils/log.py,sha256=ul9DF7-6AwjH-vE7aPB96hFM0OKrYUjjUFmd4xP4678,1063
+terryutils/mysql_util.py,sha256=UwnF8z-KNJlMuOnH03iHVPKkpSNj073YSsrvI3x8pkM,3292
+terryutils-1.0.6.dist-info/METADATA,sha256=4V80JUySGSBCEisXF1jeiBGelCvLGPz7tJ4cre5CpZk,344
+terryutils-1.0.6.dist-info/WHEEL,sha256=WnJ8fYhv8N4SYVK2lLYNI6N0kVATA7b0piVUNvqIIJE,91
+terryutils-1.0.6.dist-info/top_level.txt,sha256=LQtzsZsXViaVLHb_rJ4SJiT9ScUfK6dUoy89dP1mLIo,11
+terryutils-1.0.6.dist-info/RECORD,,

terryutils-1.0.6.dist-info/top_level.txt

@@ -0,0 +1 @@
+terryutils

terryutils-1.0.4.dist-info/RECORD

@@ -1,4 +0,0 @@
-terryutils-1.0.4.dist-info/METADATA,sha256=NfNgRDQzWtg4cW8azjgYPDNTVftt1SklFn6P2KYPe7g,344
-terryutils-1.0.4.dist-info/WHEEL,sha256=WnJ8fYhv8N4SYVK2lLYNI6N0kVATA7b0piVUNvqIIJE,91
-terryutils-1.0.4.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-terryutils-1.0.4.dist-info/RECORD,,

terryutils-1.0.4.dist-info/top_level.txt

@@ -1 +0,0 @@
-