staran 0.6.1__py3-none-any.whl → 1.0.1__py3-none-any.whl

staran/tools/tests/test_logging.py

@@ -0,0 +1,402 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+"""
+日志系统测试
+==========
+
+测试Date类的日志功能，包括：
+- 日志级别设置
+- 日志消息记录
+- 日志输出控制
+- 性能日志
+"""
+
+import unittest
+import sys
+import os
+import logging
+import io
+from contextlib import redirect_stderr
+
+# 添加项目根目录到路径
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..', '..'))
+
+from staran.tools.date import Date
+
+
+class TestLoggingSystem(unittest.TestCase):
+    """测试日志系统基本功能"""
+    
+    def setUp(self):
+        """设置测试环境"""
+        # 重置日志级别
+        Date.set_log_level(logging.DEBUG)
+        
+        # 创建字符串流来捕获日志输出
+        self.log_stream = io.StringIO()
+        
+        # 获取Date的日志记录器
+        self.logger = logging.getLogger('staran.tools.date')
+        
+        # 清除现有的处理器
+        for handler in self.logger.handlers[:]:
+            self.logger.removeHandler(handler)
+        
+        # 添加流处理器
+        self.handler = logging.StreamHandler(self.log_stream)
+        self.handler.setLevel(logging.DEBUG)
+        formatter = logging.Formatter('%(levelname)s:%(name)s:%(message)s')
+        self.handler.setFormatter(formatter)
+        self.logger.addHandler(self.handler)
+        self.logger.setLevel(logging.DEBUG)
+    
+    def tearDown(self):
+        """清理测试环境"""
+        # 移除处理器
+        self.logger.removeHandler(self.handler)
+        self.handler.close()
+        
+        # 重置日志级别为ERROR，避免干扰其他测试
+        Date.set_log_level(logging.ERROR)
+    
+    def get_log_output(self):
+        """获取日志输出"""
+        return self.log_stream.getvalue()
+    
+    def test_log_level_setting(self):
+        """测试日志级别设置"""
+        # 测试设置不同的日志级别
+        Date.set_log_level(logging.INFO)
+        Date.set_log_level(logging.WARNING)
+        Date.set_log_level(logging.ERROR)
+        Date.set_log_level(logging.DEBUG)
+        
+        # 验证级别设置成功（通过创建对象来触发日志）
+        self.log_stream.truncate(0)
+        self.log_stream.seek(0)
+        
+        Date('20250415')
+        log_output = self.get_log_output()
+        
+        # 应该有DEBUG级别的日志
+        self.assertIn('DEBUG', log_output)
+    
+    def test_object_creation_logging(self):
+        """测试对象创建日志"""
+        self.log_stream.truncate(0)
+        self.log_stream.seek(0)
+        
+        # 创建不同类型的Date对象
+        Date('20250415')
+        Date('202504')
+        Date('2025')
+        Date(2025, 4, 15)
+        
+        log_output = self.get_log_output()
+        
+        # 验证有创建日志
+        self.assertIn('创建Date对象', log_output)
+        self.assertIn('20250415', log_output)
+    
+    def test_method_call_logging(self):
+        """测试方法调用日志"""
+        self.log_stream.truncate(0)
+        self.log_stream.seek(0)
+        
+        date = Date('20250415')
+        
+        # 调用各种方法
+        date.format_iso()
+        date.add_days(10)
+        date.get_weekday()
+        date.is_weekend()
+        
+        log_output = self.get_log_output()
+        
+        # 验证有方法调用日志
+        self.assertIn('调用方法', log_output)
+    
+    def test_error_logging(self):
+        """测试错误日志"""
+        self.log_stream.truncate(0)
+        self.log_stream.seek(0)
+        
+        # 触发错误
+        try:
+            Date('invalid_date')
+        except ValueError:
+            pass
+        
+        log_output = self.get_log_output()
+        
+        # 验证有错误日志
+        self.assertIn('ERROR', log_output)
+    
+    def test_performance_logging(self):
+        """测试性能日志"""
+        self.log_stream.truncate(0)
+        self.log_stream.seek(0)
+        
+        date = Date('20250415')
+        
+        # 执行一些计算密集的操作
+        for i in range(10):
+            date.add_days(i)
+        
+        log_output = self.get_log_output()
+        
+        # 性能日志可能不会每次都出现，但应该有调用日志
+        self.assertIn('调用方法', log_output)
+
+
+class TestLoggingConfiguration(unittest.TestCase):
+    """测试日志配置"""
+    
+    def setUp(self):
+        """设置测试环境"""
+        Date.set_log_level(logging.ERROR)  # 默认高级别
+    
+    def test_log_level_effects(self):
+        """测试日志级别效果"""
+        # 创建测试流
+        log_stream = io.StringIO()
+        logger = logging.getLogger('staran.tools.date')
+        
+        # 清除现有处理器
+        for handler in logger.handlers[:]:
+            logger.removeHandler(handler)
+        
+        # 添加新处理器
+        handler = logging.StreamHandler(log_stream)
+        handler.setLevel(logging.DEBUG)
+        formatter = logging.Formatter('%(levelname)s:%(message)s')
+        handler.setFormatter(formatter)
+        logger.addHandler(handler)
+        
+        try:
+            # 测试ERROR级别
+            Date.set_log_level(logging.ERROR)
+            logger.setLevel(logging.ERROR)
+            
+            log_stream.truncate(0)
+            log_stream.seek(0)
+            
+            Date('20250415')
+            error_output = log_stream.getvalue()
+            
+            # 测试DEBUG级别
+            Date.set_log_level(logging.DEBUG)
+            logger.setLevel(logging.DEBUG)
+            
+            log_stream.truncate(0)
+            log_stream.seek(0)
+            
+            Date('20250415')
+            debug_output = log_stream.getvalue()
+            
+            # DEBUG级别应该有更多输出
+            self.assertGreaterEqual(len(debug_output), len(error_output))
+            
+        finally:
+            logger.removeHandler(handler)
+            handler.close()
+    
+    def test_log_message_format(self):
+        """测试日志消息格式"""
+        log_stream = io.StringIO()
+        logger = logging.getLogger('staran.tools.date')
+        
+        # 清除现有处理器
+        for handler in logger.handlers[:]:
+            logger.removeHandler(handler)
+        
+        handler = logging.StreamHandler(log_stream)
+        handler.setLevel(logging.DEBUG)
+        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+        handler.setFormatter(formatter)
+        logger.addHandler(handler)
+        logger.setLevel(logging.DEBUG)
+        
+        try:
+            Date.set_log_level(logging.DEBUG)
+            
+            Date('20250415')
+            output = log_stream.getvalue()
+            
+            # 验证日志格式包含时间戳、名称、级别、消息
+            if output:  # 如果有输出
+                lines = output.strip().split('\n')
+                for line in lines:
+                    parts = line.split(' - ')
+                    if len(parts) >= 4:
+                        self.assertIn('staran.tools.date', parts[1])
+                        self.assertIn(parts[2], ['DEBUG', 'INFO', 'WARNING', 'ERROR'])
+        
+        finally:
+            logger.removeHandler(handler)
+            handler.close()
+
+
+class TestLoggingIntegration(unittest.TestCase):
+    """测试日志集成"""
+    
+    def setUp(self):
+        """设置测试环境"""
+        Date.set_log_level(logging.WARNING)
+    
+    def test_logging_with_various_operations(self):
+        """测试各种操作的日志集成"""
+        log_stream = io.StringIO()
+        logger = logging.getLogger('staran.tools.date')
+        
+        # 清除现有处理器
+        for handler in logger.handlers[:]:
+            logger.removeHandler(handler)
+        
+        handler = logging.StreamHandler(log_stream)
+        handler.setLevel(logging.DEBUG)
+        formatter = logging.Formatter('%(levelname)s:%(funcName)s:%(message)s')
+        handler.setFormatter(formatter)
+        logger.addHandler(handler)
+        logger.setLevel(logging.DEBUG)
+        
+        try:
+            Date.set_log_level(logging.DEBUG)
+            
+            # 执行各种操作
+            date = Date('20250415')
+            date.format_iso()
+            date.add_days(10)
+            date.get_weekday()
+            result = date.is_weekend()
+            
+            output = log_stream.getvalue()
+            
+            # 验证不同操作都有日志记录
+            if output:
+                self.assertIsInstance(output, str)
+                # 基本验证日志不为空
+                self.assertGreater(len(output.strip()), 0)
+        
+        finally:
+            logger.removeHandler(handler)
+            handler.close()
+    
+    def test_logging_performance_impact(self):
+        """测试日志对性能的影响"""
+        import time
+        
+        # 测试无日志性能
+        Date.set_log_level(logging.CRITICAL)
+        
+        start_time = time.time()
+        for i in range(100):
+            date = Date('20250415')
+            date.add_days(i % 10)
+        no_log_time = time.time() - start_time
+        
+        # 测试有日志性能
+        Date.set_log_level(logging.DEBUG)
+        
+        start_time = time.time()
+        for i in range(100):
+            date = Date('20250415')
+            date.add_days(i % 10)
+        with_log_time = time.time() - start_time
+        
+        # 日志不应该显著影响性能（这里允许10倍的性能差异）
+        self.assertLess(with_log_time, no_log_time * 10)
+        
+        # 重置日志级别
+        Date.set_log_level(logging.ERROR)
+
+
+class TestLoggingEdgeCases(unittest.TestCase):
+    """测试日志边缘情况"""
+    
+    def setUp(self):
+        """设置测试环境"""
+        Date.set_log_level(logging.ERROR)
+    
+    def test_logging_with_invalid_operations(self):
+        """测试无效操作的日志"""
+        log_stream = io.StringIO()
+        logger = logging.getLogger('staran.tools.date')
+        
+        # 清除现有处理器
+        for handler in logger.handlers[:]:
+            logger.removeHandler(handler)
+        
+        handler = logging.StreamHandler(log_stream)
+        handler.setLevel(logging.DEBUG)
+        formatter = logging.Formatter('%(levelname)s:%(message)s')
+        handler.setFormatter(formatter)
+        logger.addHandler(handler)
+        logger.setLevel(logging.DEBUG)
+        
+        try:
+            Date.set_log_level(logging.DEBUG)
+            
+            # 尝试无效操作
+            try:
+                Date('invalid')
+            except ValueError:
+                pass
+            
+            try:
+                Date(2025, 13, 1)  # 无效月份
+            except ValueError:
+                pass
+            
+            output = log_stream.getvalue()
+            
+            # 应该有错误日志
+            if output:
+                self.assertIn('ERROR', output)
+        
+        finally:
+            logger.removeHandler(handler)
+            handler.close()
+    
+    def test_concurrent_logging(self):
+        """测试并发日志"""
+        import threading
+        import time
+        
+        Date.set_log_level(logging.DEBUG)
+        
+        results = []
+        
+        def create_dates():
+            for i in range(10):
+                try:
+                    date = Date(f'2025041{i % 10}')
+                    results.append(str(date))
+                except:
+                    pass
+        
+        # 创建多个线程
+        threads = []
+        for i in range(5):
+            thread = threading.Thread(target=create_dates)
+            threads.append(thread)
+        
+        # 启动所有线程
+        for thread in threads:
+            thread.start()
+        
+        # 等待所有线程完成
+        for thread in threads:
+            thread.join()
+        
+        # 验证结果
+        self.assertGreater(len(results), 0)
+        
+        # 重置日志级别
+        Date.set_log_level(logging.ERROR)
+
+
+if __name__ == '__main__':
+    # 运行所有日志测试
+    unittest.main(verbosity=2)

