crawlo 1.3.2__py3-none-any.whl → 1.3.4__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of crawlo might be problematic. Click here for more details.

Files changed (105) hide show
  1. crawlo/__init__.py +24 -0
  2. crawlo/__version__.py +1 -1
  3. crawlo/commands/run.py +58 -32
  4. crawlo/core/__init__.py +44 -0
  5. crawlo/core/engine.py +119 -45
  6. crawlo/core/scheduler.py +4 -3
  7. crawlo/crawler.py +603 -1133
  8. crawlo/downloader/aiohttp_downloader.py +4 -2
  9. crawlo/extension/__init__.py +1 -1
  10. crawlo/extension/logging_extension.py +23 -7
  11. crawlo/factories/__init__.py +28 -0
  12. crawlo/factories/base.py +69 -0
  13. crawlo/factories/crawler.py +104 -0
  14. crawlo/factories/registry.py +85 -0
  15. crawlo/filters/aioredis_filter.py +25 -2
  16. crawlo/framework.py +292 -0
  17. crawlo/initialization/__init__.py +40 -0
  18. crawlo/initialization/built_in.py +426 -0
  19. crawlo/initialization/context.py +142 -0
  20. crawlo/initialization/core.py +194 -0
  21. crawlo/initialization/phases.py +149 -0
  22. crawlo/initialization/registry.py +146 -0
  23. crawlo/items/base.py +2 -1
  24. crawlo/logging/__init__.py +38 -0
  25. crawlo/logging/config.py +97 -0
  26. crawlo/logging/factory.py +129 -0
  27. crawlo/logging/manager.py +112 -0
  28. crawlo/middleware/middleware_manager.py +1 -1
  29. crawlo/middleware/offsite.py +1 -1
  30. crawlo/mode_manager.py +26 -1
  31. crawlo/pipelines/pipeline_manager.py +2 -1
  32. crawlo/project.py +76 -46
  33. crawlo/queue/pqueue.py +11 -5
  34. crawlo/queue/queue_manager.py +143 -19
  35. crawlo/queue/redis_priority_queue.py +69 -49
  36. crawlo/settings/default_settings.py +110 -14
  37. crawlo/settings/setting_manager.py +29 -13
  38. crawlo/spider/__init__.py +34 -16
  39. crawlo/stats_collector.py +17 -3
  40. crawlo/task_manager.py +112 -3
  41. crawlo/templates/project/settings.py.tmpl +103 -202
  42. crawlo/templates/project/settings_distributed.py.tmpl +122 -135
  43. crawlo/templates/project/settings_gentle.py.tmpl +149 -43
  44. crawlo/templates/project/settings_high_performance.py.tmpl +127 -90
  45. crawlo/templates/project/settings_minimal.py.tmpl +46 -15
  46. crawlo/templates/project/settings_simple.py.tmpl +138 -75
  47. crawlo/templates/project/spiders/__init__.py.tmpl +5 -1
  48. crawlo/templates/run.py.tmpl +10 -14
  49. crawlo/templates/spiders_init.py.tmpl +10 -0
  50. crawlo/tools/network_diagnostic.py +365 -0
  51. crawlo/utils/class_loader.py +26 -0
  52. crawlo/utils/error_handler.py +76 -35
  53. crawlo/utils/log.py +41 -144
  54. crawlo/utils/redis_connection_pool.py +43 -6
  55. crawlo/utils/request_serializer.py +8 -1
  56. {crawlo-1.3.2.dist-info → crawlo-1.3.4.dist-info}/METADATA +120 -14
  57. {crawlo-1.3.2.dist-info → crawlo-1.3.4.dist-info}/RECORD +104 -45
  58. tests/authenticated_proxy_example.py +2 -2
  59. tests/baidu_performance_test.py +109 -0
  60. tests/baidu_test.py +60 -0
  61. tests/comprehensive_framework_test.py +213 -0
  62. tests/comprehensive_test.py +82 -0
  63. tests/comprehensive_testing_summary.md +187 -0
  64. tests/debug_configure.py +70 -0
  65. tests/debug_framework_logger.py +85 -0
  66. tests/debug_log_levels.py +64 -0
  67. tests/distributed_test.py +67 -0
  68. tests/distributed_test_debug.py +77 -0
  69. tests/final_command_test_report.md +0 -0
  70. tests/final_comprehensive_test.py +152 -0
  71. tests/final_validation_test.py +183 -0
  72. tests/framework_performance_test.py +203 -0
  73. tests/optimized_performance_test.py +212 -0
  74. tests/performance_comparison.py +246 -0
  75. tests/queue_blocking_test.py +114 -0
  76. tests/queue_test.py +90 -0
  77. tests/scrapy_comparison/ofweek_scrapy.py +139 -0
  78. tests/scrapy_comparison/scrapy_test.py +134 -0
  79. tests/simple_command_test.py +120 -0
  80. tests/simple_crawlo_test.py +128 -0
  81. tests/simple_log_test.py +58 -0
  82. tests/simple_optimization_test.py +129 -0
  83. tests/simple_spider_test.py +50 -0
  84. tests/simple_test.py +48 -0
  85. tests/test_all_commands.py +231 -0
  86. tests/test_batch_processor.py +179 -0
  87. tests/test_component_factory.py +175 -0
  88. tests/test_controlled_spider_mixin.py +80 -0
  89. tests/test_enhanced_error_handler_comprehensive.py +246 -0
  90. tests/test_factories.py +253 -0
  91. tests/test_framework_logger.py +67 -0
  92. tests/test_framework_startup.py +65 -0
  93. tests/test_large_scale_config.py +113 -0
  94. tests/test_large_scale_helper.py +236 -0
  95. tests/test_mode_change.py +73 -0
  96. tests/test_mode_consistency.py +1 -1
  97. tests/test_performance_monitor.py +116 -0
  98. tests/test_queue_empty_check.py +42 -0
  99. tests/untested_features_report.md +139 -0
  100. tests/verify_debug.py +52 -0
  101. tests/verify_log_fix.py +112 -0
  102. tests/DOUBLE_CRAWLO_PREFIX_FIX_REPORT.md +0 -82
  103. {crawlo-1.3.2.dist-info → crawlo-1.3.4.dist-info}/WHEEL +0 -0
  104. {crawlo-1.3.2.dist-info → crawlo-1.3.4.dist-info}/entry_points.txt +0 -0
  105. {crawlo-1.3.2.dist-info → crawlo-1.3.4.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,194 @@
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ 核心初始化器 - 协调整个初始化过程
5
+ """
6
+
7
+ import time
8
+ import threading
9
+ from typing import Optional, Any
10
+
11
+ from .context import InitializationContext
12
+ from .phases import InitializationPhase, PhaseResult, get_execution_order, get_phase_definition
13
+ from .registry import get_global_registry, BaseInitializer, register_initializer
14
+ from .built_in import register_built_in_initializers
15
+
16
+
17
+ class CoreInitializer:
18
+ """
19
+ 核心初始化器 - 协调整个框架的初始化过程
20
+
21
+ 职责:
22
+ 1. 管理初始化阶段的执行顺序
23
+ 2. 处理阶段间的依赖关系
24
+ 3. 提供统一的初始化入口
25
+ 4. 错误处理和降级策略
26
+ """
27
+
28
+ _instance: Optional['CoreInitializer'] = None
29
+ _lock = threading.Lock()
30
+
31
+ def __new__(cls) -> 'CoreInitializer':
32
+ if cls._instance is None:
33
+ with cls._lock:
34
+ if cls._instance is None:
35
+ cls._instance = super(CoreInitializer, cls).__new__(cls)
36
+ cls._instance._initialized = False
37
+ return cls._instance
38
+
39
+ def __init__(self):
40
+ if hasattr(self, '_initialized') and self._initialized:
41
+ return
42
+
43
+ self._context: Optional[InitializationContext] = None
44
+ self._is_ready = False
45
+ self._init_lock = threading.RLock()
46
+
47
+ # 注册内置初始化器
48
+ register_built_in_initializers()
49
+
50
+ self._initialized = True
51
+
52
+ @property
53
+ def context(self) -> Optional[InitializationContext]:
54
+ """获取初始化上下文"""
55
+ return self._context
56
+
57
+ @property
58
+ def is_ready(self) -> bool:
59
+ """检查框架是否已准备就绪"""
60
+ return self._is_ready
61
+
62
+ def initialize(self, settings=None, **kwargs) -> Any:
63
+ """
64
+ 执行框架初始化
65
+
66
+ Args:
67
+ settings: 配置对象
68
+ **kwargs: 额外的配置参数
69
+
70
+ Returns:
71
+ 初始化后的配置管理器
72
+ """
73
+ with self._init_lock:
74
+ # 如果已经初始化完成,直接返回
75
+ if self._is_ready and self._context and self._context.settings:
76
+ return self._context.settings
77
+
78
+ # 创建初始化上下文
79
+ context = InitializationContext()
80
+ context.custom_settings = kwargs
81
+ self._context = context
82
+
83
+ try:
84
+ # 执行初始化阶段
85
+ self._execute_initialization_phases(context)
86
+
87
+ # 检查关键阶段是否完成
88
+ if not context.is_phase_completed(InitializationPhase.SETTINGS):
89
+ raise RuntimeError("Settings initialization failed")
90
+
91
+ self._is_ready = True
92
+ context.finish()
93
+
94
+ return context.settings
95
+
96
+ except Exception as e:
97
+ context.add_error(f"Framework initialization failed: {e}")
98
+ context.finish()
99
+
100
+ # 降级策略
101
+ return self._fallback_initialization(settings, **kwargs)
102
+
103
+ def _execute_initialization_phases(self, context: InitializationContext):
104
+ """执行初始化阶段"""
105
+ registry = get_global_registry()
106
+ execution_order = get_execution_order()
107
+
108
+ # 只执行已注册的阶段
109
+ registered_phases = set(registry.get_all_phases())
110
+
111
+ for phase in execution_order:
112
+ if phase == InitializationPhase.ERROR:
113
+ continue
114
+
115
+ # 只执行已注册的阶段
116
+ if phase not in registered_phases:
117
+ continue
118
+
119
+ context.set_current_phase(phase)
120
+
121
+ # 检查依赖关系
122
+ if not self._check_dependencies(phase, context):
123
+ phase_def = get_phase_definition(phase)
124
+ if not (phase_def and phase_def.optional):
125
+ raise RuntimeError(f"Dependencies not satisfied for phase {phase}")
126
+ else:
127
+ # 可选阶段,跳过
128
+ continue
129
+
130
+ # 执行阶段
131
+ start_time = time.time()
132
+ try:
133
+ result = registry.execute_phase(phase, context)
134
+ result.duration = time.time() - start_time
135
+
136
+ context.mark_phase_completed(phase, result)
137
+
138
+ if not result.success and not self._is_phase_optional(phase):
139
+ raise RuntimeError(f"Phase {phase} failed: {result.error}")
140
+
141
+ except Exception as e:
142
+ duration = time.time() - start_time
143
+ result = PhaseResult(
144
+ phase=phase,
145
+ success=False,
146
+ duration=duration,
147
+ error=e
148
+ )
149
+ context.mark_phase_completed(phase, result)
150
+
151
+ if not self._is_phase_optional(phase):
152
+ raise
153
+
154
+ def _check_dependencies(self, phase: InitializationPhase,
155
+ context: InitializationContext) -> bool:
156
+ """检查阶段依赖关系"""
157
+ phase_def = get_phase_definition(phase)
158
+ if not phase_def:
159
+ return True
160
+
161
+ for dependency in phase_def.dependencies:
162
+ if not context.is_phase_completed(dependency):
163
+ return False
164
+
165
+ return True
166
+
167
+ def _is_phase_optional(self, phase: InitializationPhase) -> bool:
168
+ """检查阶段是否可选"""
169
+ phase_def = get_phase_definition(phase)
170
+ return phase_def.optional if phase_def else False
171
+
172
+ def _fallback_initialization(self, settings=None, **kwargs):
173
+ """降级初始化策略"""
174
+ try:
175
+ # 尝试创建基本的配置管理器
176
+ from crawlo.settings.setting_manager import SettingManager
177
+
178
+ if settings:
179
+ return settings
180
+ else:
181
+ fallback_settings = SettingManager()
182
+ if kwargs:
183
+ fallback_settings.update_attributes(kwargs)
184
+ return fallback_settings
185
+
186
+ except Exception:
187
+ # 如果连降级都失败,返回None
188
+ return None
189
+
190
+ def reset(self):
191
+ """重置初始化状态(主要用于测试)"""
192
+ with self._init_lock:
193
+ self._context = None
194
+ self._is_ready = False
@@ -0,0 +1,149 @@
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ 初始化阶段定义
5
+ """
6
+
7
+ from enum import Enum
8
+ from dataclasses import dataclass
9
+ from typing import List, Optional
10
+
11
+
12
+ class InitializationPhase(Enum):
13
+ """初始化阶段枚举"""
14
+
15
+ # 阶段0:准备阶段
16
+ PREPARING = "preparing"
17
+
18
+ # 阶段1:日志系统初始化
19
+ LOGGING = "logging"
20
+
21
+ # 阶段2:配置系统初始化
22
+ SETTINGS = "settings"
23
+
24
+ # 阶段3:核心组件初始化
25
+ CORE_COMPONENTS = "core_components"
26
+
27
+ # 阶段4:扩展组件初始化
28
+ EXTENSIONS = "extensions"
29
+
30
+ # 阶段5:框架启动日志记录
31
+ FRAMEWORK_STARTUP_LOG = "framework_startup_log"
32
+
33
+ # 阶段6:完成
34
+ COMPLETED = "completed"
35
+
36
+ # 错误状态
37
+ ERROR = "error"
38
+
39
+
40
+ @dataclass
41
+ class PhaseResult:
42
+ """阶段执行结果"""
43
+ phase: InitializationPhase
44
+ success: bool
45
+ duration: float = 0.0
46
+ error: Optional[Exception] = None
47
+ artifacts: dict = None # 阶段产生的工件
48
+
49
+ def __post_init__(self):
50
+ if self.artifacts is None:
51
+ self.artifacts = {}
52
+
53
+
54
+ @dataclass
55
+ class PhaseDefinition:
56
+ """阶段定义"""
57
+ phase: InitializationPhase
58
+ name: str
59
+ description: str
60
+ dependencies: List[InitializationPhase] = None
61
+ optional: bool = False
62
+ timeout: float = 30.0 # 超时时间(秒)
63
+
64
+ def __post_init__(self):
65
+ if self.dependencies is None:
66
+ self.dependencies = []
67
+
68
+
69
+ # 预定义的初始化阶段
70
+ PHASE_DEFINITIONS = [
71
+ PhaseDefinition(
72
+ phase=InitializationPhase.PREPARING,
73
+ name="准备阶段",
74
+ description="初始化基础环境和检查前置条件",
75
+ dependencies=[],
76
+ timeout=5.0
77
+ ),
78
+ PhaseDefinition(
79
+ phase=InitializationPhase.LOGGING,
80
+ name="日志系统",
81
+ description="配置和初始化日志系统",
82
+ dependencies=[], # 移除对PREPARING的依赖
83
+ timeout=10.0
84
+ ),
85
+ PhaseDefinition(
86
+ phase=InitializationPhase.SETTINGS,
87
+ name="配置系统",
88
+ description="加载和验证配置",
89
+ dependencies=[InitializationPhase.LOGGING],
90
+ timeout=15.0
91
+ ),
92
+ PhaseDefinition(
93
+ phase=InitializationPhase.CORE_COMPONENTS,
94
+ name="核心组件",
95
+ description="初始化框架核心组件",
96
+ dependencies=[InitializationPhase.SETTINGS],
97
+ timeout=20.0
98
+ ),
99
+ PhaseDefinition(
100
+ phase=InitializationPhase.EXTENSIONS,
101
+ name="扩展组件",
102
+ description="加载和初始化扩展组件",
103
+ dependencies=[InitializationPhase.CORE_COMPONENTS],
104
+ optional=True,
105
+ timeout=15.0
106
+ ),
107
+ PhaseDefinition(
108
+ phase=InitializationPhase.FRAMEWORK_STARTUP_LOG,
109
+ name="框架启动日志",
110
+ description="记录框架启动相关信息",
111
+ dependencies=[InitializationPhase.LOGGING, InitializationPhase.SETTINGS],
112
+ timeout=5.0
113
+ ),
114
+ PhaseDefinition(
115
+ phase=InitializationPhase.COMPLETED,
116
+ name="初始化完成",
117
+ description="框架初始化完成",
118
+ dependencies=[
119
+ InitializationPhase.CORE_COMPONENTS,
120
+ InitializationPhase.FRAMEWORK_STARTUP_LOG
121
+ ], # Extensions是可选的
122
+ timeout=5.0
123
+ )
124
+ ]
125
+
126
+
127
+ def get_phase_definition(phase: InitializationPhase) -> Optional[PhaseDefinition]:
128
+ """获取阶段定义"""
129
+ for definition in PHASE_DEFINITIONS:
130
+ if definition.phase == phase:
131
+ return definition
132
+ return None
133
+
134
+
135
+ def get_execution_order() -> List[InitializationPhase]:
136
+ """获取执行顺序"""
137
+ return [definition.phase for definition in PHASE_DEFINITIONS]
138
+
139
+
140
+ def validate_dependencies() -> bool:
141
+ """验证阶段依赖关系的正确性"""
142
+ phases = {definition.phase for definition in PHASE_DEFINITIONS}
143
+
144
+ for definition in PHASE_DEFINITIONS:
145
+ for dependency in definition.dependencies:
146
+ if dependency not in phases:
147
+ return False
148
+
149
+ return True
@@ -0,0 +1,146 @@
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ 初始化器注册表 - 管理所有初始化器的注册和执行
5
+ """
6
+
7
+ import threading
8
+ from typing import Dict, Optional, Callable, List
9
+ from .context import InitializationContext
10
+ from .phases import InitializationPhase, PhaseResult
11
+
12
+
13
+ class Initializer:
14
+ """初始化器基类"""
15
+
16
+ def __init__(self, phase: InitializationPhase):
17
+ self._phase = phase
18
+
19
+ @property
20
+ def phase(self) -> InitializationPhase:
21
+ """获取初始化阶段"""
22
+ return self._phase
23
+
24
+ def initialize(self, context: InitializationContext) -> PhaseResult:
25
+ """执行初始化 - 子类必须实现"""
26
+ raise NotImplementedError("Subclasses must implement initialize method")
27
+
28
+
29
+ class BaseInitializer(Initializer):
30
+ """基础初始化器类 - 为向后兼容保留"""
31
+
32
+ def __init__(self, phase: InitializationPhase):
33
+ super().__init__(phase)
34
+
35
+ def _create_result(self, success: bool, duration: float = 0.0,
36
+ artifacts: Optional[Dict] = None, error: Optional[Exception] = None) -> PhaseResult:
37
+ """创建初始化结果"""
38
+ return PhaseResult(
39
+ phase=self.phase,
40
+ success=success,
41
+ duration=duration,
42
+ artifacts=artifacts or {},
43
+ error=error
44
+ )
45
+
46
+
47
+ class InitializerRegistry:
48
+ """
49
+ 初始化器注册表 - 管理所有初始化器的注册和执行
50
+
51
+ 特点:
52
+ 1. 线程安全的注册和执行
53
+ 2. 支持函数式和类式初始化器
54
+ 3. 统一的结果处理
55
+ """
56
+
57
+ def __init__(self):
58
+ self._initializers: Dict[InitializationPhase, Initializer] = {}
59
+ self._lock = threading.RLock()
60
+
61
+ def register(self, initializer: Initializer):
62
+ """注册初始化器"""
63
+ with self._lock:
64
+ phase = initializer.phase
65
+ if phase in self._initializers:
66
+ raise ValueError(f"Initializer for phase {phase} already registered")
67
+ self._initializers[phase] = initializer
68
+
69
+ def register_function(self, phase: InitializationPhase,
70
+ init_func: Callable[[InitializationContext], PhaseResult]):
71
+ """注册函数式初始化器"""
72
+
73
+ class FunctionInitializer:
74
+ def __init__(self, phase: InitializationPhase, func: Callable):
75
+ self._phase = phase
76
+ self._func = func
77
+
78
+ @property
79
+ def phase(self) -> InitializationPhase:
80
+ return self._phase
81
+
82
+ def initialize(self, context: InitializationContext) -> PhaseResult:
83
+ return self._func(context)
84
+
85
+ self.register(FunctionInitializer(phase, init_func))
86
+
87
+ def get_initializer(self, phase: InitializationPhase) -> Optional[Initializer]:
88
+ """获取指定阶段的初始化器"""
89
+ with self._lock:
90
+ return self._initializers.get(phase)
91
+
92
+ def get_all_phases(self) -> List[InitializationPhase]:
93
+ """获取所有已注册的阶段"""
94
+ with self._lock:
95
+ return list(self._initializers.keys())
96
+
97
+ def has_initializer(self, phase: InitializationPhase) -> bool:
98
+ """检查是否有指定阶段的初始化器"""
99
+ with self._lock:
100
+ return phase in self._initializers
101
+
102
+ def clear(self):
103
+ """清空注册表"""
104
+ with self._lock:
105
+ self._initializers.clear()
106
+
107
+ def execute_phase(self, phase: InitializationPhase,
108
+ context: InitializationContext) -> PhaseResult:
109
+ """执行指定阶段的初始化"""
110
+ initializer = self.get_initializer(phase)
111
+ if not initializer:
112
+ error = ValueError(f"No initializer registered for phase {phase}")
113
+ return PhaseResult(
114
+ phase=phase,
115
+ success=False,
116
+ error=error
117
+ )
118
+
119
+ try:
120
+ return initializer.initialize(context)
121
+ except Exception as e:
122
+ return PhaseResult(
123
+ phase=phase,
124
+ success=False,
125
+ error=e
126
+ )
127
+
128
+
129
+ # 全局注册表实例
130
+ _global_registry = InitializerRegistry()
131
+
132
+
133
+ def get_global_registry() -> InitializerRegistry:
134
+ """获取全局注册表"""
135
+ return _global_registry
136
+
137
+
138
+ def register_initializer(initializer: Initializer):
139
+ """注册初始化器到全局注册表"""
140
+ _global_registry.register(initializer)
141
+
142
+
143
+ def register_phase_function(phase: InitializationPhase,
144
+ init_func: Callable[[InitializationContext], PhaseResult]):
145
+ """注册函数式初始化器到全局注册表"""
146
+ _global_registry.register_function(phase, init_func)
crawlo/items/base.py CHANGED
@@ -6,6 +6,7 @@
6
6
  from abc import ABCMeta
7
7
  from .fields import Field
8
8
 
9
+
9
10
  class ItemMeta(ABCMeta):
10
11
  def __new__(mcs, name, bases, attrs):
11
12
  fields = {}
@@ -19,4 +20,4 @@ class ItemMeta(ABCMeta):
19
20
 
20
21
  cls_instance = super().__new__(mcs, name, bases, cls_attrs)
21
22
  cls_instance.FIELDS = fields
22
- return cls_instance
23
+ return cls_instance
@@ -0,0 +1,38 @@
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ Crawlo统一日志系统
5
+ =================
6
+
7
+ 设计原则:
8
+ 1. 简单优先 - 避免过度设计
9
+ 2. 性能优先 - 减少锁竞争和复杂逻辑
10
+ 3. 一致性 - 统一的日志接口
11
+ 4. 可靠性 - 确保日志始终可用
12
+ """
13
+
14
+ from .manager import LogManager
15
+ from .factory import LoggerFactory
16
+ from .config import LogConfig
17
+
18
+ # 统一的公共接口
19
+ def get_logger(name: str = 'default'):
20
+ """获取logger实例"""
21
+ return LoggerFactory.get_logger(name)
22
+
23
+ def configure_logging(settings=None, **kwargs):
24
+ """配置日志系统"""
25
+ return LogManager().configure(settings, **kwargs)
26
+
27
+ def is_configured() -> bool:
28
+ """检查日志系统是否已配置"""
29
+ return LogManager().is_configured
30
+
31
+ __all__ = [
32
+ 'LogManager',
33
+ 'LoggerFactory',
34
+ 'LogConfig',
35
+ 'get_logger',
36
+ 'configure_logging',
37
+ 'is_configured'
38
+ ]
@@ -0,0 +1,97 @@
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ 日志配置管理
5
+ """
6
+
7
+ import os
8
+ from dataclasses import dataclass, field
9
+ from typing import Optional, Dict, Any
10
+
11
+
12
+ @dataclass
13
+ class LogConfig:
14
+ """日志配置数据类 - 简单明确的配置结构"""
15
+
16
+ # 基本配置
17
+ level: str = "INFO"
18
+ format: str = "%(asctime)s - [%(name)s] - %(levelname)s: %(message)s"
19
+ encoding: str = "utf-8"
20
+
21
+ # 文件配置
22
+ file_path: Optional[str] = None
23
+ max_bytes: int = 10 * 1024 * 1024 # 10MB
24
+ backup_count: int = 5
25
+
26
+ # 控制台配置
27
+ console_enabled: bool = True
28
+ file_enabled: bool = True
29
+
30
+ # 模块级别配置
31
+ module_levels: Dict[str, str] = field(default_factory=dict)
32
+
33
+ @classmethod
34
+ def from_settings(cls, settings) -> 'LogConfig':
35
+ """从settings对象创建配置"""
36
+ if not settings:
37
+ return cls()
38
+
39
+ # 使用settings的get方法而不是getattr
40
+ if hasattr(settings, 'get'):
41
+ get_val = settings.get
42
+ else:
43
+ get_val = lambda k, d=None: getattr(settings, k, d)
44
+
45
+ # 获取默认值
46
+ format_default_value = "%(asctime)s - [%(name)s] - %(levelname)s: %(message)s"
47
+
48
+ return cls(
49
+ level=get_val('LOG_LEVEL', 'INFO'),
50
+ format=get_val('LOG_FORMAT', format_default_value),
51
+ encoding=get_val('LOG_ENCODING', 'utf-8'),
52
+ file_path=get_val('LOG_FILE'),
53
+ max_bytes=get_val('LOG_MAX_BYTES', 10 * 1024 * 1024),
54
+ backup_count=get_val('LOG_BACKUP_COUNT', 5),
55
+ console_enabled=get_val('LOG_CONSOLE_ENABLED', True),
56
+ file_enabled=get_val('LOG_FILE_ENABLED', True),
57
+ module_levels=get_val('LOG_LEVELS', {})
58
+ )
59
+
60
+ @classmethod
61
+ def from_dict(cls, config_dict: Dict[str, Any]) -> 'LogConfig':
62
+ """从字典创建配置"""
63
+ return cls(**{k: v for k, v in config_dict.items() if k in cls.__annotations__})
64
+
65
+ def get_module_level(self, module_name: str) -> str:
66
+ """获取模块的日志级别"""
67
+ # 先查找精确匹配
68
+ if module_name in self.module_levels:
69
+ return self.module_levels[module_name]
70
+
71
+ # 查找父模块匹配
72
+ parts = module_name.split('.')
73
+ for i in range(len(parts) - 1, 0, -1):
74
+ parent_module = '.'.join(parts[:i])
75
+ if parent_module in self.module_levels:
76
+ return self.module_levels[parent_module]
77
+
78
+ # 返回默认级别
79
+ return self.level
80
+
81
+ def validate(self) -> bool:
82
+ """验证配置有效性"""
83
+ valid_levels = {'DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'}
84
+
85
+ if self.level.upper() not in valid_levels:
86
+ return False
87
+
88
+ # 确保日志目录存在
89
+ if self.file_path and self.file_enabled:
90
+ try:
91
+ log_dir = os.path.dirname(self.file_path)
92
+ if log_dir and not os.path.exists(log_dir):
93
+ os.makedirs(log_dir, exist_ok=True)
94
+ except (OSError, PermissionError):
95
+ return False
96
+
97
+ return True