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.
- crawlo/__init__.py +24 -0
- crawlo/__version__.py +1 -1
- crawlo/commands/run.py +58 -32
- crawlo/core/__init__.py +44 -0
- crawlo/core/engine.py +119 -45
- crawlo/core/scheduler.py +4 -3
- crawlo/crawler.py +603 -1133
- crawlo/downloader/aiohttp_downloader.py +4 -2
- crawlo/extension/__init__.py +1 -1
- crawlo/extension/logging_extension.py +23 -7
- crawlo/factories/__init__.py +28 -0
- crawlo/factories/base.py +69 -0
- crawlo/factories/crawler.py +104 -0
- crawlo/factories/registry.py +85 -0
- crawlo/filters/aioredis_filter.py +25 -2
- crawlo/framework.py +292 -0
- crawlo/initialization/__init__.py +40 -0
- crawlo/initialization/built_in.py +426 -0
- crawlo/initialization/context.py +142 -0
- crawlo/initialization/core.py +194 -0
- crawlo/initialization/phases.py +149 -0
- crawlo/initialization/registry.py +146 -0
- crawlo/items/base.py +2 -1
- crawlo/logging/__init__.py +38 -0
- crawlo/logging/config.py +97 -0
- crawlo/logging/factory.py +129 -0
- crawlo/logging/manager.py +112 -0
- crawlo/middleware/middleware_manager.py +1 -1
- crawlo/middleware/offsite.py +1 -1
- crawlo/mode_manager.py +26 -1
- crawlo/pipelines/pipeline_manager.py +2 -1
- crawlo/project.py +76 -46
- crawlo/queue/pqueue.py +11 -5
- crawlo/queue/queue_manager.py +143 -19
- crawlo/queue/redis_priority_queue.py +69 -49
- crawlo/settings/default_settings.py +110 -14
- crawlo/settings/setting_manager.py +29 -13
- crawlo/spider/__init__.py +34 -16
- crawlo/stats_collector.py +17 -3
- crawlo/task_manager.py +112 -3
- crawlo/templates/project/settings.py.tmpl +103 -202
- crawlo/templates/project/settings_distributed.py.tmpl +122 -135
- crawlo/templates/project/settings_gentle.py.tmpl +149 -43
- crawlo/templates/project/settings_high_performance.py.tmpl +127 -90
- crawlo/templates/project/settings_minimal.py.tmpl +46 -15
- crawlo/templates/project/settings_simple.py.tmpl +138 -75
- crawlo/templates/project/spiders/__init__.py.tmpl +5 -1
- crawlo/templates/run.py.tmpl +10 -14
- crawlo/templates/spiders_init.py.tmpl +10 -0
- crawlo/tools/network_diagnostic.py +365 -0
- crawlo/utils/class_loader.py +26 -0
- crawlo/utils/error_handler.py +76 -35
- crawlo/utils/log.py +41 -144
- crawlo/utils/redis_connection_pool.py +43 -6
- crawlo/utils/request_serializer.py +8 -1
- {crawlo-1.3.2.dist-info → crawlo-1.3.4.dist-info}/METADATA +120 -14
- {crawlo-1.3.2.dist-info → crawlo-1.3.4.dist-info}/RECORD +104 -45
- tests/authenticated_proxy_example.py +2 -2
- tests/baidu_performance_test.py +109 -0
- tests/baidu_test.py +60 -0
- tests/comprehensive_framework_test.py +213 -0
- tests/comprehensive_test.py +82 -0
- tests/comprehensive_testing_summary.md +187 -0
- tests/debug_configure.py +70 -0
- tests/debug_framework_logger.py +85 -0
- tests/debug_log_levels.py +64 -0
- tests/distributed_test.py +67 -0
- tests/distributed_test_debug.py +77 -0
- tests/final_command_test_report.md +0 -0
- tests/final_comprehensive_test.py +152 -0
- tests/final_validation_test.py +183 -0
- tests/framework_performance_test.py +203 -0
- tests/optimized_performance_test.py +212 -0
- tests/performance_comparison.py +246 -0
- tests/queue_blocking_test.py +114 -0
- tests/queue_test.py +90 -0
- tests/scrapy_comparison/ofweek_scrapy.py +139 -0
- tests/scrapy_comparison/scrapy_test.py +134 -0
- tests/simple_command_test.py +120 -0
- tests/simple_crawlo_test.py +128 -0
- tests/simple_log_test.py +58 -0
- tests/simple_optimization_test.py +129 -0
- tests/simple_spider_test.py +50 -0
- tests/simple_test.py +48 -0
- tests/test_all_commands.py +231 -0
- tests/test_batch_processor.py +179 -0
- tests/test_component_factory.py +175 -0
- tests/test_controlled_spider_mixin.py +80 -0
- tests/test_enhanced_error_handler_comprehensive.py +246 -0
- tests/test_factories.py +253 -0
- tests/test_framework_logger.py +67 -0
- tests/test_framework_startup.py +65 -0
- tests/test_large_scale_config.py +113 -0
- tests/test_large_scale_helper.py +236 -0
- tests/test_mode_change.py +73 -0
- tests/test_mode_consistency.py +1 -1
- tests/test_performance_monitor.py +116 -0
- tests/test_queue_empty_check.py +42 -0
- tests/untested_features_report.md +139 -0
- tests/verify_debug.py +52 -0
- tests/verify_log_fix.py +112 -0
- tests/DOUBLE_CRAWLO_PREFIX_FIX_REPORT.md +0 -82
- {crawlo-1.3.2.dist-info → crawlo-1.3.4.dist-info}/WHEEL +0 -0
- {crawlo-1.3.2.dist-info → crawlo-1.3.4.dist-info}/entry_points.txt +0 -0
- {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
|
+
]
|
crawlo/logging/config.py
ADDED
|
@@ -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
|