staran-1.0.1.dist-info/METADATA

@@ -0,0 +1,37 @@
+Metadata-Version: 2.4
+Name: staran
+Version: 1.0.1
+Summary: staran - 轻量级Python日期工具库
+Home-page: https://github.com/starlxa/staran
+Author: StarAn
+Author-email: starlxa@icloud.com
+License: MIT
+Project-URL: Bug Reports, https://github.com/starlxa/staran/issues
+Project-URL: Source, https://github.com/starlxa/staran
+Keywords: date datetime utilities time-processing
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Topic :: Utilities
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.7
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: requires-python
+Dynamic: summary

staran-1.0.1.dist-info/RECORD

@@ -0,0 +1,13 @@
+staran/__init__.py,sha256=KBMMJN15kcGQLCOVUqfGf7UnuJs3_HFOXZq7SLlSF_U,208
+staran/tools/__init__.py,sha256=lSEZeU6pp9smeiuVS3eYci203IFYBode0vUU5GeYb8w,1061
+staran/tools/date.py,sha256=zZz9W94jed_KYgRnnPTaG8zyDD99UJdqwa0A5C_EyOw,14208
+staran/tools/tests/__init__.py,sha256=s8WYc5XhtK7saVOeYyY85lmeLNoB8qpfbxZYSVEMfxs,3068
+staran/tools/tests/run_tests.py,sha256=i-IjimWMOLDD4XZY-Tr1GNXV6yX01-7TjKiYKjB186A,7904
+staran/tools/tests/test_api_compatibility.py,sha256=wmmM6f7nrS9J-b70xIltmtaIbiLALYCmaCsV4_wTxbE,11942
+staran/tools/tests/test_date.py,sha256=8vVe0QyRSfPqvGK0WBhmOAs03MGjDlnw-fxYV5Wjv6Q,18429
+staran/tools/tests/test_logging.py,sha256=VdoD8PJoKob01yQ93asQHz9Zo69jNfyq6jKtNSEi8UM,12041
+staran-1.0.1.dist-info/licenses/LICENSE,sha256=2EmsBIyDCono4iVXNpv5_px9qt2b7hfPq1WuyGVMNP4,1361
+staran-1.0.1.dist-info/METADATA,sha256=gx3O-5bO2L3hdJ1JgVDJuOwIh2X9VC_xL58g_pIbmfc,1295
+staran-1.0.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+staran-1.0.1.dist-info/top_level.txt,sha256=NOUZtXSh5oSIEjHrC0lQ9WmoKtD010Q00dghWyag-Zs,7
+staran-1.0.1.dist-info/RECORD,,

