crawlo 1.3.1__py3-none-any.whl → 1.3.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 +63 -63
- 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 +322 -314
- 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 +2 -2
- crawlo/core/engine.py +365 -365
- crawlo/core/processor.py +40 -40
- crawlo/core/scheduler.py +256 -256
- crawlo/crawler.py +1166 -1168
- crawlo/data/__init__.py +5 -5
- crawlo/data/user_agents.py +194 -194
- crawlo/downloader/__init__.py +273 -273
- crawlo/downloader/aiohttp_downloader.py +226 -226
- 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 +39 -39
- crawlo/extension/health_check.py +141 -141
- crawlo/extension/log_interval.py +57 -57
- crawlo/extension/log_stats.py +81 -81
- crawlo/extension/logging_extension.py +52 -45
- crawlo/extension/memory_monitor.py +104 -104
- crawlo/extension/performance_profiler.py +133 -133
- crawlo/extension/request_recorder.py +107 -107
- crawlo/filters/__init__.py +154 -154
- crawlo/filters/aioredis_filter.py +234 -234
- crawlo/filters/memory_filter.py +269 -269
- crawlo/items/__init__.py +23 -23
- crawlo/items/base.py +21 -21
- crawlo/items/fields.py +52 -52
- crawlo/items/items.py +104 -104
- 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 -115
- crawlo/middleware/proxy.py +386 -386
- crawlo/middleware/request_ignore.py +86 -86
- crawlo/middleware/response_code.py +163 -163
- crawlo/middleware/response_filter.py +136 -136
- crawlo/middleware/retry.py +124 -124
- crawlo/middleware/simple_proxy.py +65 -65
- crawlo/mode_manager.py +187 -148
- 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 +156 -156
- crawlo/pipelines/console_pipeline.py +39 -39
- crawlo/pipelines/csv_pipeline.py +316 -316
- crawlo/pipelines/database_dedup_pipeline.py +222 -222
- crawlo/pipelines/json_pipeline.py +218 -218
- crawlo/pipelines/memory_dedup_pipeline.py +115 -115
- crawlo/pipelines/mongo_pipeline.py +131 -131
- crawlo/pipelines/mysql_pipeline.py +318 -318
- crawlo/pipelines/pipeline_manager.py +75 -75
- crawlo/pipelines/redis_dedup_pipeline.py +166 -166
- crawlo/project.py +325 -297
- crawlo/queue/pqueue.py +37 -37
- crawlo/queue/queue_manager.py +379 -379
- crawlo/queue/redis_priority_queue.py +306 -306
- crawlo/settings/__init__.py +7 -7
- crawlo/settings/default_settings.py +225 -225
- crawlo/settings/setting_manager.py +198 -198
- crawlo/spider/__init__.py +639 -639
- crawlo/stats_collector.py +59 -59
- crawlo/subscriber.py +129 -129
- crawlo/task_manager.py +30 -30
- 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 +266 -261
- crawlo/templates/project/settings_distributed.py.tmpl +179 -174
- crawlo/templates/project/settings_gentle.py.tmpl +60 -95
- crawlo/templates/project/settings_high_performance.py.tmpl +130 -125
- crawlo/templates/project/settings_minimal.py.tmpl +34 -29
- crawlo/templates/project/settings_simple.py.tmpl +101 -96
- crawlo/templates/project/spiders/__init__.py.tmpl +5 -5
- crawlo/templates/run.py.tmpl +38 -47
- crawlo/templates/spider/spider.py.tmpl +143 -143
- 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 +388 -388
- crawlo/tools/encoding_converter.py +127 -127
- 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/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 +123 -123
- 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 +199 -146
- crawlo/utils/performance_monitor.py +285 -285
- crawlo/utils/queue_helper.py +175 -175
- crawlo/utils/redis_connection_pool.py +351 -351
- crawlo/utils/redis_key_validator.py +198 -198
- crawlo/utils/request.py +267 -267
- crawlo/utils/request_serializer.py +218 -218
- 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.3.1.dist-info → crawlo-1.3.3.dist-info}/METADATA +1020 -1020
- crawlo-1.3.3.dist-info/RECORD +219 -0
- examples/__init__.py +7 -7
- tests/DOUBLE_CRAWLO_PREFIX_FIX_REPORT.md +81 -81
- tests/__init__.py +7 -7
- tests/advanced_tools_example.py +275 -275
- tests/authenticated_proxy_example.py +107 -107
- tests/cleaners_example.py +160 -160
- tests/config_validation_demo.py +142 -142
- tests/controlled_spider_example.py +205 -205
- tests/date_tools_example.py +180 -180
- tests/debug_pipelines.py +66 -66
- 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/redis_key_validation_demo.py +130 -130
- tests/request_params_example.py +150 -150
- tests/response_improvements_example.py +144 -144
- tests/test_advanced_tools.py +148 -148
- tests/test_all_redis_key_configs.py +145 -145
- tests/test_authenticated_proxy.py +141 -141
- tests/test_cleaners.py +54 -54
- 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_crawlo_proxy_integration.py +108 -108
- tests/test_date_tools.py +123 -123
- tests/test_default_header_middleware.py +158 -158
- tests/test_distributed.py +65 -65
- tests/test_double_crawlo_fix.py +207 -207
- 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_env_config.py +121 -121
- tests/test_error_handler_compatibility.py +112 -112
- tests/test_final_validation.py +153 -153
- tests/test_framework_env_usage.py +103 -103
- tests/test_integration.py +169 -169
- tests/test_item_dedup_redis_key.py +122 -122
- tests/test_mode_consistency.py +51 -51
- tests/test_offsite_middleware.py +221 -221
- tests/test_parsel.py +29 -29
- tests/test_performance.py +327 -327
- 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_manager_double_crawlo.py +173 -173
- tests/test_queue_manager_redis_key.py +176 -176
- 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_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 +241 -241
- 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_agents.py +96 -96
- tests/tools_example.py +260 -260
- tests/verify_distributed.py +117 -117
- crawlo-1.3.1.dist-info/RECORD +0 -219
- {crawlo-1.3.1.dist-info → crawlo-1.3.3.dist-info}/WHEEL +0 -0
- {crawlo-1.3.1.dist-info → crawlo-1.3.3.dist-info}/entry_points.txt +0 -0
- {crawlo-1.3.1.dist-info → crawlo-1.3.3.dist-info}/top_level.txt +0 -0
crawlo/core/processor.py
CHANGED
|
@@ -1,40 +1,40 @@
|
|
|
1
|
-
#!/usr/bin/python
|
|
2
|
-
# -*- coding:UTF-8 -*-
|
|
3
|
-
from asyncio import Queue
|
|
4
|
-
from typing import Union, Optional
|
|
5
|
-
|
|
6
|
-
from crawlo import Request, Item
|
|
7
|
-
from crawlo.pipelines.pipeline_manager import PipelineManager
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
class Processor(object):
|
|
11
|
-
|
|
12
|
-
def __init__(self, crawler):
|
|
13
|
-
self.crawler = crawler
|
|
14
|
-
self.queue: Queue = Queue()
|
|
15
|
-
self.pipelines: Optional[PipelineManager] = None
|
|
16
|
-
|
|
17
|
-
def open(self):
|
|
18
|
-
self.pipelines = PipelineManager.from_crawler(self.crawler)
|
|
19
|
-
|
|
20
|
-
async def process(self):
|
|
21
|
-
while not self.idle():
|
|
22
|
-
result = await self.queue.get()
|
|
23
|
-
if isinstance(result, Request):
|
|
24
|
-
await self.crawler.engine.enqueue_request(result)
|
|
25
|
-
else:
|
|
26
|
-
assert isinstance(result, Item)
|
|
27
|
-
await self._process_item(result)
|
|
28
|
-
|
|
29
|
-
async def _process_item(self, item):
|
|
30
|
-
await self.pipelines.process_item(item=item)
|
|
31
|
-
|
|
32
|
-
async def enqueue(self, output: Union[Request, Item]):
|
|
33
|
-
await self.queue.put(output)
|
|
34
|
-
await self.process()
|
|
35
|
-
|
|
36
|
-
def idle(self) -> bool:
|
|
37
|
-
return len(self) == 0
|
|
38
|
-
|
|
39
|
-
def __len__(self):
|
|
40
|
-
return self.queue.qsize()
|
|
1
|
+
#!/usr/bin/python
|
|
2
|
+
# -*- coding:UTF-8 -*-
|
|
3
|
+
from asyncio import Queue
|
|
4
|
+
from typing import Union, Optional
|
|
5
|
+
|
|
6
|
+
from crawlo import Request, Item
|
|
7
|
+
from crawlo.pipelines.pipeline_manager import PipelineManager
|
|
8
|
+
|
|
9
|
+
|
|
10
|
+
class Processor(object):
|
|
11
|
+
|
|
12
|
+
def __init__(self, crawler):
|
|
13
|
+
self.crawler = crawler
|
|
14
|
+
self.queue: Queue = Queue()
|
|
15
|
+
self.pipelines: Optional[PipelineManager] = None
|
|
16
|
+
|
|
17
|
+
def open(self):
|
|
18
|
+
self.pipelines = PipelineManager.from_crawler(self.crawler)
|
|
19
|
+
|
|
20
|
+
async def process(self):
|
|
21
|
+
while not self.idle():
|
|
22
|
+
result = await self.queue.get()
|
|
23
|
+
if isinstance(result, Request):
|
|
24
|
+
await self.crawler.engine.enqueue_request(result)
|
|
25
|
+
else:
|
|
26
|
+
assert isinstance(result, Item)
|
|
27
|
+
await self._process_item(result)
|
|
28
|
+
|
|
29
|
+
async def _process_item(self, item):
|
|
30
|
+
await self.pipelines.process_item(item=item)
|
|
31
|
+
|
|
32
|
+
async def enqueue(self, output: Union[Request, Item]):
|
|
33
|
+
await self.queue.put(output)
|
|
34
|
+
await self.process()
|
|
35
|
+
|
|
36
|
+
def idle(self) -> bool:
|
|
37
|
+
return len(self) == 0
|
|
38
|
+
|
|
39
|
+
def __len__(self):
|
|
40
|
+
return self.queue.qsize()
|
crawlo/core/scheduler.py
CHANGED
|
@@ -1,257 +1,257 @@
|
|
|
1
|
-
#!/usr/bin/python
|
|
2
|
-
# -*- coding:UTF-8 -*-
|
|
3
|
-
from typing import Optional, Callable
|
|
4
|
-
import traceback
|
|
5
|
-
|
|
6
|
-
from crawlo.utils.log import get_logger
|
|
7
|
-
from crawlo.utils.request import set_request
|
|
8
|
-
from crawlo.utils.request_serializer import RequestSerializer
|
|
9
|
-
from crawlo.utils.error_handler import ErrorHandler
|
|
10
|
-
from crawlo.queue.queue_manager import QueueManager, QueueConfig, QueueType
|
|
11
|
-
from crawlo.project import load_class, common_call
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
class Scheduler:
|
|
15
|
-
def __init__(self, crawler, dupe_filter, stats, log_level, priority):
|
|
16
|
-
self.crawler = crawler
|
|
17
|
-
self.queue_manager: Optional[QueueManager] = None
|
|
18
|
-
self.request_serializer = RequestSerializer()
|
|
19
|
-
|
|
20
|
-
self.logger = get_logger(name=self.__class__.__name__, level=log_level)
|
|
21
|
-
self.error_handler = ErrorHandler(self.__class__.__name__, log_level)
|
|
22
|
-
self.stats = stats
|
|
23
|
-
self.dupe_filter = dupe_filter
|
|
24
|
-
self.priority = priority
|
|
25
|
-
|
|
26
|
-
@classmethod
|
|
27
|
-
def create_instance(cls, crawler):
|
|
28
|
-
filter_cls = load_class(crawler.settings.get('FILTER_CLASS'))
|
|
29
|
-
o = cls(
|
|
30
|
-
crawler=crawler,
|
|
31
|
-
dupe_filter=filter_cls.create_instance(crawler),
|
|
32
|
-
stats=crawler.stats,
|
|
33
|
-
log_level=crawler.settings.get('LOG_LEVEL'),
|
|
34
|
-
priority=crawler.settings.get('DEPTH_PRIORITY')
|
|
35
|
-
)
|
|
36
|
-
return o
|
|
37
|
-
|
|
38
|
-
async def open(self):
|
|
39
|
-
"""Initialize scheduler and queue"""
|
|
40
|
-
self.logger.debug("开始初始化调度器...")
|
|
41
|
-
try:
|
|
42
|
-
# 创建队列配置
|
|
43
|
-
queue_config = QueueConfig.from_settings(self.crawler.settings)
|
|
44
|
-
|
|
45
|
-
# 创建队列管理器
|
|
46
|
-
self.queue_manager = QueueManager(queue_config)
|
|
47
|
-
|
|
48
|
-
# 初始化队列
|
|
49
|
-
needs_config_update = await self.queue_manager.initialize()
|
|
50
|
-
|
|
51
|
-
# 检查是否需要更新过滤器配置
|
|
52
|
-
updated_configs = []
|
|
53
|
-
if needs_config_update:
|
|
54
|
-
# 如果返回True,说明队列类型发生了变化,需要检查当前队列类型来决定更新方向
|
|
55
|
-
if self.queue_manager._queue_type == QueueType.REDIS:
|
|
56
|
-
self._update_filter_config_for_redis()
|
|
57
|
-
updated_configs.append("Redis")
|
|
58
|
-
else:
|
|
59
|
-
self._update_filter_config_if_needed()
|
|
60
|
-
updated_configs.append("内存")
|
|
61
|
-
else:
|
|
62
|
-
# 检查是否需要更新配置(即使队列管理器没有要求更新)
|
|
63
|
-
if self.queue_manager._queue_type == QueueType.REDIS:
|
|
64
|
-
# 检查当前过滤器是否为内存过滤器
|
|
65
|
-
current_filter_class = self.crawler.settings.get('FILTER_CLASS', '')
|
|
66
|
-
if 'memory_filter' in current_filter_class:
|
|
67
|
-
self._update_filter_config_for_redis()
|
|
68
|
-
updated_configs.append("Redis")
|
|
69
|
-
elif self.queue_manager._queue_type == QueueType.MEMORY:
|
|
70
|
-
# 检查当前过滤器是否为Redis过滤器
|
|
71
|
-
current_filter_class = self.crawler.settings.get('FILTER_CLASS', '')
|
|
72
|
-
if 'aioredis_filter' in current_filter_class or 'redis_filter' in current_filter_class:
|
|
73
|
-
self._update_filter_config_if_needed()
|
|
74
|
-
updated_configs.append("内存")
|
|
75
|
-
|
|
76
|
-
# 只有在确实需要更新配置时才重新创建过滤器实例
|
|
77
|
-
# 检查是否真的进行了配置更新
|
|
78
|
-
filter_updated = (
|
|
79
|
-
(self.queue_manager._queue_type == QueueType.REDIS and 'memory_filter' in self.crawler.settings.get('FILTER_CLASS', '')) or
|
|
80
|
-
(self.queue_manager._queue_type == QueueType.MEMORY and ('aioredis_filter' in self.crawler.settings.get('FILTER_CLASS', '') or 'redis_filter' in self.crawler.settings.get('FILTER_CLASS', '')))
|
|
81
|
-
)
|
|
82
|
-
|
|
83
|
-
if needs_config_update or filter_updated:
|
|
84
|
-
# 重新创建过滤器实例,确保使用更新后的配置
|
|
85
|
-
filter_cls = load_class(self.crawler.settings.get('FILTER_CLASS'))
|
|
86
|
-
self.dupe_filter = filter_cls.create_instance(self.crawler)
|
|
87
|
-
|
|
88
|
-
# 如果配置被更新,记录警告信息
|
|
89
|
-
if filter_updated:
|
|
90
|
-
original_mode = "distributed" if 'memory_filter' in self.crawler.settings.get('FILTER_CLASS', '') else "standalone"
|
|
91
|
-
new_mode = "standalone" if self.queue_manager._queue_type == QueueType.MEMORY else "distributed"
|
|
92
|
-
self.logger.warning(f"runtime mode inconsistency detected: switched from {original_mode} to {new_mode} mode")
|
|
93
|
-
|
|
94
|
-
# 输出关键的调度器初始化完成信息
|
|
95
|
-
status = self.queue_manager.get_status()
|
|
96
|
-
current_filter = self.crawler.settings.get('FILTER_CLASS')
|
|
97
|
-
|
|
98
|
-
# 输出启用的过滤器信息(类似Downloader的格式)
|
|
99
|
-
self.logger.info(f"enabled filters: \n {current_filter}") # 注释掉重复的日志
|
|
100
|
-
|
|
101
|
-
# 优化日志输出,将多条日志合并为1条关键信息
|
|
102
|
-
# 显示完整路径的过滤器类名
|
|
103
|
-
if self.crawler.settings.get('QUEUE_TYPE', 'memory') == 'auto' and updated_configs:
|
|
104
|
-
concurrency = self.crawler.settings.get('CONCURRENCY', 8)
|
|
105
|
-
delay = self.crawler.settings.get('DOWNLOAD_DELAY', 1.0)
|
|
106
|
-
# Change INFO level log to DEBUG level to avoid redundant output
|
|
107
|
-
self.logger.debug(f"Scheduler initialized [Queue type: {status['type']}, Status: {status['health']}, Concurrency: {concurrency}, Delay: {delay}s]")
|
|
108
|
-
else:
|
|
109
|
-
# Change INFO level log to DEBUG level to avoid redundant output
|
|
110
|
-
self.logger.debug(f"Scheduler initialized [Queue type: {status['type']}, Status: {status['health']}]")
|
|
111
|
-
except Exception as e:
|
|
112
|
-
self.logger.error(f"Scheduler initialization failed: {e}")
|
|
113
|
-
self.logger.debug(f"Detailed error information:\n{traceback.format_exc()}")
|
|
114
|
-
raise
|
|
115
|
-
|
|
116
|
-
def _update_filter_config_if_needed(self):
|
|
117
|
-
"""Update filter configuration if queue type switches to memory mode"""
|
|
118
|
-
if self.queue_manager and self.queue_manager._queue_type == QueueType.MEMORY:
|
|
119
|
-
# 检查当前过滤器是否为Redis过滤器
|
|
120
|
-
current_filter_class = self.crawler.settings.get('FILTER_CLASS', '')
|
|
121
|
-
updated_configs = []
|
|
122
|
-
|
|
123
|
-
if 'aioredis_filter' in current_filter_class or 'redis_filter' in current_filter_class:
|
|
124
|
-
# 更新为内存过滤器
|
|
125
|
-
self.crawler.settings.set('FILTER_CLASS', 'crawlo.filters.memory_filter.MemoryFilter')
|
|
126
|
-
updated_configs.append("filter")
|
|
127
|
-
|
|
128
|
-
# 检查当前去重管道是否为Redis去重管道
|
|
129
|
-
current_dedup_pipeline = self.crawler.settings.get('DEFAULT_DEDUP_PIPELINE', '')
|
|
130
|
-
if 'redis_dedup_pipeline' in current_dedup_pipeline:
|
|
131
|
-
# 更新为内存去重管道
|
|
132
|
-
self.crawler.settings.set('DEFAULT_DEDUP_PIPELINE', 'crawlo.pipelines.memory_dedup_pipeline.MemoryDedupPipeline')
|
|
133
|
-
# 同时更新PIPELINES列表中的去重管道
|
|
134
|
-
pipelines = self.crawler.settings.get('PIPELINES', [])
|
|
135
|
-
if current_dedup_pipeline in pipelines:
|
|
136
|
-
# 找到并替换Redis去重管道为内存去重管道
|
|
137
|
-
index = pipelines.index(current_dedup_pipeline)
|
|
138
|
-
pipelines[index] = 'crawlo.pipelines.memory_dedup_pipeline.MemoryDedupPipeline'
|
|
139
|
-
self.crawler.settings.set('PIPELINES', pipelines)
|
|
140
|
-
updated_configs.append("dedup pipeline")
|
|
141
|
-
|
|
142
|
-
# 合并日志输出
|
|
143
|
-
if updated_configs:
|
|
144
|
-
self.logger.debug(f"configuration updated: {', '.join(updated_configs)} -> memory mode")
|
|
145
|
-
|
|
146
|
-
def _update_filter_config_for_redis(self):
|
|
147
|
-
"""Update filter configuration to Redis implementation if queue type is Redis"""
|
|
148
|
-
if self.queue_manager and self.queue_manager._queue_type == QueueType.REDIS:
|
|
149
|
-
# 检查当前过滤器是否为内存过滤器
|
|
150
|
-
current_filter_class = self.crawler.settings.get('FILTER_CLASS', '')
|
|
151
|
-
updated_configs = []
|
|
152
|
-
|
|
153
|
-
if 'memory_filter' in current_filter_class:
|
|
154
|
-
# 更新为Redis过滤器
|
|
155
|
-
self.crawler.settings.set('FILTER_CLASS', 'crawlo.filters.aioredis_filter.AioRedisFilter')
|
|
156
|
-
updated_configs.append("filter")
|
|
157
|
-
|
|
158
|
-
# 检查当前去重管道是否为内存去重管道
|
|
159
|
-
current_dedup_pipeline = self.crawler.settings.get('DEFAULT_DEDUP_PIPELINE', '')
|
|
160
|
-
if 'memory_dedup_pipeline' in current_dedup_pipeline:
|
|
161
|
-
# 更新为Redis去重管道
|
|
162
|
-
self.crawler.settings.set('DEFAULT_DEDUP_PIPELINE', 'crawlo.pipelines.redis_dedup_pipeline.RedisDedupPipeline')
|
|
163
|
-
# 同时更新PIPELINES列表中的去重管道
|
|
164
|
-
pipelines = self.crawler.settings.get('PIPELINES', [])
|
|
165
|
-
if current_dedup_pipeline in pipelines:
|
|
166
|
-
# 找到并替换内存去重管道为Redis去重管道
|
|
167
|
-
index = pipelines.index(current_dedup_pipeline)
|
|
168
|
-
pipelines[index] = 'crawlo.pipelines.redis_dedup_pipeline.RedisDedupPipeline'
|
|
169
|
-
self.crawler.settings.set('PIPELINES', pipelines)
|
|
170
|
-
updated_configs.append("dedup pipeline")
|
|
171
|
-
|
|
172
|
-
# 合并日志输出
|
|
173
|
-
if updated_configs:
|
|
174
|
-
self.logger.info(f"configuration updated: {', '.join(updated_configs)} -> redis mode")
|
|
175
|
-
|
|
176
|
-
async def next_request(self):
|
|
177
|
-
"""Get next request"""
|
|
178
|
-
if not self.queue_manager:
|
|
179
|
-
return None
|
|
180
|
-
|
|
181
|
-
try:
|
|
182
|
-
request = await self.queue_manager.get()
|
|
183
|
-
|
|
184
|
-
# 恢复 callback(从 Redis 队列取出时)
|
|
185
|
-
if request:
|
|
186
|
-
spider = getattr(self.crawler, 'spider', None)
|
|
187
|
-
request = self.request_serializer.restore_after_deserialization(request, spider)
|
|
188
|
-
|
|
189
|
-
return request
|
|
190
|
-
except Exception as e:
|
|
191
|
-
self.error_handler.handle_error(
|
|
192
|
-
e,
|
|
193
|
-
context="Failed to get next request",
|
|
194
|
-
raise_error=False
|
|
195
|
-
)
|
|
196
|
-
return None
|
|
197
|
-
|
|
198
|
-
async def enqueue_request(self, request):
|
|
199
|
-
"""Add request to queue"""
|
|
200
|
-
if not request.dont_filter and await common_call(self.dupe_filter.requested, request):
|
|
201
|
-
self.dupe_filter.log_stats(request)
|
|
202
|
-
return False
|
|
203
|
-
|
|
204
|
-
if not self.queue_manager:
|
|
205
|
-
self.logger.error("Queue manager not initialized")
|
|
206
|
-
return False
|
|
207
|
-
|
|
208
|
-
set_request(request, self.priority)
|
|
209
|
-
|
|
210
|
-
try:
|
|
211
|
-
# 使用统一的队列接口
|
|
212
|
-
success = await self.queue_manager.put(request, priority=getattr(request, 'priority', 0))
|
|
213
|
-
|
|
214
|
-
if success:
|
|
215
|
-
self.logger.debug(f"Request enqueued successfully: {request.url}")
|
|
216
|
-
|
|
217
|
-
return success
|
|
218
|
-
except Exception as e:
|
|
219
|
-
self.error_handler.handle_error(
|
|
220
|
-
e,
|
|
221
|
-
context="Failed to enqueue request",
|
|
222
|
-
raise_error=False
|
|
223
|
-
)
|
|
224
|
-
return False
|
|
225
|
-
|
|
226
|
-
def idle(self) -> bool:
|
|
227
|
-
"""Check if queue is empty"""
|
|
228
|
-
return len(self) == 0
|
|
229
|
-
|
|
230
|
-
async def async_idle(self) -> bool:
|
|
231
|
-
"""Asynchronously check if queue is empty (more accurate)"""
|
|
232
|
-
if not self.queue_manager:
|
|
233
|
-
return True
|
|
234
|
-
# 使用队列管理器的异步empty方法
|
|
235
|
-
return await self.queue_manager.async_empty()
|
|
236
|
-
|
|
237
|
-
async def close(self):
|
|
238
|
-
"""Close scheduler"""
|
|
239
|
-
try:
|
|
240
|
-
if isinstance(closed := getattr(self.dupe_filter, 'closed', None), Callable):
|
|
241
|
-
await closed()
|
|
242
|
-
|
|
243
|
-
if self.queue_manager:
|
|
244
|
-
await self.queue_manager.close()
|
|
245
|
-
except Exception as e:
|
|
246
|
-
self.error_handler.handle_error(
|
|
247
|
-
e,
|
|
248
|
-
context="Failed to close scheduler",
|
|
249
|
-
raise_error=False
|
|
250
|
-
)
|
|
251
|
-
|
|
252
|
-
def __len__(self):
|
|
253
|
-
"""Get queue size"""
|
|
254
|
-
if not self.queue_manager:
|
|
255
|
-
return 0
|
|
256
|
-
# 返回同步的近似值,实际大小需要异步获取
|
|
1
|
+
#!/usr/bin/python
|
|
2
|
+
# -*- coding:UTF-8 -*-
|
|
3
|
+
from typing import Optional, Callable
|
|
4
|
+
import traceback
|
|
5
|
+
|
|
6
|
+
from crawlo.utils.log import get_logger
|
|
7
|
+
from crawlo.utils.request import set_request
|
|
8
|
+
from crawlo.utils.request_serializer import RequestSerializer
|
|
9
|
+
from crawlo.utils.error_handler import ErrorHandler
|
|
10
|
+
from crawlo.queue.queue_manager import QueueManager, QueueConfig, QueueType
|
|
11
|
+
from crawlo.project import load_class, common_call
|
|
12
|
+
|
|
13
|
+
|
|
14
|
+
class Scheduler:
|
|
15
|
+
def __init__(self, crawler, dupe_filter, stats, log_level, priority):
|
|
16
|
+
self.crawler = crawler
|
|
17
|
+
self.queue_manager: Optional[QueueManager] = None
|
|
18
|
+
self.request_serializer = RequestSerializer()
|
|
19
|
+
|
|
20
|
+
self.logger = get_logger(name=self.__class__.__name__, level=log_level)
|
|
21
|
+
self.error_handler = ErrorHandler(self.__class__.__name__, log_level)
|
|
22
|
+
self.stats = stats
|
|
23
|
+
self.dupe_filter = dupe_filter
|
|
24
|
+
self.priority = priority
|
|
25
|
+
|
|
26
|
+
@classmethod
|
|
27
|
+
def create_instance(cls, crawler):
|
|
28
|
+
filter_cls = load_class(crawler.settings.get('FILTER_CLASS'))
|
|
29
|
+
o = cls(
|
|
30
|
+
crawler=crawler,
|
|
31
|
+
dupe_filter=filter_cls.create_instance(crawler),
|
|
32
|
+
stats=crawler.stats,
|
|
33
|
+
log_level=crawler.settings.get('LOG_LEVEL'),
|
|
34
|
+
priority=crawler.settings.get('DEPTH_PRIORITY')
|
|
35
|
+
)
|
|
36
|
+
return o
|
|
37
|
+
|
|
38
|
+
async def open(self):
|
|
39
|
+
"""Initialize scheduler and queue"""
|
|
40
|
+
self.logger.debug("开始初始化调度器...")
|
|
41
|
+
try:
|
|
42
|
+
# 创建队列配置
|
|
43
|
+
queue_config = QueueConfig.from_settings(self.crawler.settings)
|
|
44
|
+
|
|
45
|
+
# 创建队列管理器
|
|
46
|
+
self.queue_manager = QueueManager(queue_config)
|
|
47
|
+
|
|
48
|
+
# 初始化队列
|
|
49
|
+
needs_config_update = await self.queue_manager.initialize()
|
|
50
|
+
|
|
51
|
+
# 检查是否需要更新过滤器配置
|
|
52
|
+
updated_configs = []
|
|
53
|
+
if needs_config_update:
|
|
54
|
+
# 如果返回True,说明队列类型发生了变化,需要检查当前队列类型来决定更新方向
|
|
55
|
+
if self.queue_manager._queue_type == QueueType.REDIS:
|
|
56
|
+
self._update_filter_config_for_redis()
|
|
57
|
+
updated_configs.append("Redis")
|
|
58
|
+
else:
|
|
59
|
+
self._update_filter_config_if_needed()
|
|
60
|
+
updated_configs.append("内存")
|
|
61
|
+
else:
|
|
62
|
+
# 检查是否需要更新配置(即使队列管理器没有要求更新)
|
|
63
|
+
if self.queue_manager._queue_type == QueueType.REDIS:
|
|
64
|
+
# 检查当前过滤器是否为内存过滤器
|
|
65
|
+
current_filter_class = self.crawler.settings.get('FILTER_CLASS', '')
|
|
66
|
+
if 'memory_filter' in current_filter_class:
|
|
67
|
+
self._update_filter_config_for_redis()
|
|
68
|
+
updated_configs.append("Redis")
|
|
69
|
+
elif self.queue_manager._queue_type == QueueType.MEMORY:
|
|
70
|
+
# 检查当前过滤器是否为Redis过滤器
|
|
71
|
+
current_filter_class = self.crawler.settings.get('FILTER_CLASS', '')
|
|
72
|
+
if 'aioredis_filter' in current_filter_class or 'redis_filter' in current_filter_class:
|
|
73
|
+
self._update_filter_config_if_needed()
|
|
74
|
+
updated_configs.append("内存")
|
|
75
|
+
|
|
76
|
+
# 只有在确实需要更新配置时才重新创建过滤器实例
|
|
77
|
+
# 检查是否真的进行了配置更新
|
|
78
|
+
filter_updated = (
|
|
79
|
+
(self.queue_manager._queue_type == QueueType.REDIS and 'memory_filter' in self.crawler.settings.get('FILTER_CLASS', '')) or
|
|
80
|
+
(self.queue_manager._queue_type == QueueType.MEMORY and ('aioredis_filter' in self.crawler.settings.get('FILTER_CLASS', '') or 'redis_filter' in self.crawler.settings.get('FILTER_CLASS', '')))
|
|
81
|
+
)
|
|
82
|
+
|
|
83
|
+
if needs_config_update or filter_updated:
|
|
84
|
+
# 重新创建过滤器实例,确保使用更新后的配置
|
|
85
|
+
filter_cls = load_class(self.crawler.settings.get('FILTER_CLASS'))
|
|
86
|
+
self.dupe_filter = filter_cls.create_instance(self.crawler)
|
|
87
|
+
|
|
88
|
+
# 如果配置被更新,记录警告信息
|
|
89
|
+
if filter_updated:
|
|
90
|
+
original_mode = "distributed" if 'memory_filter' in self.crawler.settings.get('FILTER_CLASS', '') else "standalone"
|
|
91
|
+
new_mode = "standalone" if self.queue_manager._queue_type == QueueType.MEMORY else "distributed"
|
|
92
|
+
self.logger.warning(f"runtime mode inconsistency detected: switched from {original_mode} to {new_mode} mode")
|
|
93
|
+
|
|
94
|
+
# 输出关键的调度器初始化完成信息
|
|
95
|
+
status = self.queue_manager.get_status()
|
|
96
|
+
current_filter = self.crawler.settings.get('FILTER_CLASS')
|
|
97
|
+
|
|
98
|
+
# 输出启用的过滤器信息(类似Downloader的格式)
|
|
99
|
+
self.logger.info(f"enabled filters: \n {current_filter}") # 注释掉重复的日志
|
|
100
|
+
|
|
101
|
+
# 优化日志输出,将多条日志合并为1条关键信息
|
|
102
|
+
# 显示完整路径的过滤器类名
|
|
103
|
+
if self.crawler.settings.get('QUEUE_TYPE', 'memory') == 'auto' and updated_configs:
|
|
104
|
+
concurrency = self.crawler.settings.get('CONCURRENCY', 8)
|
|
105
|
+
delay = self.crawler.settings.get('DOWNLOAD_DELAY', 1.0)
|
|
106
|
+
# Change INFO level log to DEBUG level to avoid redundant output
|
|
107
|
+
self.logger.debug(f"Scheduler initialized [Queue type: {status['type']}, Status: {status['health']}, Concurrency: {concurrency}, Delay: {delay}s]")
|
|
108
|
+
else:
|
|
109
|
+
# Change INFO level log to DEBUG level to avoid redundant output
|
|
110
|
+
self.logger.debug(f"Scheduler initialized [Queue type: {status['type']}, Status: {status['health']}]")
|
|
111
|
+
except Exception as e:
|
|
112
|
+
self.logger.error(f"Scheduler initialization failed: {e}")
|
|
113
|
+
self.logger.debug(f"Detailed error information:\n{traceback.format_exc()}")
|
|
114
|
+
raise
|
|
115
|
+
|
|
116
|
+
def _update_filter_config_if_needed(self):
|
|
117
|
+
"""Update filter configuration if queue type switches to memory mode"""
|
|
118
|
+
if self.queue_manager and self.queue_manager._queue_type == QueueType.MEMORY:
|
|
119
|
+
# 检查当前过滤器是否为Redis过滤器
|
|
120
|
+
current_filter_class = self.crawler.settings.get('FILTER_CLASS', '')
|
|
121
|
+
updated_configs = []
|
|
122
|
+
|
|
123
|
+
if 'aioredis_filter' in current_filter_class or 'redis_filter' in current_filter_class:
|
|
124
|
+
# 更新为内存过滤器
|
|
125
|
+
self.crawler.settings.set('FILTER_CLASS', 'crawlo.filters.memory_filter.MemoryFilter')
|
|
126
|
+
updated_configs.append("filter")
|
|
127
|
+
|
|
128
|
+
# 检查当前去重管道是否为Redis去重管道
|
|
129
|
+
current_dedup_pipeline = self.crawler.settings.get('DEFAULT_DEDUP_PIPELINE', '')
|
|
130
|
+
if 'redis_dedup_pipeline' in current_dedup_pipeline:
|
|
131
|
+
# 更新为内存去重管道
|
|
132
|
+
self.crawler.settings.set('DEFAULT_DEDUP_PIPELINE', 'crawlo.pipelines.memory_dedup_pipeline.MemoryDedupPipeline')
|
|
133
|
+
# 同时更新PIPELINES列表中的去重管道
|
|
134
|
+
pipelines = self.crawler.settings.get('PIPELINES', [])
|
|
135
|
+
if current_dedup_pipeline in pipelines:
|
|
136
|
+
# 找到并替换Redis去重管道为内存去重管道
|
|
137
|
+
index = pipelines.index(current_dedup_pipeline)
|
|
138
|
+
pipelines[index] = 'crawlo.pipelines.memory_dedup_pipeline.MemoryDedupPipeline'
|
|
139
|
+
self.crawler.settings.set('PIPELINES', pipelines)
|
|
140
|
+
updated_configs.append("dedup pipeline")
|
|
141
|
+
|
|
142
|
+
# 合并日志输出
|
|
143
|
+
if updated_configs:
|
|
144
|
+
self.logger.debug(f"configuration updated: {', '.join(updated_configs)} -> memory mode")
|
|
145
|
+
|
|
146
|
+
def _update_filter_config_for_redis(self):
|
|
147
|
+
"""Update filter configuration to Redis implementation if queue type is Redis"""
|
|
148
|
+
if self.queue_manager and self.queue_manager._queue_type == QueueType.REDIS:
|
|
149
|
+
# 检查当前过滤器是否为内存过滤器
|
|
150
|
+
current_filter_class = self.crawler.settings.get('FILTER_CLASS', '')
|
|
151
|
+
updated_configs = []
|
|
152
|
+
|
|
153
|
+
if 'memory_filter' in current_filter_class:
|
|
154
|
+
# 更新为Redis过滤器
|
|
155
|
+
self.crawler.settings.set('FILTER_CLASS', 'crawlo.filters.aioredis_filter.AioRedisFilter')
|
|
156
|
+
updated_configs.append("filter")
|
|
157
|
+
|
|
158
|
+
# 检查当前去重管道是否为内存去重管道
|
|
159
|
+
current_dedup_pipeline = self.crawler.settings.get('DEFAULT_DEDUP_PIPELINE', '')
|
|
160
|
+
if 'memory_dedup_pipeline' in current_dedup_pipeline:
|
|
161
|
+
# 更新为Redis去重管道
|
|
162
|
+
self.crawler.settings.set('DEFAULT_DEDUP_PIPELINE', 'crawlo.pipelines.redis_dedup_pipeline.RedisDedupPipeline')
|
|
163
|
+
# 同时更新PIPELINES列表中的去重管道
|
|
164
|
+
pipelines = self.crawler.settings.get('PIPELINES', [])
|
|
165
|
+
if current_dedup_pipeline in pipelines:
|
|
166
|
+
# 找到并替换内存去重管道为Redis去重管道
|
|
167
|
+
index = pipelines.index(current_dedup_pipeline)
|
|
168
|
+
pipelines[index] = 'crawlo.pipelines.redis_dedup_pipeline.RedisDedupPipeline'
|
|
169
|
+
self.crawler.settings.set('PIPELINES', pipelines)
|
|
170
|
+
updated_configs.append("dedup pipeline")
|
|
171
|
+
|
|
172
|
+
# 合并日志输出
|
|
173
|
+
if updated_configs:
|
|
174
|
+
self.logger.info(f"configuration updated: {', '.join(updated_configs)} -> redis mode")
|
|
175
|
+
|
|
176
|
+
async def next_request(self):
|
|
177
|
+
"""Get next request"""
|
|
178
|
+
if not self.queue_manager:
|
|
179
|
+
return None
|
|
180
|
+
|
|
181
|
+
try:
|
|
182
|
+
request = await self.queue_manager.get()
|
|
183
|
+
|
|
184
|
+
# 恢复 callback(从 Redis 队列取出时)
|
|
185
|
+
if request:
|
|
186
|
+
spider = getattr(self.crawler, 'spider', None)
|
|
187
|
+
request = self.request_serializer.restore_after_deserialization(request, spider)
|
|
188
|
+
|
|
189
|
+
return request
|
|
190
|
+
except Exception as e:
|
|
191
|
+
self.error_handler.handle_error(
|
|
192
|
+
e,
|
|
193
|
+
context="Failed to get next request",
|
|
194
|
+
raise_error=False
|
|
195
|
+
)
|
|
196
|
+
return None
|
|
197
|
+
|
|
198
|
+
async def enqueue_request(self, request):
|
|
199
|
+
"""Add request to queue"""
|
|
200
|
+
if not request.dont_filter and await common_call(self.dupe_filter.requested, request):
|
|
201
|
+
self.dupe_filter.log_stats(request)
|
|
202
|
+
return False
|
|
203
|
+
|
|
204
|
+
if not self.queue_manager:
|
|
205
|
+
self.logger.error("Queue manager not initialized")
|
|
206
|
+
return False
|
|
207
|
+
|
|
208
|
+
set_request(request, self.priority)
|
|
209
|
+
|
|
210
|
+
try:
|
|
211
|
+
# 使用统一的队列接口
|
|
212
|
+
success = await self.queue_manager.put(request, priority=getattr(request, 'priority', 0))
|
|
213
|
+
|
|
214
|
+
if success:
|
|
215
|
+
self.logger.debug(f"Request enqueued successfully: {request.url}")
|
|
216
|
+
|
|
217
|
+
return success
|
|
218
|
+
except Exception as e:
|
|
219
|
+
self.error_handler.handle_error(
|
|
220
|
+
e,
|
|
221
|
+
context="Failed to enqueue request",
|
|
222
|
+
raise_error=False
|
|
223
|
+
)
|
|
224
|
+
return False
|
|
225
|
+
|
|
226
|
+
def idle(self) -> bool:
|
|
227
|
+
"""Check if queue is empty"""
|
|
228
|
+
return len(self) == 0
|
|
229
|
+
|
|
230
|
+
async def async_idle(self) -> bool:
|
|
231
|
+
"""Asynchronously check if queue is empty (more accurate)"""
|
|
232
|
+
if not self.queue_manager:
|
|
233
|
+
return True
|
|
234
|
+
# 使用队列管理器的异步empty方法
|
|
235
|
+
return await self.queue_manager.async_empty()
|
|
236
|
+
|
|
237
|
+
async def close(self):
|
|
238
|
+
"""Close scheduler"""
|
|
239
|
+
try:
|
|
240
|
+
if isinstance(closed := getattr(self.dupe_filter, 'closed', None), Callable):
|
|
241
|
+
await closed()
|
|
242
|
+
|
|
243
|
+
if self.queue_manager:
|
|
244
|
+
await self.queue_manager.close()
|
|
245
|
+
except Exception as e:
|
|
246
|
+
self.error_handler.handle_error(
|
|
247
|
+
e,
|
|
248
|
+
context="Failed to close scheduler",
|
|
249
|
+
raise_error=False
|
|
250
|
+
)
|
|
251
|
+
|
|
252
|
+
def __len__(self):
|
|
253
|
+
"""Get queue size"""
|
|
254
|
+
if not self.queue_manager:
|
|
255
|
+
return 0
|
|
256
|
+
# 返回同步的近似值,实际大小需要异步获取
|
|
257
257
|
return 0 if self.queue_manager.empty() else 1
|