crawlo 1.4.1__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 -228
  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.1.dist-info/METADATA +0 -1199
  323. crawlo-1.4.1.dist-info/RECORD +0 -309
  324. {crawlo-1.4.1.dist-info → crawlo-1.4.3.dist-info}/WHEEL +0 -0
  325. {crawlo-1.4.1.dist-info → crawlo-1.4.3.dist-info}/entry_points.txt +0 -0
  326. {crawlo-1.4.1.dist-info → crawlo-1.4.3.dist-info}/top_level.txt +0 -0
crawlo/network/request.py CHANGED
@@ -1,380 +1,380 @@
1
- #!/usr/bin/python
2
- # -*- coding: UTF-8 -*-
3
- """
4
- HTTP Request 封装模块
5
- ====================
6
- 提供功能完善的HTTP请求封装,支持:
7
- - JSON/表单数据自动处理
8
- - GET请求参数处理
9
- - 优先级排序机制
10
- - 安全的深拷贝操作
11
- - 灵活的请求配置
12
- """
13
- import json
14
- from copy import deepcopy
15
- from urllib.parse import urlencode, urlparse, urlunparse, parse_qsl
16
- from w3lib.url import safe_url_string
17
- from typing import Dict, Optional, Callable, Union, Any, TypeVar, List
18
-
19
- from crawlo.utils.url import escape_ajax
20
-
21
-
22
- _Request = TypeVar("_Request", bound="Request")
23
-
24
-
25
- class RequestPriority:
26
- """请求优先级常量和工具类"""
27
- URGENT = -200 # 紧急任务
28
- HIGH = -100 # 高优先级
29
- NORMAL = 0 # 正常优先级(默认)
30
- LOW = 100 # 低优先级
31
- BACKGROUND = 200 # 后台任务
32
-
33
- @classmethod
34
- def get_all_priorities(cls) -> Dict[str, int]:
35
- """获取所有优先级常量"""
36
- return {
37
- 'URGENT': cls.URGENT,
38
- 'HIGH': cls.HIGH,
39
- 'NORMAL': cls.NORMAL,
40
- 'LOW': cls.LOW,
41
- 'BACKGROUND': cls.BACKGROUND
42
- }
43
-
44
- @classmethod
45
- def from_string(cls, priority_str: str) -> int:
46
- """从字符串获取优先级值"""
47
- priorities = cls.get_all_priorities()
48
- if priority_str.upper() not in priorities:
49
- raise ValueError(f"不支持的优先级: {priority_str}, 支持: {list(priorities.keys())}")
50
- return priorities[priority_str.upper()]
51
-
52
-
53
- class Request:
54
- """
55
- 封装一个 HTTP 请求对象,用于爬虫框架中表示一个待抓取的请求任务。
56
- 支持 JSON、表单、GET参数、原始 body 提交,自动处理 Content-Type 与编码。
57
- 不支持文件上传(multipart/form-data),保持轻量。
58
- """
59
-
60
- __slots__ = (
61
- '_url',
62
- '_meta',
63
- 'callback',
64
- 'cb_kwargs',
65
- 'err_back',
66
- 'headers',
67
- 'body',
68
- 'method',
69
- 'cookies',
70
- 'priority',
71
- 'encoding',
72
- 'dont_filter',
73
- 'timeout',
74
- 'proxy',
75
- 'allow_redirects',
76
- 'auth',
77
- 'verify',
78
- 'flags',
79
- '_json_body',
80
- '_form_data',
81
- '_params',
82
- 'use_dynamic_loader',
83
- 'dynamic_loader_options'
84
- )
85
-
86
- def __init__(
87
- self,
88
- url: str,
89
- callback: Optional[Callable] = None,
90
- method: Optional[str] = 'GET',
91
- headers: Optional[Dict[str, str]] = None,
92
- body: Optional[Union[bytes, str, Dict[Any, Any]]] = None,
93
- form_data: Optional[Dict[str, Any]] = None,
94
- json_body: Optional[Dict[str, Any]] = None,
95
- params: Optional[Dict[str, Any]] = None,
96
- cb_kwargs: Optional[Dict[str, Any]] = None,
97
- cookies: Optional[Dict[str, str]] = None,
98
- meta: Optional[Dict[str, Any]] = None,
99
- priority: int = RequestPriority.NORMAL,
100
- dont_filter: bool = False,
101
- timeout: Optional[float] = None,
102
- proxy: Optional[str] = None,
103
- allow_redirects: bool = True,
104
- auth: Optional[tuple] = None,
105
- verify: bool = True,
106
- flags: Optional[List[str]] = None,
107
- encoding: str = 'utf-8',
108
- # 动态加载相关参数
109
- use_dynamic_loader: bool = False,
110
- dynamic_loader_options: Optional[Dict[str, Any]] = None
111
- ):
112
- """
113
- 初始化请求对象。
114
-
115
- :param url: 请求 URL(必须)
116
- :param callback: 成功回调函数
117
- :param method: HTTP 方法,默认 GET
118
- :param headers: 请求头
119
- :param body: 原始请求体(bytes/str),若为 dict 且未使用 json_body/form_data,则自动转为 JSON
120
- :param form_data: 表单数据,POST请求时自动转为 application/x-www-form-urlencoded
121
- :param json_body: JSON 数据,自动序列化并设置 Content-Type
122
- :param params: GET请求参数,会自动附加到URL上
123
- :param cb_kwargs: 传递给 callback 的额外参数
124
- :param cookies: Cookies 字典
125
- :param meta: 元数据(跨中间件传递数据)
126
- :param priority: 优先级(数值越小越优先)
127
- :param dont_filter: 是否跳过去重
128
- :param timeout: 超时时间(秒)
129
- :param proxy: 代理地址,如 http://127.0.0.1:8080
130
- :param allow_redirects: 是否允许重定向
131
- :param auth: 认证元组 (username, password)
132
- :param verify: 是否验证 SSL 证书
133
- :param flags: 标记(用于调试或分类)
134
- :param encoding: 字符编码,默认 utf-8
135
- """
136
- self.callback = callback
137
- self.method = str(method).upper()
138
- self.headers = headers or {}
139
- self.cookies = cookies or {}
140
- self.priority = -priority # 用于排序:值越小优先级越高
141
-
142
- # 安全处理 meta,移除 logger 后再 deepcopy
143
- self._meta = self._safe_deepcopy_meta(meta) if meta is not None else {}
144
-
145
- self.timeout = self._meta.get('download_timeout', timeout)
146
- self.proxy = proxy
147
- self.allow_redirects = allow_redirects
148
- self.auth = auth
149
- self.verify = verify
150
- self.flags = flags or []
151
- self.encoding = encoding
152
- self.cb_kwargs = cb_kwargs or {}
153
- self.body = body
154
- # 保存高层语义参数(用于 copy)
155
- self._json_body = json_body
156
- self._form_data = form_data
157
- self._params = params
158
-
159
- # 动态加载相关属性
160
- self.use_dynamic_loader = use_dynamic_loader
161
- self.dynamic_loader_options = dynamic_loader_options or {}
162
-
163
- # 处理GET参数
164
- if params is not None and self.method == 'GET':
165
- # 将GET参数附加到URL上
166
- self._url = self._add_params_to_url(url, params)
167
- else:
168
- self._url = url
169
-
170
- # 构建 body
171
- if json_body is not None:
172
- if 'Content-Type' not in self.headers:
173
- self.headers['Content-Type'] = 'application/json'
174
- self.body = json.dumps(json_body, ensure_ascii=False).encode(encoding)
175
- if self.method == 'GET':
176
- self.method = 'POST'
177
-
178
- elif form_data is not None:
179
- if self.method == 'GET':
180
- # 对于GET请求,将form_data作为GET参数处理
181
- self._url = self._add_params_to_url(self._url, form_data)
182
- else:
183
- # 对于POST等请求,将form_data作为请求体处理
184
- if 'Content-Type' not in self.headers:
185
- self.headers['Content-Type'] = 'application/x-www-form-urlencoded'
186
- query_str = urlencode(form_data)
187
- self.body = query_str.encode(encoding) # 显式编码为 bytes
188
-
189
-
190
- else:
191
- # 处理原始 body
192
- if isinstance(self.body, dict):
193
- if 'Content-Type' not in self.headers:
194
- self.headers['Content-Type'] = 'application/json'
195
- self.body = json.dumps(self.body, ensure_ascii=False).encode(encoding)
196
- elif isinstance(self.body, str):
197
- self.body = self.body.encode(encoding)
198
-
199
- self.dont_filter = dont_filter
200
- self._set_url(self._url)
201
-
202
- @staticmethod
203
- def _add_params_to_url(url: str, params: Dict[str, Any]) -> str:
204
- """将参数添加到URL中"""
205
- if not params:
206
- return url
207
-
208
- # 解析URL
209
- parsed = urlparse(url)
210
- # 解析现有查询参数
211
- query_params = parse_qsl(parsed.query, keep_blank_values=True)
212
- # 添加新参数
213
- for key, value in params.items():
214
- query_params.append((str(key), str(value)))
215
- # 重新构建查询字符串
216
- new_query = urlencode(query_params)
217
- # 构建新的URL
218
- return urlunparse((
219
- parsed.scheme,
220
- parsed.netloc,
221
- parsed.path,
222
- parsed.params,
223
- new_query,
224
- parsed.fragment
225
- ))
226
-
227
- @staticmethod
228
- def _safe_deepcopy_meta(meta: Dict[str, Any]) -> Dict[str, Any]:
229
- """安全地 deepcopy meta,移除 logger 后再复制"""
230
- import logging
231
-
232
- def clean_logger_recursive(obj):
233
- """递归移除 logger 对象"""
234
- if isinstance(obj, logging.Logger):
235
- return None
236
- elif isinstance(obj, dict):
237
- cleaned = {}
238
- for k, v in obj.items():
239
- if not (k == 'logger' or isinstance(v, logging.Logger)):
240
- cleaned[k] = clean_logger_recursive(v)
241
- return cleaned
242
- elif isinstance(obj, (list, tuple)):
243
- cleaned_list = []
244
- for item in obj:
245
- cleaned_item = clean_logger_recursive(item)
246
- if cleaned_item is not None:
247
- cleaned_list.append(cleaned_item)
248
- return type(obj)(cleaned_list)
249
- else:
250
- return obj
251
-
252
- # 先清理 logger,再 deepcopy
253
- cleaned_meta = clean_logger_recursive(meta)
254
- return deepcopy(cleaned_meta)
255
-
256
- def copy(self: _Request) -> _Request:
257
- """
258
- 创建当前请求的副本,保留所有高层语义(json_body/form_data/params)。
259
- """
260
- return type(self)(
261
- url=self.url,
262
- callback=self.callback,
263
- method=self.method,
264
- headers=self.headers.copy(),
265
- body=None, # 由 form_data/json_body/params 重新生成
266
- form_data=self._form_data,
267
- json_body=self._json_body,
268
- params=self._params,
269
- cb_kwargs=deepcopy(self.cb_kwargs),
270
- err_back=self.err_back,
271
- cookies=self.cookies.copy(),
272
- meta=deepcopy(self._meta),
273
- priority=-self.priority,
274
- dont_filter=self.dont_filter,
275
- timeout=self.timeout,
276
- proxy=self.proxy,
277
- allow_redirects=self.allow_redirects,
278
- auth=self.auth,
279
- verify=self.verify,
280
- flags=self.flags.copy(),
281
- encoding=self.encoding,
282
- use_dynamic_loader=self.use_dynamic_loader,
283
- dynamic_loader_options=deepcopy(self.dynamic_loader_options)
284
- )
285
-
286
- def set_meta(self, key: str, value: Any) -> 'Request':
287
- """设置 meta 中的某个键值,支持链式调用。"""
288
- self._meta[key] = value
289
- return self
290
-
291
- def add_header(self, key: str, value: str) -> 'Request':
292
- """添加请求头,支持链式调用。"""
293
- self.headers[key] = value
294
- return self
295
-
296
- def add_headers(self, headers: Dict[str, str]) -> 'Request':
297
- """批量添加请求头,支持链式调用。"""
298
- self.headers.update(headers)
299
- return self
300
-
301
- def set_proxy(self, proxy: str) -> 'Request':
302
- """设置代理,支持链式调用。"""
303
- self.proxy = proxy
304
- return self
305
-
306
- def set_timeout(self, timeout: float) -> 'Request':
307
- """设置超时时间,支持链式调用。"""
308
- self.timeout = timeout
309
- return self
310
-
311
- def add_flag(self, flag: str) -> 'Request':
312
- """添加标记,支持链式调用。"""
313
- if flag not in self.flags:
314
- self.flags.append(flag)
315
- return self
316
-
317
- def remove_flag(self, flag: str) -> 'Request':
318
- """移除标记,支持链式调用。"""
319
- if flag in self.flags:
320
- self.flags.remove(flag)
321
- return self
322
-
323
- def set_dynamic_loader(self, use_dynamic: bool = True, options: Optional[Dict[str, Any]] = None) -> 'Request':
324
- """设置使用动态加载器,支持链式调用。"""
325
- self.use_dynamic_loader = use_dynamic
326
- if options:
327
- self.dynamic_loader_options = options
328
- # 同时在meta中设置标记,供混合下载器使用
329
- self._meta['use_dynamic_loader'] = use_dynamic
330
- return self
331
-
332
- def set_protocol_loader(self) -> 'Request':
333
- """强制使用协议加载器,支持链式调用。"""
334
- self.use_dynamic_loader = False
335
- self._meta['use_dynamic_loader'] = False
336
- self._meta['use_protocol_loader'] = True
337
- return self
338
-
339
- def _set_url(self, url: str) -> None:
340
- """安全设置 URL,确保格式正确。"""
341
- if not isinstance(url, str):
342
- raise TypeError(f"Request url 必须为字符串,当前类型: {type(url).__name__}")
343
-
344
- if not url.strip():
345
- raise ValueError("URL 不能为空")
346
-
347
- # 检查危险的 URL scheme
348
- dangerous_schemes = ['file://', 'ftp://', 'javascript:', 'data:']
349
- if any(url.lower().startswith(scheme) for scheme in dangerous_schemes):
350
- raise ValueError(f"URL scheme 不安全: {url[:20]}...")
351
-
352
- s = safe_url_string(url, self.encoding)
353
- escaped_url = escape_ajax(s)
354
-
355
- if not escaped_url.startswith(('http://', 'https://')):
356
- raise ValueError(f"URL 缺少 HTTP(S) scheme: {escaped_url[:50]}...")
357
-
358
- # 检查 URL 长度
359
- if len(escaped_url) > 8192: # 大多数服务器支持的最大 URL 长度
360
- raise ValueError(f"URL 过长 (超过 8192 字符): {len(escaped_url)} 字符")
361
-
362
- self._url = escaped_url
363
-
364
- @property
365
- def url(self) -> str:
366
- return self._url
367
-
368
- @property
369
- def meta(self) -> Dict[str, Any]:
370
- return self._meta
371
-
372
- def __str__(self) -> str:
373
- return f'<Request url={self.url} method={self.method}>'
374
-
375
- def __repr__(self) -> str:
376
- return str(self)
377
-
378
- def __lt__(self, other: _Request) -> bool:
379
- """用于按优先级排序"""
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ HTTP Request 封装模块
5
+ ====================
6
+ 提供功能完善的HTTP请求封装,支持:
7
+ - JSON/表单数据自动处理
8
+ - GET请求参数处理
9
+ - 优先级排序机制
10
+ - 安全的深拷贝操作
11
+ - 灵活的请求配置
12
+ """
13
+ import json
14
+ from copy import deepcopy
15
+ from urllib.parse import urlencode, urlparse, urlunparse, parse_qsl
16
+ from w3lib.url import safe_url_string
17
+ from typing import Dict, Optional, Callable, Union, Any, TypeVar, List
18
+
19
+ from crawlo.utils.url import escape_ajax
20
+
21
+
22
+ _Request = TypeVar("_Request", bound="Request")
23
+
24
+
25
+ class RequestPriority:
26
+ """请求优先级常量和工具类"""
27
+ URGENT = -200 # 紧急任务
28
+ HIGH = -100 # 高优先级
29
+ NORMAL = 0 # 正常优先级(默认)
30
+ LOW = 100 # 低优先级
31
+ BACKGROUND = 200 # 后台任务
32
+
33
+ @classmethod
34
+ def get_all_priorities(cls) -> Dict[str, int]:
35
+ """获取所有优先级常量"""
36
+ return {
37
+ 'URGENT': cls.URGENT,
38
+ 'HIGH': cls.HIGH,
39
+ 'NORMAL': cls.NORMAL,
40
+ 'LOW': cls.LOW,
41
+ 'BACKGROUND': cls.BACKGROUND
42
+ }
43
+
44
+ @classmethod
45
+ def from_string(cls, priority_str: str) -> int:
46
+ """从字符串获取优先级值"""
47
+ priorities = cls.get_all_priorities()
48
+ if priority_str.upper() not in priorities:
49
+ raise ValueError(f"不支持的优先级: {priority_str}, 支持: {list(priorities.keys())}")
50
+ return priorities[priority_str.upper()]
51
+
52
+
53
+ class Request:
54
+ """
55
+ 封装一个 HTTP 请求对象,用于爬虫框架中表示一个待抓取的请求任务。
56
+ 支持 JSON、表单、GET参数、原始 body 提交,自动处理 Content-Type 与编码。
57
+ 不支持文件上传(multipart/form-data),保持轻量。
58
+ """
59
+
60
+ __slots__ = (
61
+ '_url',
62
+ '_meta',
63
+ 'callback',
64
+ 'cb_kwargs',
65
+ 'err_back',
66
+ 'headers',
67
+ 'body',
68
+ 'method',
69
+ 'cookies',
70
+ 'priority',
71
+ 'encoding',
72
+ 'dont_filter',
73
+ 'timeout',
74
+ 'proxy',
75
+ 'allow_redirects',
76
+ 'auth',
77
+ 'verify',
78
+ 'flags',
79
+ '_json_body',
80
+ '_form_data',
81
+ '_params',
82
+ 'use_dynamic_loader',
83
+ 'dynamic_loader_options'
84
+ )
85
+
86
+ def __init__(
87
+ self,
88
+ url: str,
89
+ callback: Optional[Callable] = None,
90
+ method: Optional[str] = 'GET',
91
+ headers: Optional[Dict[str, str]] = None,
92
+ body: Optional[Union[bytes, str, Dict[Any, Any]]] = None,
93
+ form_data: Optional[Dict[str, Any]] = None,
94
+ json_body: Optional[Dict[str, Any]] = None,
95
+ params: Optional[Dict[str, Any]] = None,
96
+ cb_kwargs: Optional[Dict[str, Any]] = None,
97
+ cookies: Optional[Dict[str, str]] = None,
98
+ meta: Optional[Dict[str, Any]] = None,
99
+ priority: int = RequestPriority.NORMAL,
100
+ dont_filter: bool = False,
101
+ timeout: Optional[float] = None,
102
+ proxy: Optional[str] = None,
103
+ allow_redirects: bool = True,
104
+ auth: Optional[tuple] = None,
105
+ verify: bool = True,
106
+ flags: Optional[List[str]] = None,
107
+ encoding: str = 'utf-8',
108
+ # 动态加载相关参数
109
+ use_dynamic_loader: bool = False,
110
+ dynamic_loader_options: Optional[Dict[str, Any]] = None
111
+ ):
112
+ """
113
+ 初始化请求对象。
114
+
115
+ :param url: 请求 URL(必须)
116
+ :param callback: 成功回调函数
117
+ :param method: HTTP 方法,默认 GET
118
+ :param headers: 请求头
119
+ :param body: 原始请求体(bytes/str),若为 dict 且未使用 json_body/form_data,则自动转为 JSON
120
+ :param form_data: 表单数据,POST请求时自动转为 application/x-www-form-urlencoded
121
+ :param json_body: JSON 数据,自动序列化并设置 Content-Type
122
+ :param params: GET请求参数,会自动附加到URL上
123
+ :param cb_kwargs: 传递给 callback 的额外参数
124
+ :param cookies: Cookies 字典
125
+ :param meta: 元数据(跨中间件传递数据)
126
+ :param priority: 优先级(数值越小越优先)
127
+ :param dont_filter: 是否跳过去重
128
+ :param timeout: 超时时间(秒)
129
+ :param proxy: 代理地址,如 http://127.0.0.1:8080
130
+ :param allow_redirects: 是否允许重定向
131
+ :param auth: 认证元组 (username, password)
132
+ :param verify: 是否验证 SSL 证书
133
+ :param flags: 标记(用于调试或分类)
134
+ :param encoding: 字符编码,默认 utf-8
135
+ """
136
+ self.callback = callback
137
+ self.method = str(method).upper()
138
+ self.headers = headers or {}
139
+ self.cookies = cookies or {}
140
+ self.priority = -priority # 用于排序:值越小优先级越高
141
+
142
+ # 安全处理 meta,移除 logger 后再 deepcopy
143
+ self._meta = self._safe_deepcopy_meta(meta) if meta is not None else {}
144
+
145
+ self.timeout = self._meta.get('download_timeout', timeout)
146
+ self.proxy = proxy
147
+ self.allow_redirects = allow_redirects
148
+ self.auth = auth
149
+ self.verify = verify
150
+ self.flags = flags or []
151
+ self.encoding = encoding
152
+ self.cb_kwargs = cb_kwargs or {}
153
+ self.body = body
154
+ # 保存高层语义参数(用于 copy)
155
+ self._json_body = json_body
156
+ self._form_data = form_data
157
+ self._params = params
158
+
159
+ # 动态加载相关属性
160
+ self.use_dynamic_loader = use_dynamic_loader
161
+ self.dynamic_loader_options = dynamic_loader_options or {}
162
+
163
+ # 处理GET参数
164
+ if params is not None and self.method == 'GET':
165
+ # 将GET参数附加到URL上
166
+ self._url = self._add_params_to_url(url, params)
167
+ else:
168
+ self._url = url
169
+
170
+ # 构建 body
171
+ if json_body is not None:
172
+ if 'Content-Type' not in self.headers:
173
+ self.headers['Content-Type'] = 'application/json'
174
+ self.body = json.dumps(json_body, ensure_ascii=False).encode(encoding)
175
+ if self.method == 'GET':
176
+ self.method = 'POST'
177
+
178
+ elif form_data is not None:
179
+ if self.method == 'GET':
180
+ # 对于GET请求,将form_data作为GET参数处理
181
+ self._url = self._add_params_to_url(self._url, form_data)
182
+ else:
183
+ # 对于POST等请求,将form_data作为请求体处理
184
+ if 'Content-Type' not in self.headers:
185
+ self.headers['Content-Type'] = 'application/x-www-form-urlencoded'
186
+ query_str = urlencode(form_data)
187
+ self.body = query_str.encode(encoding) # 显式编码为 bytes
188
+
189
+
190
+ else:
191
+ # 处理原始 body
192
+ if isinstance(self.body, dict):
193
+ if 'Content-Type' not in self.headers:
194
+ self.headers['Content-Type'] = 'application/json'
195
+ self.body = json.dumps(self.body, ensure_ascii=False).encode(encoding)
196
+ elif isinstance(self.body, str):
197
+ self.body = self.body.encode(encoding)
198
+
199
+ self.dont_filter = dont_filter
200
+ self._set_url(self._url)
201
+
202
+ @staticmethod
203
+ def _add_params_to_url(url: str, params: Dict[str, Any]) -> str:
204
+ """将参数添加到URL中"""
205
+ if not params:
206
+ return url
207
+
208
+ # 解析URL
209
+ parsed = urlparse(url)
210
+ # 解析现有查询参数
211
+ query_params = parse_qsl(parsed.query, keep_blank_values=True)
212
+ # 添加新参数
213
+ for key, value in params.items():
214
+ query_params.append((str(key), str(value)))
215
+ # 重新构建查询字符串
216
+ new_query = urlencode(query_params)
217
+ # 构建新的URL
218
+ return urlunparse((
219
+ parsed.scheme,
220
+ parsed.netloc,
221
+ parsed.path,
222
+ parsed.params,
223
+ new_query,
224
+ parsed.fragment
225
+ ))
226
+
227
+ @staticmethod
228
+ def _safe_deepcopy_meta(meta: Dict[str, Any]) -> Dict[str, Any]:
229
+ """安全地 deepcopy meta,移除 logger 后再复制"""
230
+ import logging
231
+
232
+ def clean_logger_recursive(obj):
233
+ """递归移除 logger 对象"""
234
+ if isinstance(obj, logging.Logger):
235
+ return None
236
+ elif isinstance(obj, dict):
237
+ cleaned = {}
238
+ for k, v in obj.items():
239
+ if not (k == 'logger' or isinstance(v, logging.Logger)):
240
+ cleaned[k] = clean_logger_recursive(v)
241
+ return cleaned
242
+ elif isinstance(obj, (list, tuple)):
243
+ cleaned_list = []
244
+ for item in obj:
245
+ cleaned_item = clean_logger_recursive(item)
246
+ if cleaned_item is not None:
247
+ cleaned_list.append(cleaned_item)
248
+ return type(obj)(cleaned_list)
249
+ else:
250
+ return obj
251
+
252
+ # 先清理 logger,再 deepcopy
253
+ cleaned_meta = clean_logger_recursive(meta)
254
+ return deepcopy(cleaned_meta)
255
+
256
+ def copy(self: _Request) -> _Request:
257
+ """
258
+ 创建当前请求的副本,保留所有高层语义(json_body/form_data/params)。
259
+ """
260
+ return type(self)(
261
+ url=self.url,
262
+ callback=self.callback,
263
+ method=self.method,
264
+ headers=self.headers.copy(),
265
+ body=None, # 由 form_data/json_body/params 重新生成
266
+ form_data=self._form_data,
267
+ json_body=self._json_body,
268
+ params=self._params,
269
+ cb_kwargs=deepcopy(self.cb_kwargs),
270
+ err_back=self.err_back,
271
+ cookies=self.cookies.copy(),
272
+ meta=deepcopy(self._meta),
273
+ priority=-self.priority,
274
+ dont_filter=self.dont_filter,
275
+ timeout=self.timeout,
276
+ proxy=self.proxy,
277
+ allow_redirects=self.allow_redirects,
278
+ auth=self.auth,
279
+ verify=self.verify,
280
+ flags=self.flags.copy(),
281
+ encoding=self.encoding,
282
+ use_dynamic_loader=self.use_dynamic_loader,
283
+ dynamic_loader_options=deepcopy(self.dynamic_loader_options)
284
+ )
285
+
286
+ def set_meta(self, key: str, value: Any) -> 'Request':
287
+ """设置 meta 中的某个键值,支持链式调用。"""
288
+ self._meta[key] = value
289
+ return self
290
+
291
+ def add_header(self, key: str, value: str) -> 'Request':
292
+ """添加请求头,支持链式调用。"""
293
+ self.headers[key] = value
294
+ return self
295
+
296
+ def add_headers(self, headers: Dict[str, str]) -> 'Request':
297
+ """批量添加请求头,支持链式调用。"""
298
+ self.headers.update(headers)
299
+ return self
300
+
301
+ def set_proxy(self, proxy: str) -> 'Request':
302
+ """设置代理,支持链式调用。"""
303
+ self.proxy = proxy
304
+ return self
305
+
306
+ def set_timeout(self, timeout: float) -> 'Request':
307
+ """设置超时时间,支持链式调用。"""
308
+ self.timeout = timeout
309
+ return self
310
+
311
+ def add_flag(self, flag: str) -> 'Request':
312
+ """添加标记,支持链式调用。"""
313
+ if flag not in self.flags:
314
+ self.flags.append(flag)
315
+ return self
316
+
317
+ def remove_flag(self, flag: str) -> 'Request':
318
+ """移除标记,支持链式调用。"""
319
+ if flag in self.flags:
320
+ self.flags.remove(flag)
321
+ return self
322
+
323
+ def set_dynamic_loader(self, use_dynamic: bool = True, options: Optional[Dict[str, Any]] = None) -> 'Request':
324
+ """设置使用动态加载器,支持链式调用。"""
325
+ self.use_dynamic_loader = use_dynamic
326
+ if options:
327
+ self.dynamic_loader_options = options
328
+ # 同时在meta中设置标记,供混合下载器使用
329
+ self._meta['use_dynamic_loader'] = use_dynamic
330
+ return self
331
+
332
+ def set_protocol_loader(self) -> 'Request':
333
+ """强制使用协议加载器,支持链式调用。"""
334
+ self.use_dynamic_loader = False
335
+ self._meta['use_dynamic_loader'] = False
336
+ self._meta['use_protocol_loader'] = True
337
+ return self
338
+
339
+ def _set_url(self, url: str) -> None:
340
+ """安全设置 URL,确保格式正确。"""
341
+ if not isinstance(url, str):
342
+ raise TypeError(f"Request url 必须为字符串,当前类型: {type(url).__name__}")
343
+
344
+ if not url.strip():
345
+ raise ValueError("URL 不能为空")
346
+
347
+ # 检查危险的 URL scheme
348
+ dangerous_schemes = ['file://', 'ftp://', 'javascript:', 'data:']
349
+ if any(url.lower().startswith(scheme) for scheme in dangerous_schemes):
350
+ raise ValueError(f"URL scheme 不安全: {url[:20]}...")
351
+
352
+ s = safe_url_string(url, self.encoding)
353
+ escaped_url = escape_ajax(s)
354
+
355
+ if not escaped_url.startswith(('http://', 'https://')):
356
+ raise ValueError(f"URL 缺少 HTTP(S) scheme: {escaped_url[:50]}...")
357
+
358
+ # 检查 URL 长度
359
+ if len(escaped_url) > 8192: # 大多数服务器支持的最大 URL 长度
360
+ raise ValueError(f"URL 过长 (超过 8192 字符): {len(escaped_url)} 字符")
361
+
362
+ self._url = escaped_url
363
+
364
+ @property
365
+ def url(self) -> str:
366
+ return self._url
367
+
368
+ @property
369
+ def meta(self) -> Dict[str, Any]:
370
+ return self._meta
371
+
372
+ def __str__(self) -> str:
373
+ return f'<Request url={self.url} method={self.method}>'
374
+
375
+ def __repr__(self) -> str:
376
+ return str(self)
377
+
378
+ def __lt__(self, other: _Request) -> bool:
379
+ """用于按优先级排序"""
380
380
  return self.priority < other.priority