staran/banks/__init__.py

@@ -1,30 +0,0 @@
-"""
-staran.banks - 银行配置模块
-
-该模块包含不同银行的特定配置，包括：
-- 数据库连接配置
-- 表结构定义
-- 业务规则设置
-- 模型配置
-
-支持的银行：
-- xinjiang_icbc: 新疆工行配置
-
-版本: 0.6.0
-"""
-
-from .xinjiang_icbc import (
-    XinjiangICBCConfig,
-    get_xinjiang_icbc_tables,
-    get_xinjiang_icbc_models,
-    xinjiang_icbc_config
-)
-
-__all__ = [
-    'XinjiangICBCConfig',
-    'xinjiang_icbc_config',
-    'get_xinjiang_icbc_tables', 
-    'get_xinjiang_icbc_models'
-]
-
-__version__ = "0.6.0"

staran/banks/xinjiang_icbc/__init__.py

@@ -1,90 +0,0 @@
-"""
-新疆工行银行配置模块
-
-专门针对新疆工行代发长尾客户的配置：
-- 数据库表结构定义（代发长尾客户专用）
-- 业务规则配置
-- 模型配置（提升模型和防流失模型）
-
-数据库: xinjiang_icbc_daifa_longtail
-业务范围: 代发长尾客户
-"""
-
-from dataclasses import dataclass
-from typing import Dict, List, Optional
-from datetime import datetime
-
-
-@dataclass 
-class XinjiangICBCConfig:
-    """新疆工行配置类"""
-    
-    # 数据库配置
-    database_name: str = "xinjiang_icbc_daifa_longtail"
-    schema_name: str = "daifa_longtail"
-    
-    # 业务配置
-    business_domain: str = "代发长尾客户"
-    customer_segment: str = "代发长尾"
-    
-    # 模型配置
-    available_models: List[str] = None
-    
-    # 业务规则
-    longtail_asset_min: float = 10000  # 长尾客户最小资产
-    longtail_asset_max: float = 100000  # 长尾客户最大资产
-    upgrade_target: float = 3000  # 提升目标金额
-    churn_threshold: float = 1500  # 流失阈值金额
-    
-    def __post_init__(self):
-        if self.available_models is None:
-            self.available_models = [
-                "daifa_longtail_upgrade_3k",    # 代发长尾提升3k模型
-                "daifa_longtail_churn_1_5k"     # 代发长尾防流失1.5k模型
-            ]
-
-
-def get_xinjiang_icbc_tables() -> Dict[str, str]:
-    """获取新疆工行代发长尾客户表配置"""
-    return {
-        # 代发长尾客户行为表
-        "daifa_longtail_behavior": "xinjiang_icbc_daifa_hlwj_dfcw_f1_f4_wy",
-        
-        # 代发长尾客户资产平均表  
-        "daifa_longtail_asset_avg": "xinjiang_icbc_daifa_hlwj_zi_chan_avg_wy",
-        
-        # 代发长尾客户资产配置表
-        "daifa_longtail_asset_config": "xinjiang_icbc_daifa_hlwj_zi_chan_config_wy",
-        
-        # 代发长尾客户月度统计表
-        "daifa_longtail_monthly_stat": "xinjiang_icbc_daifa_hlwj_monthly_stat_wy"
-    }
-
-
-def get_xinjiang_icbc_models() -> Dict[str, Dict]:
-    """获取新疆工行代发长尾客户模型配置"""
-    return {
-        "daifa_longtail_upgrade_3k": {
-            "name": "代发长尾客户提升3k预测模型",
-            "description": "预测下个月代发长尾客户资产提升3000元的概率",
-            "target": "upgrade_3k_next_month",
-            "model_type": "binary_classification",
-            "business_objective": "识别有潜力提升资产的代发长尾客户",
-            "target_threshold": 3000,
-            "prediction_window": "1_month"
-        },
-        
-        "daifa_longtail_churn_1_5k": {
-            "name": "代发长尾客户防流失1.5k预测模型", 
-            "description": "预测下个月代发长尾客户流失1500元资产的风险",
-            "target": "churn_1_5k_next_month",
-            "model_type": "binary_classification", 
-            "business_objective": "识别有流失风险的代发长尾客户",
-            "target_threshold": 1500,
-            "prediction_window": "1_month"
-        }
-    }
-
-
-# 创建默认配置实例
-xinjiang_icbc_config = XinjiangICBCConfig()

