crawlo 1.3.5__py3-none-any.whl → 1.3.6__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 (289) hide show
  1. crawlo/__init__.py +87 -87
  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 +45 -45
  16. crawlo/core/engine.py +439 -439
  17. crawlo/core/processor.py +40 -40
  18. crawlo/core/scheduler.py +257 -257
  19. crawlo/crawler.py +638 -638
  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 +228 -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 +39 -39
  32. crawlo/extension/health_check.py +141 -141
  33. crawlo/extension/log_interval.py +57 -57
  34. crawlo/extension/log_stats.py +81 -81
  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 -291
  47. crawlo/initialization/__init__.py +39 -39
  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 +37 -37
  58. crawlo/logging/config.py +96 -96
  59. crawlo/logging/factory.py +128 -128
  60. crawlo/logging/manager.py +111 -111
  61. crawlo/middleware/__init__.py +21 -21
  62. crawlo/middleware/default_header.py +132 -132
  63. crawlo/middleware/download_delay.py +104 -104
  64. crawlo/middleware/middleware_manager.py +135 -135
  65. crawlo/middleware/offsite.py +123 -123
  66. crawlo/middleware/proxy.py +386 -386
  67. crawlo/middleware/request_ignore.py +86 -86
  68. crawlo/middleware/response_code.py +163 -163
  69. crawlo/middleware/response_filter.py +136 -136
  70. crawlo/middleware/retry.py +124 -124
  71. crawlo/middleware/simple_proxy.py +65 -65
  72. crawlo/mode_manager.py +212 -212
  73. crawlo/network/__init__.py +21 -21
  74. crawlo/network/request.py +379 -379
  75. crawlo/network/response.py +359 -359
  76. crawlo/pipelines/__init__.py +21 -21
  77. crawlo/pipelines/bloom_dedup_pipeline.py +156 -156
  78. crawlo/pipelines/console_pipeline.py +39 -39
  79. crawlo/pipelines/csv_pipeline.py +316 -316
  80. crawlo/pipelines/database_dedup_pipeline.py +222 -222
  81. crawlo/pipelines/json_pipeline.py +218 -218
  82. crawlo/pipelines/memory_dedup_pipeline.py +115 -115
  83. crawlo/pipelines/mongo_pipeline.py +131 -131
  84. crawlo/pipelines/mysql_pipeline.py +325 -325
  85. crawlo/pipelines/pipeline_manager.py +76 -76
  86. crawlo/pipelines/redis_dedup_pipeline.py +166 -166
  87. crawlo/project.py +327 -327
  88. crawlo/queue/pqueue.py +42 -42
  89. crawlo/queue/queue_manager.py +503 -503
  90. crawlo/queue/redis_priority_queue.py +326 -326
  91. crawlo/settings/__init__.py +7 -7
  92. crawlo/settings/default_settings.py +321 -321
  93. crawlo/settings/setting_manager.py +214 -214
  94. crawlo/spider/__init__.py +657 -657
  95. crawlo/stats_collector.py +73 -73
  96. crawlo/subscriber.py +129 -129
  97. crawlo/task_manager.py +138 -138
  98. crawlo/templates/crawlo.cfg.tmpl +10 -10
  99. crawlo/templates/project/__init__.py.tmpl +3 -3
  100. crawlo/templates/project/items.py.tmpl +17 -17
  101. crawlo/templates/project/middlewares.py.tmpl +118 -118
  102. crawlo/templates/project/pipelines.py.tmpl +96 -96
  103. crawlo/templates/project/settings.py.tmpl +167 -167
  104. crawlo/templates/project/settings_distributed.py.tmpl +166 -166
  105. crawlo/templates/project/settings_gentle.py.tmpl +166 -166
  106. crawlo/templates/project/settings_high_performance.py.tmpl +167 -167
  107. crawlo/templates/project/settings_minimal.py.tmpl +65 -65
  108. crawlo/templates/project/settings_simple.py.tmpl +164 -164
  109. crawlo/templates/project/spiders/__init__.py.tmpl +9 -9
  110. crawlo/templates/run.py.tmpl +34 -34
  111. crawlo/templates/spider/spider.py.tmpl +143 -143
  112. crawlo/templates/spiders_init.py.tmpl +9 -9
  113. crawlo/tools/__init__.py +200 -200
  114. crawlo/tools/anti_crawler.py +268 -268
  115. crawlo/tools/authenticated_proxy.py +240 -240
  116. crawlo/tools/data_formatter.py +225 -225
  117. crawlo/tools/data_validator.py +180 -180
  118. crawlo/tools/date_tools.py +289 -289
  119. crawlo/tools/distributed_coordinator.py +388 -388
  120. crawlo/tools/encoding_converter.py +127 -127
  121. crawlo/tools/network_diagnostic.py +364 -364
  122. crawlo/tools/request_tools.py +82 -82
  123. crawlo/tools/retry_mechanism.py +224 -224
  124. crawlo/tools/scenario_adapter.py +262 -262
  125. crawlo/tools/text_cleaner.py +232 -232
  126. crawlo/utils/__init__.py +34 -34
  127. crawlo/utils/batch_processor.py +259 -259
  128. crawlo/utils/class_loader.py +25 -25
  129. crawlo/utils/controlled_spider_mixin.py +439 -439
  130. crawlo/utils/db_helper.py +343 -343
  131. crawlo/utils/enhanced_error_handler.py +356 -356
  132. crawlo/utils/env_config.py +142 -142
  133. crawlo/utils/error_handler.py +165 -165
  134. crawlo/utils/func_tools.py +82 -82
  135. crawlo/utils/large_scale_config.py +286 -286
  136. crawlo/utils/large_scale_helper.py +344 -344
  137. crawlo/utils/log.py +79 -79
  138. crawlo/utils/performance_monitor.py +285 -285
  139. crawlo/utils/queue_helper.py +175 -175
  140. crawlo/utils/redis_connection_pool.py +388 -388
  141. crawlo/utils/redis_key_validator.py +198 -198
  142. crawlo/utils/request.py +267 -267
  143. crawlo/utils/request_serializer.py +225 -225
  144. crawlo/utils/spider_loader.py +61 -61
  145. crawlo/utils/system.py +11 -11
  146. crawlo/utils/tools.py +4 -4
  147. crawlo/utils/url.py +39 -39
  148. {crawlo-1.3.5.dist-info → crawlo-1.3.6.dist-info}/METADATA +1126 -1126
  149. crawlo-1.3.6.dist-info/RECORD +290 -0
  150. examples/__init__.py +7 -7
  151. tests/__init__.py +7 -7
  152. tests/advanced_tools_example.py +275 -275
  153. tests/authenticated_proxy_example.py +106 -106
  154. tests/baidu_performance_test.py +108 -108
  155. tests/baidu_test.py +59 -59
  156. tests/cleaners_example.py +160 -160
  157. tests/comprehensive_framework_test.py +212 -212
  158. tests/comprehensive_test.py +81 -81
  159. tests/comprehensive_testing_summary.md +186 -186
  160. tests/config_validation_demo.py +142 -142
  161. tests/controlled_spider_example.py +205 -205
  162. tests/date_tools_example.py +180 -180
  163. tests/debug_configure.py +69 -69
  164. tests/debug_framework_logger.py +84 -84
  165. tests/debug_log_config.py +126 -126
  166. tests/debug_log_levels.py +63 -63
  167. tests/debug_pipelines.py +66 -66
  168. tests/detailed_log_test.py +233 -233
  169. tests/distributed_test.py +66 -66
  170. tests/distributed_test_debug.py +76 -76
  171. tests/dynamic_loading_example.py +523 -523
  172. tests/dynamic_loading_test.py +104 -104
  173. tests/env_config_example.py +133 -133
  174. tests/error_handling_example.py +171 -171
  175. tests/final_comprehensive_test.py +151 -151
  176. tests/final_log_test.py +260 -260
  177. tests/final_validation_test.py +182 -182
  178. tests/fix_log_test.py +142 -142
  179. tests/framework_performance_test.py +202 -202
  180. tests/log_buffering_test.py +111 -111
  181. tests/log_generation_timing_test.py +153 -153
  182. tests/optimized_performance_test.py +211 -211
  183. tests/performance_comparison.py +245 -245
  184. tests/queue_blocking_test.py +113 -113
  185. tests/queue_test.py +89 -89
  186. tests/redis_key_validation_demo.py +130 -130
  187. tests/request_params_example.py +150 -150
  188. tests/response_improvements_example.py +144 -144
  189. tests/scrapy_comparison/ofweek_scrapy.py +138 -138
  190. tests/scrapy_comparison/scrapy_test.py +133 -133
  191. tests/simple_command_test.py +119 -119
  192. tests/simple_crawlo_test.py +127 -127
  193. tests/simple_log_test.py +57 -57
  194. tests/simple_log_test2.py +137 -137
  195. tests/simple_optimization_test.py +128 -128
  196. tests/simple_queue_type_test.py +42 -0
  197. tests/simple_spider_test.py +49 -49
  198. tests/simple_test.py +47 -47
  199. tests/spider_log_timing_test.py +177 -177
  200. tests/test_advanced_tools.py +148 -148
  201. tests/test_all_commands.py +230 -230
  202. tests/test_all_redis_key_configs.py +145 -145
  203. tests/test_authenticated_proxy.py +141 -141
  204. tests/test_batch_processor.py +178 -178
  205. tests/test_cleaners.py +54 -54
  206. tests/test_component_factory.py +174 -174
  207. tests/test_comprehensive.py +146 -146
  208. tests/test_config_consistency.py +80 -80
  209. tests/test_config_merge.py +152 -152
  210. tests/test_config_validator.py +182 -182
  211. tests/test_controlled_spider_mixin.py +79 -79
  212. tests/test_crawlo_proxy_integration.py +108 -108
  213. tests/test_date_tools.py +123 -123
  214. tests/test_default_header_middleware.py +158 -158
  215. tests/test_distributed.py +65 -65
  216. tests/test_double_crawlo_fix.py +207 -207
  217. tests/test_double_crawlo_fix_simple.py +124 -124
  218. tests/test_download_delay_middleware.py +221 -221
  219. tests/test_downloader_proxy_compatibility.py +268 -268
  220. tests/test_dynamic_downloaders_proxy.py +124 -124
  221. tests/test_dynamic_proxy.py +92 -92
  222. tests/test_dynamic_proxy_config.py +146 -146
  223. tests/test_dynamic_proxy_real.py +109 -109
  224. tests/test_edge_cases.py +303 -303
  225. tests/test_enhanced_error_handler.py +270 -270
  226. tests/test_enhanced_error_handler_comprehensive.py +245 -245
  227. tests/test_env_config.py +121 -121
  228. tests/test_error_handler_compatibility.py +112 -112
  229. tests/test_factories.py +252 -252
  230. tests/test_final_validation.py +153 -153
  231. tests/test_framework_env_usage.py +103 -103
  232. tests/test_framework_logger.py +66 -66
  233. tests/test_framework_startup.py +64 -64
  234. tests/test_get_component_logger.py +83 -83
  235. tests/test_integration.py +169 -169
  236. tests/test_item_dedup_redis_key.py +122 -122
  237. tests/test_large_scale_config.py +112 -112
  238. tests/test_large_scale_helper.py +235 -235
  239. tests/test_logging_system.py +282 -282
  240. tests/test_mode_change.py +72 -72
  241. tests/test_mode_consistency.py +51 -51
  242. tests/test_offsite_middleware.py +221 -221
  243. tests/test_parsel.py +29 -29
  244. tests/test_performance.py +327 -327
  245. tests/test_performance_monitor.py +115 -115
  246. tests/test_proxy_api.py +264 -264
  247. tests/test_proxy_health_check.py +32 -32
  248. tests/test_proxy_middleware.py +121 -121
  249. tests/test_proxy_middleware_enhanced.py +216 -216
  250. tests/test_proxy_middleware_integration.py +136 -136
  251. tests/test_proxy_middleware_refactored.py +184 -184
  252. tests/test_proxy_providers.py +56 -56
  253. tests/test_proxy_stats.py +19 -19
  254. tests/test_proxy_strategies.py +59 -59
  255. tests/test_queue_empty_check.py +41 -41
  256. tests/test_queue_manager_double_crawlo.py +173 -173
  257. tests/test_queue_manager_redis_key.py +176 -176
  258. tests/test_queue_type.py +107 -0
  259. tests/test_random_user_agent.py +72 -72
  260. tests/test_real_scenario_proxy.py +195 -195
  261. tests/test_redis_config.py +28 -28
  262. tests/test_redis_connection_pool.py +294 -294
  263. tests/test_redis_key_naming.py +181 -181
  264. tests/test_redis_key_validator.py +123 -123
  265. tests/test_redis_queue.py +224 -224
  266. tests/test_request_ignore_middleware.py +182 -182
  267. tests/test_request_params.py +111 -111
  268. tests/test_request_serialization.py +70 -70
  269. tests/test_response_code_middleware.py +349 -349
  270. tests/test_response_filter_middleware.py +427 -427
  271. tests/test_response_improvements.py +152 -152
  272. tests/test_retry_middleware.py +241 -241
  273. tests/test_scheduler.py +252 -252
  274. tests/test_scheduler_config_update.py +133 -133
  275. tests/test_simple_response.py +61 -61
  276. tests/test_telecom_spider_redis_key.py +205 -205
  277. tests/test_template_content.py +87 -87
  278. tests/test_template_redis_key.py +134 -134
  279. tests/test_tools.py +159 -159
  280. tests/test_user_agents.py +96 -96
  281. tests/tools_example.py +260 -260
  282. tests/untested_features_report.md +138 -138
  283. tests/verify_debug.py +51 -51
  284. tests/verify_distributed.py +117 -117
  285. tests/verify_log_fix.py +111 -111
  286. crawlo-1.3.5.dist-info/RECORD +0 -288
  287. {crawlo-1.3.5.dist-info → crawlo-1.3.6.dist-info}/WHEEL +0 -0
  288. {crawlo-1.3.5.dist-info → crawlo-1.3.6.dist-info}/entry_points.txt +0 -0
  289. {crawlo-1.3.5.dist-info → crawlo-1.3.6.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