crawlo 1.4.2__py3-none-any.whl → 1.4.3__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 +93 -93
- crawlo/__version__.py +1 -1
- crawlo/cli.py +75 -75
- crawlo/commands/__init__.py +14 -14
- crawlo/commands/check.py +594 -594
- crawlo/commands/genspider.py +151 -151
- crawlo/commands/help.py +138 -138
- crawlo/commands/list.py +155 -155
- crawlo/commands/run.py +341 -341
- crawlo/commands/startproject.py +436 -436
- crawlo/commands/stats.py +187 -187
- crawlo/commands/utils.py +196 -196
- crawlo/config.py +312 -312
- crawlo/config_validator.py +277 -277
- crawlo/core/__init__.py +52 -52
- crawlo/core/engine.py +438 -439
- crawlo/core/processor.py +47 -47
- crawlo/core/scheduler.py +291 -257
- crawlo/crawler.py +650 -650
- crawlo/data/__init__.py +5 -5
- crawlo/data/user_agents.py +194 -194
- crawlo/downloader/__init__.py +273 -273
- crawlo/downloader/aiohttp_downloader.py +233 -233
- crawlo/downloader/cffi_downloader.py +245 -245
- crawlo/downloader/httpx_downloader.py +259 -259
- crawlo/downloader/hybrid_downloader.py +212 -212
- crawlo/downloader/playwright_downloader.py +402 -402
- crawlo/downloader/selenium_downloader.py +472 -472
- crawlo/event.py +11 -11
- crawlo/exceptions.py +81 -81
- crawlo/extension/__init__.py +63 -63
- crawlo/extension/health_check.py +141 -141
- crawlo/extension/log_interval.py +94 -94
- crawlo/extension/log_stats.py +70 -70
- crawlo/extension/logging_extension.py +61 -61
- crawlo/extension/memory_monitor.py +104 -104
- crawlo/extension/performance_profiler.py +133 -133
- crawlo/extension/request_recorder.py +107 -107
- crawlo/factories/__init__.py +27 -27
- crawlo/factories/base.py +68 -68
- crawlo/factories/crawler.py +103 -103
- crawlo/factories/registry.py +84 -84
- crawlo/filters/__init__.py +154 -154
- crawlo/filters/aioredis_filter.py +257 -257
- crawlo/filters/memory_filter.py +269 -269
- crawlo/framework.py +292 -292
- crawlo/initialization/__init__.py +44 -44
- crawlo/initialization/built_in.py +425 -425
- crawlo/initialization/context.py +141 -141
- crawlo/initialization/core.py +193 -193
- crawlo/initialization/phases.py +148 -148
- crawlo/initialization/registry.py +145 -145
- crawlo/items/__init__.py +23 -23
- crawlo/items/base.py +23 -23
- crawlo/items/fields.py +52 -52
- crawlo/items/items.py +104 -104
- crawlo/logging/__init__.py +45 -37
- crawlo/logging/async_handler.py +181 -0
- crawlo/logging/config.py +196 -96
- crawlo/logging/factory.py +171 -128
- crawlo/logging/manager.py +111 -111
- crawlo/logging/monitor.py +153 -0
- crawlo/logging/sampler.py +167 -0
- crawlo/middleware/__init__.py +21 -21
- crawlo/middleware/default_header.py +132 -132
- crawlo/middleware/download_delay.py +104 -104
- crawlo/middleware/middleware_manager.py +135 -135
- crawlo/middleware/offsite.py +123 -123
- crawlo/middleware/proxy.py +386 -386
- crawlo/middleware/request_ignore.py +86 -86
- crawlo/middleware/response_code.py +150 -150
- crawlo/middleware/response_filter.py +136 -136
- crawlo/middleware/retry.py +124 -124
- crawlo/middleware/simple_proxy.py +65 -65
- crawlo/mode_manager.py +219 -219
- crawlo/network/__init__.py +21 -21
- crawlo/network/request.py +379 -379
- crawlo/network/response.py +359 -359
- crawlo/pipelines/__init__.py +21 -21
- crawlo/pipelines/bloom_dedup_pipeline.py +146 -146
- crawlo/pipelines/console_pipeline.py +39 -39
- crawlo/pipelines/csv_pipeline.py +316 -316
- crawlo/pipelines/database_dedup_pipeline.py +197 -197
- crawlo/pipelines/json_pipeline.py +218 -218
- crawlo/pipelines/memory_dedup_pipeline.py +105 -105
- crawlo/pipelines/mongo_pipeline.py +131 -131
- crawlo/pipelines/mysql_pipeline.py +325 -325
- crawlo/pipelines/pipeline_manager.py +100 -84
- crawlo/pipelines/redis_dedup_pipeline.py +156 -156
- crawlo/project.py +349 -338
- crawlo/queue/pqueue.py +42 -42
- crawlo/queue/queue_manager.py +526 -522
- crawlo/queue/redis_priority_queue.py +370 -367
- crawlo/settings/__init__.py +7 -7
- crawlo/settings/default_settings.py +284 -284
- crawlo/settings/setting_manager.py +219 -219
- crawlo/spider/__init__.py +657 -657
- crawlo/stats_collector.py +73 -73
- crawlo/subscriber.py +129 -129
- crawlo/task_manager.py +138 -138
- crawlo/templates/crawlo.cfg.tmpl +10 -10
- crawlo/templates/project/__init__.py.tmpl +3 -3
- crawlo/templates/project/items.py.tmpl +17 -17
- crawlo/templates/project/middlewares.py.tmpl +118 -118
- crawlo/templates/project/pipelines.py.tmpl +96 -96
- crawlo/templates/project/settings.py.tmpl +170 -170
- crawlo/templates/project/settings_distributed.py.tmpl +169 -169
- crawlo/templates/project/settings_gentle.py.tmpl +166 -166
- crawlo/templates/project/settings_high_performance.py.tmpl +167 -167
- crawlo/templates/project/settings_minimal.py.tmpl +65 -65
- crawlo/templates/project/settings_simple.py.tmpl +164 -164
- crawlo/templates/project/spiders/__init__.py.tmpl +9 -9
- crawlo/templates/run.py.tmpl +34 -34
- crawlo/templates/spider/spider.py.tmpl +143 -143
- crawlo/templates/spiders_init.py.tmpl +9 -9
- crawlo/tools/__init__.py +200 -200
- crawlo/tools/anti_crawler.py +268 -268
- crawlo/tools/authenticated_proxy.py +240 -240
- crawlo/tools/data_formatter.py +225 -225
- crawlo/tools/data_validator.py +180 -180
- crawlo/tools/date_tools.py +289 -289
- crawlo/tools/distributed_coordinator.py +384 -384
- crawlo/tools/encoding_converter.py +127 -127
- crawlo/tools/network_diagnostic.py +364 -364
- crawlo/tools/request_tools.py +82 -82
- crawlo/tools/retry_mechanism.py +224 -224
- crawlo/tools/scenario_adapter.py +262 -262
- crawlo/tools/text_cleaner.py +232 -232
- crawlo/utils/__init__.py +34 -34
- crawlo/utils/batch_processor.py +259 -259
- crawlo/utils/class_loader.py +25 -25
- crawlo/utils/controlled_spider_mixin.py +439 -439
- crawlo/utils/db_helper.py +343 -343
- crawlo/utils/enhanced_error_handler.py +356 -356
- crawlo/utils/env_config.py +142 -142
- crawlo/utils/error_handler.py +165 -165
- crawlo/utils/fingerprint.py +122 -122
- crawlo/utils/func_tools.py +82 -82
- crawlo/utils/large_scale_config.py +286 -286
- crawlo/utils/large_scale_helper.py +344 -344
- crawlo/utils/log.py +79 -79
- crawlo/utils/performance_monitor.py +285 -285
- crawlo/utils/queue_helper.py +175 -175
- crawlo/utils/redis_connection_pool.py +388 -388
- crawlo/utils/redis_key_validator.py +198 -198
- crawlo/utils/request.py +267 -267
- crawlo/utils/request_serializer.py +225 -225
- crawlo/utils/spider_loader.py +61 -61
- crawlo/utils/system.py +11 -11
- crawlo/utils/tools.py +4 -4
- crawlo/utils/url.py +39 -39
- crawlo-1.4.3.dist-info/METADATA +190 -0
- crawlo-1.4.3.dist-info/RECORD +326 -0
- examples/__init__.py +7 -7
- examples/test_project/__init__.py +7 -7
- examples/test_project/run.py +34 -34
- examples/test_project/test_project/__init__.py +3 -3
- examples/test_project/test_project/items.py +17 -17
- examples/test_project/test_project/middlewares.py +118 -118
- examples/test_project/test_project/pipelines.py +96 -96
- examples/test_project/test_project/settings.py +169 -169
- examples/test_project/test_project/spiders/__init__.py +9 -9
- examples/test_project/test_project/spiders/of_week_dis.py +143 -143
- tests/__init__.py +7 -7
- tests/advanced_tools_example.py +275 -275
- tests/authenticated_proxy_example.py +106 -106
- tests/baidu_performance_test.py +108 -108
- tests/baidu_test.py +59 -59
- tests/cleaners_example.py +160 -160
- tests/comprehensive_framework_test.py +212 -212
- tests/comprehensive_test.py +81 -81
- tests/comprehensive_testing_summary.md +186 -186
- tests/config_validation_demo.py +142 -142
- tests/controlled_spider_example.py +205 -205
- tests/date_tools_example.py +180 -180
- tests/debug_configure.py +69 -69
- tests/debug_framework_logger.py +84 -84
- tests/debug_log_config.py +126 -126
- tests/debug_log_levels.py +63 -63
- tests/debug_pipelines.py +66 -66
- tests/detailed_log_test.py +233 -233
- tests/distributed_test.py +66 -66
- tests/distributed_test_debug.py +76 -76
- tests/dynamic_loading_example.py +523 -523
- tests/dynamic_loading_test.py +104 -104
- tests/env_config_example.py +133 -133
- tests/error_handling_example.py +171 -171
- tests/final_comprehensive_test.py +151 -151
- tests/final_log_test.py +260 -260
- tests/final_validation_test.py +182 -182
- tests/fix_log_test.py +142 -142
- tests/framework_performance_test.py +202 -202
- tests/log_buffering_test.py +111 -111
- tests/log_generation_timing_test.py +153 -153
- tests/optimized_performance_test.py +211 -211
- tests/performance_comparison.py +245 -245
- tests/queue_blocking_test.py +113 -113
- tests/queue_test.py +89 -89
- tests/redis_key_validation_demo.py +130 -130
- tests/request_params_example.py +150 -150
- tests/response_improvements_example.py +144 -144
- tests/scrapy_comparison/ofweek_scrapy.py +138 -138
- tests/scrapy_comparison/scrapy_test.py +133 -133
- tests/simple_command_test.py +119 -119
- tests/simple_crawlo_test.py +127 -127
- tests/simple_log_test.py +57 -57
- tests/simple_log_test2.py +137 -137
- tests/simple_optimization_test.py +128 -128
- tests/simple_queue_type_test.py +41 -41
- tests/simple_spider_test.py +49 -49
- tests/simple_test.py +47 -47
- tests/spider_log_timing_test.py +177 -177
- tests/test_advanced_tools.py +148 -148
- tests/test_all_commands.py +230 -230
- tests/test_all_pipeline_fingerprints.py +133 -133
- tests/test_all_redis_key_configs.py +145 -145
- tests/test_authenticated_proxy.py +141 -141
- tests/test_batch_processor.py +178 -178
- tests/test_cleaners.py +54 -54
- tests/test_component_factory.py +174 -174
- tests/test_comprehensive.py +146 -146
- tests/test_config_consistency.py +80 -80
- tests/test_config_merge.py +152 -152
- tests/test_config_validator.py +182 -182
- tests/test_controlled_spider_mixin.py +79 -79
- tests/test_crawlo_proxy_integration.py +108 -108
- tests/test_date_tools.py +123 -123
- tests/test_dedup_fix.py +220 -220
- tests/test_dedup_pipeline_consistency.py +125 -0
- tests/test_default_header_middleware.py +313 -313
- tests/test_distributed.py +65 -65
- tests/test_double_crawlo_fix.py +204 -204
- tests/test_double_crawlo_fix_simple.py +124 -124
- tests/test_download_delay_middleware.py +221 -221
- tests/test_downloader_proxy_compatibility.py +268 -268
- tests/test_dynamic_downloaders_proxy.py +124 -124
- tests/test_dynamic_proxy.py +92 -92
- tests/test_dynamic_proxy_config.py +146 -146
- tests/test_dynamic_proxy_real.py +109 -109
- tests/test_edge_cases.py +303 -303
- tests/test_enhanced_error_handler.py +270 -270
- tests/test_enhanced_error_handler_comprehensive.py +245 -245
- tests/test_env_config.py +121 -121
- tests/test_error_handler_compatibility.py +112 -112
- tests/test_factories.py +252 -252
- tests/test_final_validation.py +153 -153
- tests/test_fingerprint_consistency.py +135 -135
- tests/test_fingerprint_simple.py +51 -51
- tests/test_framework_env_usage.py +103 -103
- tests/test_framework_logger.py +66 -66
- tests/test_framework_startup.py +64 -64
- tests/test_get_component_logger.py +83 -83
- tests/test_hash_performance.py +99 -99
- tests/test_integration.py +169 -169
- tests/test_item_dedup_redis_key.py +122 -122
- tests/test_large_scale_config.py +112 -112
- tests/test_large_scale_helper.py +235 -235
- tests/test_logging_enhancements.py +375 -0
- tests/test_logging_final.py +185 -0
- tests/test_logging_integration.py +313 -0
- tests/test_logging_system.py +282 -282
- tests/test_middleware_debug.py +142 -0
- tests/test_mode_change.py +72 -72
- tests/test_mode_consistency.py +51 -51
- tests/test_offsite_middleware.py +244 -244
- tests/test_offsite_middleware_simple.py +203 -203
- tests/test_parsel.py +29 -29
- tests/test_performance.py +327 -327
- tests/test_performance_monitor.py +115 -115
- tests/test_pipeline_fingerprint_consistency.py +86 -86
- tests/test_priority_behavior.py +212 -0
- tests/test_priority_consistency.py +152 -0
- tests/test_priority_consistency_fixed.py +250 -0
- tests/test_proxy_api.py +264 -264
- tests/test_proxy_health_check.py +32 -32
- tests/test_proxy_middleware.py +121 -121
- tests/test_proxy_middleware_enhanced.py +216 -216
- tests/test_proxy_middleware_integration.py +136 -136
- tests/test_proxy_middleware_refactored.py +184 -184
- tests/test_proxy_providers.py +56 -56
- tests/test_proxy_stats.py +19 -19
- tests/test_proxy_strategies.py +59 -59
- tests/test_queue_empty_check.py +41 -41
- tests/test_queue_manager_double_crawlo.py +173 -173
- tests/test_queue_manager_redis_key.py +179 -179
- tests/test_queue_naming.py +154 -154
- tests/test_queue_type.py +106 -106
- tests/test_queue_type_redis_config_consistency.py +131 -0
- tests/test_random_headers_default.py +323 -0
- tests/test_random_headers_necessity.py +309 -0
- tests/test_random_user_agent.py +72 -72
- tests/test_real_scenario_proxy.py +195 -195
- tests/test_redis_config.py +28 -28
- tests/test_redis_connection_pool.py +294 -294
- tests/test_redis_key_naming.py +181 -181
- tests/test_redis_key_validator.py +123 -123
- tests/test_redis_queue.py +224 -224
- tests/test_redis_queue_name_fix.py +175 -175
- tests/test_redis_queue_type_fallback.py +130 -0
- tests/test_request_ignore_middleware.py +182 -182
- tests/test_request_params.py +111 -111
- tests/test_request_serialization.py +70 -70
- tests/test_response_code_middleware.py +349 -349
- tests/test_response_filter_middleware.py +427 -427
- tests/test_response_improvements.py +152 -152
- tests/test_retry_middleware.py +334 -242
- tests/test_retry_middleware_realistic.py +274 -0
- tests/test_scheduler.py +252 -252
- tests/test_scheduler_config_update.py +133 -133
- tests/test_simple_response.py +61 -61
- tests/test_telecom_spider_redis_key.py +205 -205
- tests/test_template_content.py +87 -87
- tests/test_template_redis_key.py +134 -134
- tests/test_tools.py +159 -159
- tests/test_user_agent_randomness.py +177 -0
- tests/test_user_agents.py +96 -96
- tests/tools_example.py +260 -260
- tests/untested_features_report.md +138 -138
- tests/verify_debug.py +51 -51
- tests/verify_distributed.py +117 -117
- tests/verify_log_fix.py +111 -111
- crawlo-1.4.2.dist-info/METADATA +0 -1199
- crawlo-1.4.2.dist-info/RECORD +0 -309
- {crawlo-1.4.2.dist-info → crawlo-1.4.3.dist-info}/WHEEL +0 -0
- {crawlo-1.4.2.dist-info → crawlo-1.4.3.dist-info}/entry_points.txt +0 -0
- {crawlo-1.4.2.dist-info → crawlo-1.4.3.dist-info}/top_level.txt +0 -0
crawlo/project.py
CHANGED
|
@@ -1,338 +1,349 @@
|
|
|
1
|
-
import os
|
|
2
|
-
import sys
|
|
3
|
-
import importlib
|
|
4
|
-
import configparser
|
|
5
|
-
from importlib import import_module
|
|
6
|
-
from inspect import iscoroutinefunction
|
|
7
|
-
from typing import Callable, Optional, Any
|
|
8
|
-
|
|
9
|
-
from crawlo.settings.setting_manager import SettingManager
|
|
10
|
-
from crawlo.utils.log import get_logger
|
|
11
|
-
|
|
12
|
-
# 使用全局logger,避免每个模块都创建自己的延迟初始化函数
|
|
13
|
-
# 延迟获取logger,确保在日志系统配置之后获取
|
|
14
|
-
_logger = None
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
def logger():
|
|
18
|
-
"""延迟获取logger实例,确保在日志系统配置之后获取"""
|
|
19
|
-
global _logger
|
|
20
|
-
if _logger is None:
|
|
21
|
-
_logger = get_logger(__name__)
|
|
22
|
-
return _logger
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
# 添加一个临时的日志函数,用于在日志系统配置之前输出信息
|
|
26
|
-
def _temp_debug(message):
|
|
27
|
-
"""临时调试函数,在日志系统配置之前使用"""
|
|
28
|
-
# 直接输出到控制台,避免循环依赖
|
|
29
|
-
import os
|
|
30
|
-
if os.environ.get('CRAWLO_DEBUG'):
|
|
31
|
-
print(f"[CRAWLO_DEBUG] {message}")
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
def load_class(path: str) -> Any:
|
|
35
|
-
"""
|
|
36
|
-
动态加载类
|
|
37
|
-
|
|
38
|
-
Args:
|
|
39
|
-
path: 类的完整路径,如 'package.module.ClassName'
|
|
40
|
-
|
|
41
|
-
Returns:
|
|
42
|
-
加载的类对象
|
|
43
|
-
"""
|
|
44
|
-
# 使用工具模块的实现,避免循环依赖
|
|
45
|
-
from crawlo.utils.class_loader import load_class as _load_class
|
|
46
|
-
return _load_class(path)
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
def merge_settings(spider, settings):
|
|
50
|
-
"""
|
|
51
|
-
合并爬虫的自定义设置到全局设置中
|
|
52
|
-
|
|
53
|
-
Args:
|
|
54
|
-
spider: 爬虫实例
|
|
55
|
-
settings: 全局设置管理器
|
|
56
|
-
"""
|
|
57
|
-
spider_name = getattr(spider, 'name', 'UnknownSpider')
|
|
58
|
-
# 检查 settings 是否为 SettingManager 实例
|
|
59
|
-
if not hasattr(settings, 'update_attributes'):
|
|
60
|
-
_temp_debug(f"merge_settings 接收到的 settings 不是 SettingManager 实例: {type(settings)}")
|
|
61
|
-
# 如果是字典,创建一个新的 SettingManager 实例
|
|
62
|
-
if isinstance(settings, dict):
|
|
63
|
-
from crawlo.settings.setting_manager import SettingManager
|
|
64
|
-
new_settings = SettingManager()
|
|
65
|
-
new_settings.update_attributes(settings)
|
|
66
|
-
settings = new_settings
|
|
67
|
-
else:
|
|
68
|
-
_temp_debug("无法处理的 settings 类型")
|
|
69
|
-
return
|
|
70
|
-
|
|
71
|
-
if hasattr(spider, 'custom_settings'):
|
|
72
|
-
custom_settings = getattr(spider, 'custom_settings')
|
|
73
|
-
settings.update_attributes(custom_settings)
|
|
74
|
-
else:
|
|
75
|
-
_temp_debug(f"爬虫 '{spider_name}' 无 custom_settings,跳过合并")
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
async def common_call(func: Callable, *args, **kwargs):
|
|
79
|
-
"""
|
|
80
|
-
通用调用函数,自动处理同步和异步函数
|
|
81
|
-
|
|
82
|
-
Args:
|
|
83
|
-
func: 要调用的函数
|
|
84
|
-
*args: 位置参数
|
|
85
|
-
**kwargs: 关键字参数
|
|
86
|
-
|
|
87
|
-
Returns:
|
|
88
|
-
函数调用结果
|
|
89
|
-
"""
|
|
90
|
-
if iscoroutinefunction(func):
|
|
91
|
-
return await func(*args, **kwargs)
|
|
92
|
-
else:
|
|
93
|
-
return func(*args, **kwargs)
|
|
94
|
-
|
|
95
|
-
|
|
96
|
-
def _get_settings_module_from_cfg(cfg_path: str) -> str:
|
|
97
|
-
"""从 crawlo.cfg 读取 settings 模块路径"""
|
|
98
|
-
config = configparser.ConfigParser()
|
|
99
|
-
try:
|
|
100
|
-
config.read(cfg_path, encoding="utf-8")
|
|
101
|
-
if config.has_section("settings") and config.has_option("settings", "default"):
|
|
102
|
-
module_path = config.get("settings", "default")
|
|
103
|
-
_temp_debug(f"📄 从 crawlo.cfg 加载 settings 模块: {module_path}")
|
|
104
|
-
return module_path
|
|
105
|
-
else:
|
|
106
|
-
raise RuntimeError(f"配置文件缺少 [settings] 或 default 选项: {cfg_path}")
|
|
107
|
-
except Exception as e:
|
|
108
|
-
raise RuntimeError(f"解析 crawlo.cfg 失败: {e}")
|
|
109
|
-
|
|
110
|
-
|
|
111
|
-
def _find_project_root(start_path: str = ".") -> Optional[str]:
|
|
112
|
-
"""
|
|
113
|
-
从指定路径向上查找项目根目录。
|
|
114
|
-
识别依据:
|
|
115
|
-
1. 存在 'crawlo.cfg'
|
|
116
|
-
2. 存在 '__init__.py' 和 'settings.py'(即 Python 包)
|
|
117
|
-
"""
|
|
118
|
-
path = os.path.abspath(start_path)
|
|
119
|
-
|
|
120
|
-
# 首先检查当前目录及其子目录
|
|
121
|
-
for root, dirs, files in os.walk(path):
|
|
122
|
-
if "crawlo.cfg" in files:
|
|
123
|
-
cfg_path = os.path.join(root, "crawlo.cfg")
|
|
124
|
-
_temp_debug(f"✅ 找到项目配置文件: {cfg_path}")
|
|
125
|
-
return root
|
|
126
|
-
|
|
127
|
-
# 向上查找直到找到 crawlo.cfg 或包含 settings.py 和 __init__.py 的目录
|
|
128
|
-
original_path = path
|
|
129
|
-
checked_paths = set()
|
|
130
|
-
|
|
131
|
-
while True:
|
|
132
|
-
# 避免无限循环
|
|
133
|
-
if path in checked_paths:
|
|
134
|
-
break
|
|
135
|
-
checked_paths.add(path)
|
|
136
|
-
|
|
137
|
-
# 检查 crawlo.cfg
|
|
138
|
-
cfg_file = os.path.join(path, "crawlo.cfg")
|
|
139
|
-
if os.path.isfile(cfg_file):
|
|
140
|
-
_temp_debug(f"✅ 找到项目配置文件: {cfg_file}")
|
|
141
|
-
return path
|
|
142
|
-
|
|
143
|
-
# 检查 settings.py 和 __init__.py
|
|
144
|
-
settings_file = os.path.join(path, "settings.py")
|
|
145
|
-
init_file = os.path.join(path, "__init__.py")
|
|
146
|
-
if os.path.isfile(settings_file) and os.path.isfile(init_file):
|
|
147
|
-
_temp_debug(f"✅ 找到项目模块: {path}")
|
|
148
|
-
# 即使找到了项目模块,也继续向上查找是否有 crawlo.cfg
|
|
149
|
-
parent = os.path.dirname(path)
|
|
150
|
-
if parent != path:
|
|
151
|
-
parent_cfg = os.path.join(parent, "crawlo.cfg")
|
|
152
|
-
if os.path.isfile(parent_cfg):
|
|
153
|
-
_temp_debug(f"✅ 在上层目录找到项目配置文件: {parent_cfg}")
|
|
154
|
-
return parent
|
|
155
|
-
return path
|
|
156
|
-
|
|
157
|
-
parent = os.path.dirname(path)
|
|
158
|
-
if parent == path:
|
|
159
|
-
break
|
|
160
|
-
path = parent
|
|
161
|
-
|
|
162
|
-
# 如果向上查找也没找到,尝试从脚本所在目录查找
|
|
163
|
-
# 获取当前脚本文件的路径
|
|
164
|
-
try:
|
|
165
|
-
script_path = os.path.dirname(os.path.abspath(sys.argv[0]))
|
|
166
|
-
if script_path != original_path:
|
|
167
|
-
path = script_path
|
|
168
|
-
checked_paths = set() # 重置已检查路径
|
|
169
|
-
while True:
|
|
170
|
-
# 避免无限循环
|
|
171
|
-
if path in checked_paths:
|
|
172
|
-
break
|
|
173
|
-
checked_paths.add(path)
|
|
174
|
-
|
|
175
|
-
cfg_file = os.path.join(path, "crawlo.cfg")
|
|
176
|
-
if os.path.isfile(cfg_file):
|
|
177
|
-
_temp_debug(f"✅ 找到项目配置文件: {cfg_file}")
|
|
178
|
-
return path
|
|
179
|
-
|
|
180
|
-
settings_file = os.path.join(path, "settings.py")
|
|
181
|
-
init_file = os.path.join(path, "__init__.py")
|
|
182
|
-
if os.path.isfile(settings_file) and os.path.isfile(init_file):
|
|
183
|
-
_temp_debug(f"✅ 找到项目模块: {path}")
|
|
184
|
-
# 即使找到了项目模块,也继续向上查找是否有 crawlo.cfg
|
|
185
|
-
parent = os.path.dirname(path)
|
|
186
|
-
if parent != path:
|
|
187
|
-
parent_cfg = os.path.join(parent, "crawlo.cfg")
|
|
188
|
-
if os.path.isfile(parent_cfg):
|
|
189
|
-
_temp_debug(f"✅ 在上层目录找到项目配置文件: {parent_cfg}")
|
|
190
|
-
return parent
|
|
191
|
-
return path
|
|
192
|
-
|
|
193
|
-
parent = os.path.dirname(path)
|
|
194
|
-
if parent == path:
|
|
195
|
-
break
|
|
196
|
-
path = parent
|
|
197
|
-
except Exception:
|
|
198
|
-
pass
|
|
199
|
-
|
|
200
|
-
# 最后尝试从当前工作目录查找
|
|
201
|
-
try:
|
|
202
|
-
cwd = os.getcwd()
|
|
203
|
-
if cwd != original_path and cwd != script_path:
|
|
204
|
-
path = cwd
|
|
205
|
-
checked_paths = set() # 重置已检查路径
|
|
206
|
-
while True:
|
|
207
|
-
# 避免无限循环
|
|
208
|
-
if path in checked_paths:
|
|
209
|
-
break
|
|
210
|
-
checked_paths.add(path)
|
|
211
|
-
|
|
212
|
-
cfg_file = os.path.join(path, "crawlo.cfg")
|
|
213
|
-
if os.path.isfile(cfg_file):
|
|
214
|
-
_temp_debug(f"找到项目配置文件: {cfg_file}")
|
|
215
|
-
return path
|
|
216
|
-
|
|
217
|
-
settings_file = os.path.join(path, "settings.py")
|
|
218
|
-
init_file = os.path.join(path, "__init__.py")
|
|
219
|
-
if os.path.isfile(settings_file) and os.path.isfile(init_file):
|
|
220
|
-
_temp_debug(f"找到项目模块: {path}")
|
|
221
|
-
# 即使找到了项目模块,也继续向上查找是否有 crawlo.cfg
|
|
222
|
-
parent = os.path.dirname(path)
|
|
223
|
-
if parent != path:
|
|
224
|
-
parent_cfg = os.path.join(parent, "crawlo.cfg")
|
|
225
|
-
if os.path.isfile(parent_cfg):
|
|
226
|
-
_temp_debug(f"在上层目录找到项目配置文件: {parent_cfg}")
|
|
227
|
-
return parent
|
|
228
|
-
return path
|
|
229
|
-
|
|
230
|
-
parent = os.path.dirname(path)
|
|
231
|
-
if parent == path:
|
|
232
|
-
break
|
|
233
|
-
path = parent
|
|
234
|
-
except Exception:
|
|
235
|
-
pass
|
|
236
|
-
|
|
237
|
-
_temp_debug("未找到 Crawlo 项目根目录。请确保在包含 'crawlo.cfg' 或 'settings.py' 的目录运行。")
|
|
238
|
-
return None
|
|
239
|
-
|
|
240
|
-
|
|
241
|
-
def _load_project_settings(custom_settings: Optional[dict] = None) -> SettingManager:
|
|
242
|
-
"""
|
|
243
|
-
内部函数:加载项目配置(不处理日志初始化)
|
|
244
|
-
这个函数专门负责配置加载逻辑,避免与初始化管理器产生循环依赖
|
|
245
|
-
|
|
246
|
-
Args:
|
|
247
|
-
custom_settings: 运行时自定义配置,会覆盖 settings.py
|
|
248
|
-
|
|
249
|
-
Returns:
|
|
250
|
-
SettingManager: 已加载配置的实例
|
|
251
|
-
"""
|
|
252
|
-
_temp_debug("🚀 正在加载 Crawlo 项目配置...")
|
|
253
|
-
|
|
254
|
-
# 1. 查找项目根
|
|
255
|
-
project_root = _find_project_root()
|
|
256
|
-
if not project_root:
|
|
257
|
-
raise RuntimeError("未找到 Crawlo 项目,请检查项目结构")
|
|
258
|
-
|
|
259
|
-
# 2. 确定 settings 模块
|
|
260
|
-
settings_module_path = None
|
|
261
|
-
cfg_file = os.path.join(project_root, "crawlo.cfg")
|
|
262
|
-
|
|
263
|
-
if os.path.isfile(cfg_file):
|
|
264
|
-
settings_module_path = _get_settings_module_from_cfg(cfg_file)
|
|
265
|
-
else:
|
|
266
|
-
# 推断:项目目录名.settings
|
|
267
|
-
project_name = os.path.basename(project_root)
|
|
268
|
-
settings_module_path = f"{project_name}.settings"
|
|
269
|
-
_temp_debug(f"⚠️ 未找到 crawlo.cfg,推断 settings 模块为: {settings_module_path}")
|
|
270
|
-
|
|
271
|
-
# 3. 注入 sys.path
|
|
272
|
-
project_root_str = os.path.abspath(project_root)
|
|
273
|
-
if project_root_str not in sys.path:
|
|
274
|
-
sys.path.insert(0, project_root_str)
|
|
275
|
-
_temp_debug(f"📁 项目根目录已加入 sys.path: {project_root_str}")
|
|
276
|
-
|
|
277
|
-
# 4. 加载 SettingManager
|
|
278
|
-
_temp_debug(f"⚙️ 正在加载配置模块: {settings_module_path}")
|
|
279
|
-
settings = SettingManager()
|
|
280
|
-
|
|
281
|
-
try:
|
|
282
|
-
settings.set_settings(settings_module_path)
|
|
283
|
-
_temp_debug("✅ settings 模块加载成功")
|
|
284
|
-
except Exception as e:
|
|
285
|
-
raise ImportError(f"加载 settings 模块失败 '{settings_module_path}': {e}")
|
|
286
|
-
|
|
287
|
-
# 5. 根据 RUN_MODE 获取相应配置
|
|
288
|
-
run_mode = settings.get('RUN_MODE', 'standalone')
|
|
289
|
-
if run_mode:
|
|
290
|
-
from crawlo.mode_manager import ModeManager
|
|
291
|
-
mode_manager = ModeManager()
|
|
292
|
-
# 获取项目名称并传递给模式配置
|
|
293
|
-
project_name = settings.get('PROJECT_NAME', 'crawlo')
|
|
294
|
-
mode_settings = mode_manager.resolve_mode_settings(run_mode, project_name=project_name)
|
|
295
|
-
|
|
296
|
-
# 特殊处理:如果用户在settings.py中明确设置了QUEUE_TYPE
|
|
297
|
-
#
|
|
298
|
-
user_queue_type = settings.get('QUEUE_TYPE')
|
|
299
|
-
if user_queue_type == '
|
|
300
|
-
|
|
301
|
-
|
|
302
|
-
|
|
303
|
-
|
|
304
|
-
#
|
|
305
|
-
|
|
306
|
-
|
|
307
|
-
|
|
308
|
-
|
|
309
|
-
|
|
310
|
-
|
|
311
|
-
|
|
312
|
-
|
|
313
|
-
|
|
314
|
-
|
|
315
|
-
|
|
316
|
-
|
|
317
|
-
|
|
318
|
-
|
|
319
|
-
|
|
320
|
-
|
|
321
|
-
|
|
322
|
-
|
|
323
|
-
|
|
324
|
-
|
|
325
|
-
|
|
326
|
-
|
|
327
|
-
|
|
328
|
-
|
|
329
|
-
|
|
330
|
-
|
|
331
|
-
|
|
332
|
-
|
|
333
|
-
|
|
334
|
-
|
|
335
|
-
|
|
336
|
-
|
|
337
|
-
|
|
338
|
-
|
|
1
|
+
import os
|
|
2
|
+
import sys
|
|
3
|
+
import importlib
|
|
4
|
+
import configparser
|
|
5
|
+
from importlib import import_module
|
|
6
|
+
from inspect import iscoroutinefunction
|
|
7
|
+
from typing import Callable, Optional, Any
|
|
8
|
+
|
|
9
|
+
from crawlo.settings.setting_manager import SettingManager
|
|
10
|
+
from crawlo.utils.log import get_logger
|
|
11
|
+
|
|
12
|
+
# 使用全局logger,避免每个模块都创建自己的延迟初始化函数
|
|
13
|
+
# 延迟获取logger,确保在日志系统配置之后获取
|
|
14
|
+
_logger = None
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
def logger():
|
|
18
|
+
"""延迟获取logger实例,确保在日志系统配置之后获取"""
|
|
19
|
+
global _logger
|
|
20
|
+
if _logger is None:
|
|
21
|
+
_logger = get_logger(__name__)
|
|
22
|
+
return _logger
|
|
23
|
+
|
|
24
|
+
|
|
25
|
+
# 添加一个临时的日志函数,用于在日志系统配置之前输出信息
|
|
26
|
+
def _temp_debug(message):
|
|
27
|
+
"""临时调试函数,在日志系统配置之前使用"""
|
|
28
|
+
# 直接输出到控制台,避免循环依赖
|
|
29
|
+
import os
|
|
30
|
+
if os.environ.get('CRAWLO_DEBUG'):
|
|
31
|
+
print(f"[CRAWLO_DEBUG] {message}")
|
|
32
|
+
|
|
33
|
+
|
|
34
|
+
def load_class(path: str) -> Any:
|
|
35
|
+
"""
|
|
36
|
+
动态加载类
|
|
37
|
+
|
|
38
|
+
Args:
|
|
39
|
+
path: 类的完整路径,如 'package.module.ClassName'
|
|
40
|
+
|
|
41
|
+
Returns:
|
|
42
|
+
加载的类对象
|
|
43
|
+
"""
|
|
44
|
+
# 使用工具模块的实现,避免循环依赖
|
|
45
|
+
from crawlo.utils.class_loader import load_class as _load_class
|
|
46
|
+
return _load_class(path)
|
|
47
|
+
|
|
48
|
+
|
|
49
|
+
def merge_settings(spider, settings):
|
|
50
|
+
"""
|
|
51
|
+
合并爬虫的自定义设置到全局设置中
|
|
52
|
+
|
|
53
|
+
Args:
|
|
54
|
+
spider: 爬虫实例
|
|
55
|
+
settings: 全局设置管理器
|
|
56
|
+
"""
|
|
57
|
+
spider_name = getattr(spider, 'name', 'UnknownSpider')
|
|
58
|
+
# 检查 settings 是否为 SettingManager 实例
|
|
59
|
+
if not hasattr(settings, 'update_attributes'):
|
|
60
|
+
_temp_debug(f"merge_settings 接收到的 settings 不是 SettingManager 实例: {type(settings)}")
|
|
61
|
+
# 如果是字典,创建一个新的 SettingManager 实例
|
|
62
|
+
if isinstance(settings, dict):
|
|
63
|
+
from crawlo.settings.setting_manager import SettingManager
|
|
64
|
+
new_settings = SettingManager()
|
|
65
|
+
new_settings.update_attributes(settings)
|
|
66
|
+
settings = new_settings
|
|
67
|
+
else:
|
|
68
|
+
_temp_debug("无法处理的 settings 类型")
|
|
69
|
+
return
|
|
70
|
+
|
|
71
|
+
if hasattr(spider, 'custom_settings'):
|
|
72
|
+
custom_settings = getattr(spider, 'custom_settings')
|
|
73
|
+
settings.update_attributes(custom_settings)
|
|
74
|
+
else:
|
|
75
|
+
_temp_debug(f"爬虫 '{spider_name}' 无 custom_settings,跳过合并")
|
|
76
|
+
|
|
77
|
+
|
|
78
|
+
async def common_call(func: Callable, *args, **kwargs):
|
|
79
|
+
"""
|
|
80
|
+
通用调用函数,自动处理同步和异步函数
|
|
81
|
+
|
|
82
|
+
Args:
|
|
83
|
+
func: 要调用的函数
|
|
84
|
+
*args: 位置参数
|
|
85
|
+
**kwargs: 关键字参数
|
|
86
|
+
|
|
87
|
+
Returns:
|
|
88
|
+
函数调用结果
|
|
89
|
+
"""
|
|
90
|
+
if iscoroutinefunction(func):
|
|
91
|
+
return await func(*args, **kwargs)
|
|
92
|
+
else:
|
|
93
|
+
return func(*args, **kwargs)
|
|
94
|
+
|
|
95
|
+
|
|
96
|
+
def _get_settings_module_from_cfg(cfg_path: str) -> str:
|
|
97
|
+
"""从 crawlo.cfg 读取 settings 模块路径"""
|
|
98
|
+
config = configparser.ConfigParser()
|
|
99
|
+
try:
|
|
100
|
+
config.read(cfg_path, encoding="utf-8")
|
|
101
|
+
if config.has_section("settings") and config.has_option("settings", "default"):
|
|
102
|
+
module_path = config.get("settings", "default")
|
|
103
|
+
_temp_debug(f"📄 从 crawlo.cfg 加载 settings 模块: {module_path}")
|
|
104
|
+
return module_path
|
|
105
|
+
else:
|
|
106
|
+
raise RuntimeError(f"配置文件缺少 [settings] 或 default 选项: {cfg_path}")
|
|
107
|
+
except Exception as e:
|
|
108
|
+
raise RuntimeError(f"解析 crawlo.cfg 失败: {e}")
|
|
109
|
+
|
|
110
|
+
|
|
111
|
+
def _find_project_root(start_path: str = ".") -> Optional[str]:
|
|
112
|
+
"""
|
|
113
|
+
从指定路径向上查找项目根目录。
|
|
114
|
+
识别依据:
|
|
115
|
+
1. 存在 'crawlo.cfg'
|
|
116
|
+
2. 存在 '__init__.py' 和 'settings.py'(即 Python 包)
|
|
117
|
+
"""
|
|
118
|
+
path = os.path.abspath(start_path)
|
|
119
|
+
|
|
120
|
+
# 首先检查当前目录及其子目录
|
|
121
|
+
for root, dirs, files in os.walk(path):
|
|
122
|
+
if "crawlo.cfg" in files:
|
|
123
|
+
cfg_path = os.path.join(root, "crawlo.cfg")
|
|
124
|
+
_temp_debug(f"✅ 找到项目配置文件: {cfg_path}")
|
|
125
|
+
return root
|
|
126
|
+
|
|
127
|
+
# 向上查找直到找到 crawlo.cfg 或包含 settings.py 和 __init__.py 的目录
|
|
128
|
+
original_path = path
|
|
129
|
+
checked_paths = set()
|
|
130
|
+
|
|
131
|
+
while True:
|
|
132
|
+
# 避免无限循环
|
|
133
|
+
if path in checked_paths:
|
|
134
|
+
break
|
|
135
|
+
checked_paths.add(path)
|
|
136
|
+
|
|
137
|
+
# 检查 crawlo.cfg
|
|
138
|
+
cfg_file = os.path.join(path, "crawlo.cfg")
|
|
139
|
+
if os.path.isfile(cfg_file):
|
|
140
|
+
_temp_debug(f"✅ 找到项目配置文件: {cfg_file}")
|
|
141
|
+
return path
|
|
142
|
+
|
|
143
|
+
# 检查 settings.py 和 __init__.py
|
|
144
|
+
settings_file = os.path.join(path, "settings.py")
|
|
145
|
+
init_file = os.path.join(path, "__init__.py")
|
|
146
|
+
if os.path.isfile(settings_file) and os.path.isfile(init_file):
|
|
147
|
+
_temp_debug(f"✅ 找到项目模块: {path}")
|
|
148
|
+
# 即使找到了项目模块,也继续向上查找是否有 crawlo.cfg
|
|
149
|
+
parent = os.path.dirname(path)
|
|
150
|
+
if parent != path:
|
|
151
|
+
parent_cfg = os.path.join(parent, "crawlo.cfg")
|
|
152
|
+
if os.path.isfile(parent_cfg):
|
|
153
|
+
_temp_debug(f"✅ 在上层目录找到项目配置文件: {parent_cfg}")
|
|
154
|
+
return parent
|
|
155
|
+
return path
|
|
156
|
+
|
|
157
|
+
parent = os.path.dirname(path)
|
|
158
|
+
if parent == path:
|
|
159
|
+
break
|
|
160
|
+
path = parent
|
|
161
|
+
|
|
162
|
+
# 如果向上查找也没找到,尝试从脚本所在目录查找
|
|
163
|
+
# 获取当前脚本文件的路径
|
|
164
|
+
try:
|
|
165
|
+
script_path = os.path.dirname(os.path.abspath(sys.argv[0]))
|
|
166
|
+
if script_path != original_path:
|
|
167
|
+
path = script_path
|
|
168
|
+
checked_paths = set() # 重置已检查路径
|
|
169
|
+
while True:
|
|
170
|
+
# 避免无限循环
|
|
171
|
+
if path in checked_paths:
|
|
172
|
+
break
|
|
173
|
+
checked_paths.add(path)
|
|
174
|
+
|
|
175
|
+
cfg_file = os.path.join(path, "crawlo.cfg")
|
|
176
|
+
if os.path.isfile(cfg_file):
|
|
177
|
+
_temp_debug(f"✅ 找到项目配置文件: {cfg_file}")
|
|
178
|
+
return path
|
|
179
|
+
|
|
180
|
+
settings_file = os.path.join(path, "settings.py")
|
|
181
|
+
init_file = os.path.join(path, "__init__.py")
|
|
182
|
+
if os.path.isfile(settings_file) and os.path.isfile(init_file):
|
|
183
|
+
_temp_debug(f"✅ 找到项目模块: {path}")
|
|
184
|
+
# 即使找到了项目模块,也继续向上查找是否有 crawlo.cfg
|
|
185
|
+
parent = os.path.dirname(path)
|
|
186
|
+
if parent != path:
|
|
187
|
+
parent_cfg = os.path.join(parent, "crawlo.cfg")
|
|
188
|
+
if os.path.isfile(parent_cfg):
|
|
189
|
+
_temp_debug(f"✅ 在上层目录找到项目配置文件: {parent_cfg}")
|
|
190
|
+
return parent
|
|
191
|
+
return path
|
|
192
|
+
|
|
193
|
+
parent = os.path.dirname(path)
|
|
194
|
+
if parent == path:
|
|
195
|
+
break
|
|
196
|
+
path = parent
|
|
197
|
+
except Exception:
|
|
198
|
+
pass
|
|
199
|
+
|
|
200
|
+
# 最后尝试从当前工作目录查找
|
|
201
|
+
try:
|
|
202
|
+
cwd = os.getcwd()
|
|
203
|
+
if cwd != original_path and cwd != script_path:
|
|
204
|
+
path = cwd
|
|
205
|
+
checked_paths = set() # 重置已检查路径
|
|
206
|
+
while True:
|
|
207
|
+
# 避免无限循环
|
|
208
|
+
if path in checked_paths:
|
|
209
|
+
break
|
|
210
|
+
checked_paths.add(path)
|
|
211
|
+
|
|
212
|
+
cfg_file = os.path.join(path, "crawlo.cfg")
|
|
213
|
+
if os.path.isfile(cfg_file):
|
|
214
|
+
_temp_debug(f"找到项目配置文件: {cfg_file}")
|
|
215
|
+
return path
|
|
216
|
+
|
|
217
|
+
settings_file = os.path.join(path, "settings.py")
|
|
218
|
+
init_file = os.path.join(path, "__init__.py")
|
|
219
|
+
if os.path.isfile(settings_file) and os.path.isfile(init_file):
|
|
220
|
+
_temp_debug(f"找到项目模块: {path}")
|
|
221
|
+
# 即使找到了项目模块,也继续向上查找是否有 crawlo.cfg
|
|
222
|
+
parent = os.path.dirname(path)
|
|
223
|
+
if parent != path:
|
|
224
|
+
parent_cfg = os.path.join(parent, "crawlo.cfg")
|
|
225
|
+
if os.path.isfile(parent_cfg):
|
|
226
|
+
_temp_debug(f"在上层目录找到项目配置文件: {parent_cfg}")
|
|
227
|
+
return parent
|
|
228
|
+
return path
|
|
229
|
+
|
|
230
|
+
parent = os.path.dirname(path)
|
|
231
|
+
if parent == path:
|
|
232
|
+
break
|
|
233
|
+
path = parent
|
|
234
|
+
except Exception:
|
|
235
|
+
pass
|
|
236
|
+
|
|
237
|
+
_temp_debug("未找到 Crawlo 项目根目录。请确保在包含 'crawlo.cfg' 或 'settings.py' 的目录运行。")
|
|
238
|
+
return None
|
|
239
|
+
|
|
240
|
+
|
|
241
|
+
def _load_project_settings(custom_settings: Optional[dict] = None) -> SettingManager:
|
|
242
|
+
"""
|
|
243
|
+
内部函数:加载项目配置(不处理日志初始化)
|
|
244
|
+
这个函数专门负责配置加载逻辑,避免与初始化管理器产生循环依赖
|
|
245
|
+
|
|
246
|
+
Args:
|
|
247
|
+
custom_settings: 运行时自定义配置,会覆盖 settings.py
|
|
248
|
+
|
|
249
|
+
Returns:
|
|
250
|
+
SettingManager: 已加载配置的实例
|
|
251
|
+
"""
|
|
252
|
+
_temp_debug("🚀 正在加载 Crawlo 项目配置...")
|
|
253
|
+
|
|
254
|
+
# 1. 查找项目根
|
|
255
|
+
project_root = _find_project_root()
|
|
256
|
+
if not project_root:
|
|
257
|
+
raise RuntimeError("未找到 Crawlo 项目,请检查项目结构")
|
|
258
|
+
|
|
259
|
+
# 2. 确定 settings 模块
|
|
260
|
+
settings_module_path = None
|
|
261
|
+
cfg_file = os.path.join(project_root, "crawlo.cfg")
|
|
262
|
+
|
|
263
|
+
if os.path.isfile(cfg_file):
|
|
264
|
+
settings_module_path = _get_settings_module_from_cfg(cfg_file)
|
|
265
|
+
else:
|
|
266
|
+
# 推断:项目目录名.settings
|
|
267
|
+
project_name = os.path.basename(project_root)
|
|
268
|
+
settings_module_path = f"{project_name}.settings"
|
|
269
|
+
_temp_debug(f"⚠️ 未找到 crawlo.cfg,推断 settings 模块为: {settings_module_path}")
|
|
270
|
+
|
|
271
|
+
# 3. 注入 sys.path
|
|
272
|
+
project_root_str = os.path.abspath(project_root)
|
|
273
|
+
if project_root_str not in sys.path:
|
|
274
|
+
sys.path.insert(0, project_root_str)
|
|
275
|
+
_temp_debug(f"📁 项目根目录已加入 sys.path: {project_root_str}")
|
|
276
|
+
|
|
277
|
+
# 4. 加载 SettingManager
|
|
278
|
+
_temp_debug(f"⚙️ 正在加载配置模块: {settings_module_path}")
|
|
279
|
+
settings = SettingManager()
|
|
280
|
+
|
|
281
|
+
try:
|
|
282
|
+
settings.set_settings(settings_module_path)
|
|
283
|
+
_temp_debug("✅ settings 模块加载成功")
|
|
284
|
+
except Exception as e:
|
|
285
|
+
raise ImportError(f"加载 settings 模块失败 '{settings_module_path}': {e}")
|
|
286
|
+
|
|
287
|
+
# 5. 根据 RUN_MODE 获取相应配置
|
|
288
|
+
run_mode = settings.get('RUN_MODE', 'standalone')
|
|
289
|
+
if run_mode:
|
|
290
|
+
from crawlo.mode_manager import ModeManager
|
|
291
|
+
mode_manager = ModeManager()
|
|
292
|
+
# 获取项目名称并传递给模式配置
|
|
293
|
+
project_name = settings.get('PROJECT_NAME', 'crawlo')
|
|
294
|
+
mode_settings = mode_manager.resolve_mode_settings(run_mode, project_name=project_name)
|
|
295
|
+
|
|
296
|
+
# 特殊处理:如果用户在settings.py中明确设置了QUEUE_TYPE,
|
|
297
|
+
# 应该尊重用户的设置,除非是standalone模式下的redis设置
|
|
298
|
+
user_queue_type = settings.get('QUEUE_TYPE')
|
|
299
|
+
if user_queue_type and run_mode == 'standalone' and user_queue_type != 'memory':
|
|
300
|
+
# 在单机模式下,如果用户明确设置了QUEUE_TYPE(且不是memory),应该保留用户的设置
|
|
301
|
+
# 但需要确保配置的一致性
|
|
302
|
+
mode_settings['QUEUE_TYPE'] = user_queue_type
|
|
303
|
+
|
|
304
|
+
# 根据QUEUE_TYPE更新其他相关配置
|
|
305
|
+
if user_queue_type == 'redis':
|
|
306
|
+
mode_settings['FILTER_CLASS'] = 'crawlo.filters.aioredis_filter.AioRedisFilter'
|
|
307
|
+
mode_settings['DEFAULT_DEDUP_PIPELINE'] = 'crawlo.pipelines.redis_dedup_pipeline.RedisDedupPipeline'
|
|
308
|
+
elif user_queue_type == 'auto':
|
|
309
|
+
mode_settings['FILTER_CLASS'] = settings.get('FILTER_CLASS', 'crawlo.filters.memory_filter.MemoryFilter')
|
|
310
|
+
mode_settings['DEFAULT_DEDUP_PIPELINE'] = settings.get('DEFAULT_DEDUP_PIPELINE', 'crawlo.pipelines.memory_dedup_pipeline.MemoryDedupPipeline')
|
|
311
|
+
|
|
312
|
+
# 合并模式配置
|
|
313
|
+
for key, value in mode_settings.items():
|
|
314
|
+
# 对于特定的配置项,模式配置应该优先于用户配置
|
|
315
|
+
# 特别是与运行模式密切相关的配置项
|
|
316
|
+
# 但如果用户明确设置了某些关键配置且与运行模式不冲突,则应保留用户设置
|
|
317
|
+
priority_keys = ['QUEUE_TYPE', 'FILTER_CLASS', 'DEFAULT_DEDUP_PIPELINE']
|
|
318
|
+
if key in priority_keys or key not in settings.attributes:
|
|
319
|
+
settings.set(key, value)
|
|
320
|
+
_temp_debug(f"🔧 已应用 {run_mode} 模式配置")
|
|
321
|
+
|
|
322
|
+
# 6. 合并运行时配置
|
|
323
|
+
if custom_settings:
|
|
324
|
+
settings.update_attributes(custom_settings)
|
|
325
|
+
_temp_debug(f"🔧 已应用运行时自定义配置: {list(custom_settings.keys())}")
|
|
326
|
+
|
|
327
|
+
_temp_debug("🎉 Crawlo 项目配置加载完成!")
|
|
328
|
+
return settings
|
|
329
|
+
|
|
330
|
+
|
|
331
|
+
def get_settings(custom_settings: Optional[dict] = None) -> SettingManager:
|
|
332
|
+
"""
|
|
333
|
+
获取配置管理器实例(主入口函数)
|
|
334
|
+
|
|
335
|
+
注意:这个函数现在作为向后兼容的入口,实际的初始化逻辑已经移到
|
|
336
|
+
crawlo.initialization 模块中。建议使用新的初始化方式:
|
|
337
|
+
|
|
338
|
+
>>> from crawlo.initialization import initialize_framework
|
|
339
|
+
>>> settings = initialize_framework(custom_settings)
|
|
340
|
+
|
|
341
|
+
Args:
|
|
342
|
+
custom_settings: 运行时自定义配置,会覆盖 settings.py
|
|
343
|
+
|
|
344
|
+
Returns:
|
|
345
|
+
SettingManager: 已加载配置的实例
|
|
346
|
+
"""
|
|
347
|
+
# 使用新的统一初始化管理器
|
|
348
|
+
from crawlo.initialization import initialize_framework
|
|
349
|
+
return initialize_framework(custom_settings)
|