staran/engines/__init__.py

@@ -1,65 +0,0 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-
-"""
-数据库引擎模块
-提供统一的数据库引擎接口
-"""
-
-# 基础组件
-from .base import BaseEngine, DatabaseType
-
-# 具体引擎实现
-from .spark import SparkEngine
-from .hive import HiveEngine
-
-# 图灵平台引擎 (可选导入)
-try:
-    from .turing import TuringEngine, create_turing_engine
-    _TURING_AVAILABLE = True
-except ImportError:
-    TuringEngine = None
-    create_turing_engine = None
-    _TURING_AVAILABLE = False
-
-# 便捷创建函数
-def create_engine(engine_type: str, database_name: str, **kwargs) -> BaseEngine:
-    """
-    创建数据库引擎的便捷函数
-    
-    Args:
-        engine_type: 引擎类型 ('spark', 'hive', 'turing')
-        database_name: 数据库名称
-        **kwargs: 其他参数
-        
-    Returns:
-        数据库引擎实例
-    """
-    engine_type = engine_type.lower()
-    
-    if engine_type == 'spark':
-        return SparkEngine(database_name, **kwargs)
-    elif engine_type == 'hive':
-        return HiveEngine(database_name, **kwargs)
-    elif engine_type == 'turing':
-        if not _TURING_AVAILABLE:
-            raise ImportError("TuringEngine不可用，请确保turingPythonLib已安装")
-        return TuringEngine(database_name, **kwargs)
-    else:
-        raise ValueError(f"不支持的引擎类型: {engine_type}")
-
-# 主要导出
-__all__ = [
-    'BaseEngine',
-    'DatabaseType',
-    'SparkEngine',
-    'HiveEngine',
-    'create_engine'
-]
-
-# 如果图灵引擎可用，添加到导出
-if _TURING_AVAILABLE:
-    __all__.extend([
-        'TuringEngine',
-        'create_turing_engine'
-    ])