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.

Files changed (326) hide show
  1. crawlo/__init__.py +93 -93
  2. crawlo/__version__.py +1 -1
  3. crawlo/cli.py +75 -75
  4. crawlo/commands/__init__.py +14 -14
  5. crawlo/commands/check.py +594 -594
  6. crawlo/commands/genspider.py +151 -151
  7. crawlo/commands/help.py +138 -138
  8. crawlo/commands/list.py +155 -155
  9. crawlo/commands/run.py +341 -341
  10. crawlo/commands/startproject.py +436 -436
  11. crawlo/commands/stats.py +187 -187
  12. crawlo/commands/utils.py +196 -196
  13. crawlo/config.py +312 -312
  14. crawlo/config_validator.py +277 -277
  15. crawlo/core/__init__.py +52 -52
  16. crawlo/core/engine.py +438 -439
  17. crawlo/core/processor.py +47 -47
  18. crawlo/core/scheduler.py +291 -257
  19. crawlo/crawler.py +650 -650
  20. crawlo/data/__init__.py +5 -5
  21. crawlo/data/user_agents.py +194 -194
  22. crawlo/downloader/__init__.py +273 -273
  23. crawlo/downloader/aiohttp_downloader.py +233 -233
  24. crawlo/downloader/cffi_downloader.py +245 -245
  25. crawlo/downloader/httpx_downloader.py +259 -259
  26. crawlo/downloader/hybrid_downloader.py +212 -212
  27. crawlo/downloader/playwright_downloader.py +402 -402
  28. crawlo/downloader/selenium_downloader.py +472 -472
  29. crawlo/event.py +11 -11
  30. crawlo/exceptions.py +81 -81
  31. crawlo/extension/__init__.py +63 -63
  32. crawlo/extension/health_check.py +141 -141
  33. crawlo/extension/log_interval.py +94 -94
  34. crawlo/extension/log_stats.py +70 -70
  35. crawlo/extension/logging_extension.py +61 -61
  36. crawlo/extension/memory_monitor.py +104 -104
  37. crawlo/extension/performance_profiler.py +133 -133
  38. crawlo/extension/request_recorder.py +107 -107
  39. crawlo/factories/__init__.py +27 -27
  40. crawlo/factories/base.py +68 -68
  41. crawlo/factories/crawler.py +103 -103
  42. crawlo/factories/registry.py +84 -84
  43. crawlo/filters/__init__.py +154 -154
  44. crawlo/filters/aioredis_filter.py +257 -257
  45. crawlo/filters/memory_filter.py +269 -269
  46. crawlo/framework.py +292 -292
  47. crawlo/initialization/__init__.py +44 -44
  48. crawlo/initialization/built_in.py +425 -425
  49. crawlo/initialization/context.py +141 -141
  50. crawlo/initialization/core.py +193 -193
  51. crawlo/initialization/phases.py +148 -148
  52. crawlo/initialization/registry.py +145 -145
  53. crawlo/items/__init__.py +23 -23
  54. crawlo/items/base.py +23 -23
  55. crawlo/items/fields.py +52 -52
  56. crawlo/items/items.py +104 -104
  57. crawlo/logging/__init__.py +45 -37
  58. crawlo/logging/async_handler.py +181 -0
  59. crawlo/logging/config.py +196 -96
  60. crawlo/logging/factory.py +171 -128
  61. crawlo/logging/manager.py +111 -111
  62. crawlo/logging/monitor.py +153 -0
  63. crawlo/logging/sampler.py +167 -0
  64. crawlo/middleware/__init__.py +21 -21
  65. crawlo/middleware/default_header.py +132 -132
  66. crawlo/middleware/download_delay.py +104 -104
  67. crawlo/middleware/middleware_manager.py +135 -135
  68. crawlo/middleware/offsite.py +123 -123
  69. crawlo/middleware/proxy.py +386 -386
  70. crawlo/middleware/request_ignore.py +86 -86
  71. crawlo/middleware/response_code.py +150 -150
  72. crawlo/middleware/response_filter.py +136 -136
  73. crawlo/middleware/retry.py +124 -124
  74. crawlo/middleware/simple_proxy.py +65 -65
  75. crawlo/mode_manager.py +219 -219
  76. crawlo/network/__init__.py +21 -21
  77. crawlo/network/request.py +379 -379
  78. crawlo/network/response.py +359 -359
  79. crawlo/pipelines/__init__.py +21 -21
  80. crawlo/pipelines/bloom_dedup_pipeline.py +146 -146
  81. crawlo/pipelines/console_pipeline.py +39 -39
  82. crawlo/pipelines/csv_pipeline.py +316 -316
  83. crawlo/pipelines/database_dedup_pipeline.py +197 -197
  84. crawlo/pipelines/json_pipeline.py +218 -218
  85. crawlo/pipelines/memory_dedup_pipeline.py +105 -105
  86. crawlo/pipelines/mongo_pipeline.py +131 -131
  87. crawlo/pipelines/mysql_pipeline.py +325 -325
  88. crawlo/pipelines/pipeline_manager.py +100 -84
  89. crawlo/pipelines/redis_dedup_pipeline.py +156 -156
  90. crawlo/project.py +349 -338
  91. crawlo/queue/pqueue.py +42 -42
  92. crawlo/queue/queue_manager.py +526 -522
  93. crawlo/queue/redis_priority_queue.py +370 -367
  94. crawlo/settings/__init__.py +7 -7
  95. crawlo/settings/default_settings.py +284 -284
  96. crawlo/settings/setting_manager.py +219 -219
  97. crawlo/spider/__init__.py +657 -657
  98. crawlo/stats_collector.py +73 -73
  99. crawlo/subscriber.py +129 -129
  100. crawlo/task_manager.py +138 -138
  101. crawlo/templates/crawlo.cfg.tmpl +10 -10
  102. crawlo/templates/project/__init__.py.tmpl +3 -3
  103. crawlo/templates/project/items.py.tmpl +17 -17
  104. crawlo/templates/project/middlewares.py.tmpl +118 -118
  105. crawlo/templates/project/pipelines.py.tmpl +96 -96
  106. crawlo/templates/project/settings.py.tmpl +170 -170
  107. crawlo/templates/project/settings_distributed.py.tmpl +169 -169
  108. crawlo/templates/project/settings_gentle.py.tmpl +166 -166
  109. crawlo/templates/project/settings_high_performance.py.tmpl +167 -167
  110. crawlo/templates/project/settings_minimal.py.tmpl +65 -65
  111. crawlo/templates/project/settings_simple.py.tmpl +164 -164
  112. crawlo/templates/project/spiders/__init__.py.tmpl +9 -9
  113. crawlo/templates/run.py.tmpl +34 -34
  114. crawlo/templates/spider/spider.py.tmpl +143 -143
  115. crawlo/templates/spiders_init.py.tmpl +9 -9
  116. crawlo/tools/__init__.py +200 -200
  117. crawlo/tools/anti_crawler.py +268 -268
  118. crawlo/tools/authenticated_proxy.py +240 -240
  119. crawlo/tools/data_formatter.py +225 -225
  120. crawlo/tools/data_validator.py +180 -180
  121. crawlo/tools/date_tools.py +289 -289
  122. crawlo/tools/distributed_coordinator.py +384 -384
  123. crawlo/tools/encoding_converter.py +127 -127
  124. crawlo/tools/network_diagnostic.py +364 -364
  125. crawlo/tools/request_tools.py +82 -82
  126. crawlo/tools/retry_mechanism.py +224 -224
  127. crawlo/tools/scenario_adapter.py +262 -262
  128. crawlo/tools/text_cleaner.py +232 -232
  129. crawlo/utils/__init__.py +34 -34
  130. crawlo/utils/batch_processor.py +259 -259
  131. crawlo/utils/class_loader.py +25 -25
  132. crawlo/utils/controlled_spider_mixin.py +439 -439
  133. crawlo/utils/db_helper.py +343 -343
  134. crawlo/utils/enhanced_error_handler.py +356 -356
  135. crawlo/utils/env_config.py +142 -142
  136. crawlo/utils/error_handler.py +165 -165
  137. crawlo/utils/fingerprint.py +122 -122
  138. crawlo/utils/func_tools.py +82 -82
  139. crawlo/utils/large_scale_config.py +286 -286
  140. crawlo/utils/large_scale_helper.py +344 -344
  141. crawlo/utils/log.py +79 -79
  142. crawlo/utils/performance_monitor.py +285 -285
  143. crawlo/utils/queue_helper.py +175 -175
  144. crawlo/utils/redis_connection_pool.py +388 -388
  145. crawlo/utils/redis_key_validator.py +198 -198
  146. crawlo/utils/request.py +267 -267
  147. crawlo/utils/request_serializer.py +225 -225
  148. crawlo/utils/spider_loader.py +61 -61
  149. crawlo/utils/system.py +11 -11
  150. crawlo/utils/tools.py +4 -4
  151. crawlo/utils/url.py +39 -39
  152. crawlo-1.4.3.dist-info/METADATA +190 -0
  153. crawlo-1.4.3.dist-info/RECORD +326 -0
  154. examples/__init__.py +7 -7
  155. examples/test_project/__init__.py +7 -7
  156. examples/test_project/run.py +34 -34
  157. examples/test_project/test_project/__init__.py +3 -3
  158. examples/test_project/test_project/items.py +17 -17
  159. examples/test_project/test_project/middlewares.py +118 -118
  160. examples/test_project/test_project/pipelines.py +96 -96
  161. examples/test_project/test_project/settings.py +169 -169
  162. examples/test_project/test_project/spiders/__init__.py +9 -9
  163. examples/test_project/test_project/spiders/of_week_dis.py +143 -143
  164. tests/__init__.py +7 -7
  165. tests/advanced_tools_example.py +275 -275
  166. tests/authenticated_proxy_example.py +106 -106
  167. tests/baidu_performance_test.py +108 -108
  168. tests/baidu_test.py +59 -59
  169. tests/cleaners_example.py +160 -160
  170. tests/comprehensive_framework_test.py +212 -212
  171. tests/comprehensive_test.py +81 -81
  172. tests/comprehensive_testing_summary.md +186 -186
  173. tests/config_validation_demo.py +142 -142
  174. tests/controlled_spider_example.py +205 -205
  175. tests/date_tools_example.py +180 -180
  176. tests/debug_configure.py +69 -69
  177. tests/debug_framework_logger.py +84 -84
  178. tests/debug_log_config.py +126 -126
  179. tests/debug_log_levels.py +63 -63
  180. tests/debug_pipelines.py +66 -66
  181. tests/detailed_log_test.py +233 -233
  182. tests/distributed_test.py +66 -66
  183. tests/distributed_test_debug.py +76 -76
  184. tests/dynamic_loading_example.py +523 -523
  185. tests/dynamic_loading_test.py +104 -104
  186. tests/env_config_example.py +133 -133
  187. tests/error_handling_example.py +171 -171
  188. tests/final_comprehensive_test.py +151 -151
  189. tests/final_log_test.py +260 -260
  190. tests/final_validation_test.py +182 -182
  191. tests/fix_log_test.py +142 -142
  192. tests/framework_performance_test.py +202 -202
  193. tests/log_buffering_test.py +111 -111
  194. tests/log_generation_timing_test.py +153 -153
  195. tests/optimized_performance_test.py +211 -211
  196. tests/performance_comparison.py +245 -245
  197. tests/queue_blocking_test.py +113 -113
  198. tests/queue_test.py +89 -89
  199. tests/redis_key_validation_demo.py +130 -130
  200. tests/request_params_example.py +150 -150
  201. tests/response_improvements_example.py +144 -144
  202. tests/scrapy_comparison/ofweek_scrapy.py +138 -138
  203. tests/scrapy_comparison/scrapy_test.py +133 -133
  204. tests/simple_command_test.py +119 -119
  205. tests/simple_crawlo_test.py +127 -127
  206. tests/simple_log_test.py +57 -57
  207. tests/simple_log_test2.py +137 -137
  208. tests/simple_optimization_test.py +128 -128
  209. tests/simple_queue_type_test.py +41 -41
  210. tests/simple_spider_test.py +49 -49
  211. tests/simple_test.py +47 -47
  212. tests/spider_log_timing_test.py +177 -177
  213. tests/test_advanced_tools.py +148 -148
  214. tests/test_all_commands.py +230 -230
  215. tests/test_all_pipeline_fingerprints.py +133 -133
  216. tests/test_all_redis_key_configs.py +145 -145
  217. tests/test_authenticated_proxy.py +141 -141
  218. tests/test_batch_processor.py +178 -178
  219. tests/test_cleaners.py +54 -54
  220. tests/test_component_factory.py +174 -174
  221. tests/test_comprehensive.py +146 -146
  222. tests/test_config_consistency.py +80 -80
  223. tests/test_config_merge.py +152 -152
  224. tests/test_config_validator.py +182 -182
  225. tests/test_controlled_spider_mixin.py +79 -79
  226. tests/test_crawlo_proxy_integration.py +108 -108
  227. tests/test_date_tools.py +123 -123
  228. tests/test_dedup_fix.py +220 -220
  229. tests/test_dedup_pipeline_consistency.py +125 -0
  230. tests/test_default_header_middleware.py +313 -313
  231. tests/test_distributed.py +65 -65
  232. tests/test_double_crawlo_fix.py +204 -204
  233. tests/test_double_crawlo_fix_simple.py +124 -124
  234. tests/test_download_delay_middleware.py +221 -221
  235. tests/test_downloader_proxy_compatibility.py +268 -268
  236. tests/test_dynamic_downloaders_proxy.py +124 -124
  237. tests/test_dynamic_proxy.py +92 -92
  238. tests/test_dynamic_proxy_config.py +146 -146
  239. tests/test_dynamic_proxy_real.py +109 -109
  240. tests/test_edge_cases.py +303 -303
  241. tests/test_enhanced_error_handler.py +270 -270
  242. tests/test_enhanced_error_handler_comprehensive.py +245 -245
  243. tests/test_env_config.py +121 -121
  244. tests/test_error_handler_compatibility.py +112 -112
  245. tests/test_factories.py +252 -252
  246. tests/test_final_validation.py +153 -153
  247. tests/test_fingerprint_consistency.py +135 -135
  248. tests/test_fingerprint_simple.py +51 -51
  249. tests/test_framework_env_usage.py +103 -103
  250. tests/test_framework_logger.py +66 -66
  251. tests/test_framework_startup.py +64 -64
  252. tests/test_get_component_logger.py +83 -83
  253. tests/test_hash_performance.py +99 -99
  254. tests/test_integration.py +169 -169
  255. tests/test_item_dedup_redis_key.py +122 -122
  256. tests/test_large_scale_config.py +112 -112
  257. tests/test_large_scale_helper.py +235 -235
  258. tests/test_logging_enhancements.py +375 -0
  259. tests/test_logging_final.py +185 -0
  260. tests/test_logging_integration.py +313 -0
  261. tests/test_logging_system.py +282 -282
  262. tests/test_middleware_debug.py +142 -0
  263. tests/test_mode_change.py +72 -72
  264. tests/test_mode_consistency.py +51 -51
  265. tests/test_offsite_middleware.py +244 -244
  266. tests/test_offsite_middleware_simple.py +203 -203
  267. tests/test_parsel.py +29 -29
  268. tests/test_performance.py +327 -327
  269. tests/test_performance_monitor.py +115 -115
  270. tests/test_pipeline_fingerprint_consistency.py +86 -86
  271. tests/test_priority_behavior.py +212 -0
  272. tests/test_priority_consistency.py +152 -0
  273. tests/test_priority_consistency_fixed.py +250 -0
  274. tests/test_proxy_api.py +264 -264
  275. tests/test_proxy_health_check.py +32 -32
  276. tests/test_proxy_middleware.py +121 -121
  277. tests/test_proxy_middleware_enhanced.py +216 -216
  278. tests/test_proxy_middleware_integration.py +136 -136
  279. tests/test_proxy_middleware_refactored.py +184 -184
  280. tests/test_proxy_providers.py +56 -56
  281. tests/test_proxy_stats.py +19 -19
  282. tests/test_proxy_strategies.py +59 -59
  283. tests/test_queue_empty_check.py +41 -41
  284. tests/test_queue_manager_double_crawlo.py +173 -173
  285. tests/test_queue_manager_redis_key.py +179 -179
  286. tests/test_queue_naming.py +154 -154
  287. tests/test_queue_type.py +106 -106
  288. tests/test_queue_type_redis_config_consistency.py +131 -0
  289. tests/test_random_headers_default.py +323 -0
  290. tests/test_random_headers_necessity.py +309 -0
  291. tests/test_random_user_agent.py +72 -72
  292. tests/test_real_scenario_proxy.py +195 -195
  293. tests/test_redis_config.py +28 -28
  294. tests/test_redis_connection_pool.py +294 -294
  295. tests/test_redis_key_naming.py +181 -181
  296. tests/test_redis_key_validator.py +123 -123
  297. tests/test_redis_queue.py +224 -224
  298. tests/test_redis_queue_name_fix.py +175 -175
  299. tests/test_redis_queue_type_fallback.py +130 -0
  300. tests/test_request_ignore_middleware.py +182 -182
  301. tests/test_request_params.py +111 -111
  302. tests/test_request_serialization.py +70 -70
  303. tests/test_response_code_middleware.py +349 -349
  304. tests/test_response_filter_middleware.py +427 -427
  305. tests/test_response_improvements.py +152 -152
  306. tests/test_retry_middleware.py +334 -242
  307. tests/test_retry_middleware_realistic.py +274 -0
  308. tests/test_scheduler.py +252 -252
  309. tests/test_scheduler_config_update.py +133 -133
  310. tests/test_simple_response.py +61 -61
  311. tests/test_telecom_spider_redis_key.py +205 -205
  312. tests/test_template_content.py +87 -87
  313. tests/test_template_redis_key.py +134 -134
  314. tests/test_tools.py +159 -159
  315. tests/test_user_agent_randomness.py +177 -0
  316. tests/test_user_agents.py +96 -96
  317. tests/tools_example.py +260 -260
  318. tests/untested_features_report.md +138 -138
  319. tests/verify_debug.py +51 -51
  320. tests/verify_distributed.py +117 -117
  321. tests/verify_log_fix.py +111 -111
  322. crawlo-1.4.2.dist-info/METADATA +0 -1199
  323. crawlo-1.4.2.dist-info/RECORD +0 -309
  324. {crawlo-1.4.2.dist-info → crawlo-1.4.3.dist-info}/WHEEL +0 -0
  325. {crawlo-1.4.2.dist-info → crawlo-1.4.3.dist-info}/entry_points.txt +0 -0
  326. {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为'auto'
297
- # 即使在单机模式下也应该保留这个设置
298
- user_queue_type = settings.get('QUEUE_TYPE')
299
- if user_queue_type == 'auto' and run_mode == 'standalone':
300
- mode_settings['QUEUE_TYPE'] = 'auto'
301
-
302
- # 合并模式配置
303
- for key, value in mode_settings.items():
304
- # 对于特定的配置项,模式配置应该优先于用户配置
305
- # 特别是与运行模式密切相关的配置项
306
- priority_keys = ['QUEUE_TYPE', 'FILTER_CLASS', 'DEFAULT_DEDUP_PIPELINE']
307
- if key in priority_keys or key not in settings.attributes:
308
- settings.set(key, value)
309
- _temp_debug(f"🔧 已应用 {run_mode} 模式配置")
310
-
311
- # 6. 合并运行时配置
312
- if custom_settings:
313
- settings.update_attributes(custom_settings)
314
- _temp_debug(f"🔧 已应用运行时自定义配置: {list(custom_settings.keys())}")
315
-
316
- _temp_debug("🎉 Crawlo 项目配置加载完成!")
317
- return settings
318
-
319
-
320
- def get_settings(custom_settings: Optional[dict] = None) -> SettingManager:
321
- """
322
- 获取配置管理器实例(主入口函数)
323
-
324
- 注意:这个函数现在作为向后兼容的入口,实际的初始化逻辑已经移到
325
- crawlo.initialization 模块中。建议使用新的初始化方式:
326
-
327
- >>> from crawlo.initialization import initialize_framework
328
- >>> settings = initialize_framework(custom_settings)
329
-
330
- Args:
331
- custom_settings: 运行时自定义配置,会覆盖 settings.py
332
-
333
- Returns:
334
- SettingManager: 已加载配置的实例
335
- """
336
- # 使用新的统一初始化管理器
337
- from crawlo.initialization import initialize_framework
338
- return initialize_framework(custom_settings)
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)