crawlo 1.4.5__py3-none-any.whl → 1.4.7__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 (375) hide show
  1. crawlo/__init__.py +90 -89
  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 +186 -186
  7. crawlo/commands/help.py +140 -138
  8. crawlo/commands/list.py +155 -155
  9. crawlo/commands/run.py +379 -341
  10. crawlo/commands/startproject.py +460 -460
  11. crawlo/commands/stats.py +187 -187
  12. crawlo/commands/utils.py +196 -196
  13. crawlo/config.py +320 -312
  14. crawlo/config_validator.py +277 -277
  15. crawlo/core/__init__.py +52 -52
  16. crawlo/core/engine.py +451 -438
  17. crawlo/core/processor.py +47 -47
  18. crawlo/core/scheduler.py +290 -291
  19. crawlo/crawler.py +698 -657
  20. crawlo/data/__init__.py +5 -5
  21. crawlo/data/user_agents.py +194 -194
  22. crawlo/downloader/__init__.py +280 -276
  23. crawlo/downloader/aiohttp_downloader.py +233 -233
  24. crawlo/downloader/cffi_downloader.py +250 -245
  25. crawlo/downloader/httpx_downloader.py +265 -259
  26. crawlo/downloader/hybrid_downloader.py +212 -212
  27. crawlo/downloader/playwright_downloader.py +425 -402
  28. crawlo/downloader/selenium_downloader.py +486 -472
  29. crawlo/event.py +45 -11
  30. crawlo/exceptions.py +215 -82
  31. crawlo/extension/__init__.py +65 -64
  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 +53 -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 +104 -103
  42. crawlo/factories/registry.py +84 -84
  43. crawlo/factories/utils.py +135 -0
  44. crawlo/filters/__init__.py +170 -153
  45. crawlo/filters/aioredis_filter.py +348 -264
  46. crawlo/filters/memory_filter.py +261 -276
  47. crawlo/framework.py +306 -292
  48. crawlo/initialization/__init__.py +44 -44
  49. crawlo/initialization/built_in.py +391 -434
  50. crawlo/initialization/context.py +141 -141
  51. crawlo/initialization/core.py +240 -194
  52. crawlo/initialization/phases.py +230 -149
  53. crawlo/initialization/registry.py +143 -145
  54. crawlo/initialization/utils.py +49 -0
  55. crawlo/interfaces.py +23 -23
  56. crawlo/items/__init__.py +23 -23
  57. crawlo/items/base.py +23 -23
  58. crawlo/items/fields.py +52 -52
  59. crawlo/items/items.py +104 -104
  60. crawlo/logging/__init__.py +42 -46
  61. crawlo/logging/config.py +277 -197
  62. crawlo/logging/factory.py +175 -171
  63. crawlo/logging/manager.py +104 -112
  64. crawlo/middleware/__init__.py +87 -24
  65. crawlo/middleware/default_header.py +132 -132
  66. crawlo/middleware/download_delay.py +104 -104
  67. crawlo/middleware/middleware_manager.py +142 -142
  68. crawlo/middleware/offsite.py +123 -123
  69. crawlo/middleware/proxy.py +209 -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/mode_manager.py +287 -253
  75. crawlo/network/__init__.py +21 -21
  76. crawlo/network/request.py +375 -379
  77. crawlo/network/response.py +569 -664
  78. crawlo/pipelines/__init__.py +53 -22
  79. crawlo/pipelines/base_pipeline.py +452 -0
  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 +140 -132
  87. crawlo/pipelines/mysql_pipeline.py +470 -326
  88. crawlo/pipelines/pipeline_manager.py +100 -100
  89. crawlo/pipelines/redis_dedup_pipeline.py +155 -156
  90. crawlo/project.py +347 -347
  91. crawlo/queue/__init__.py +10 -0
  92. crawlo/queue/pqueue.py +38 -38
  93. crawlo/queue/queue_manager.py +591 -525
  94. crawlo/queue/redis_priority_queue.py +519 -370
  95. crawlo/settings/__init__.py +7 -7
  96. crawlo/settings/default_settings.py +285 -270
  97. crawlo/settings/setting_manager.py +219 -219
  98. crawlo/spider/__init__.py +657 -657
  99. crawlo/stats_collector.py +82 -73
  100. crawlo/subscriber.py +129 -129
  101. crawlo/task_manager.py +138 -138
  102. crawlo/templates/crawlo.cfg.tmpl +10 -10
  103. crawlo/templates/project/__init__.py.tmpl +2 -4
  104. crawlo/templates/project/items.py.tmpl +13 -17
  105. crawlo/templates/project/middlewares.py.tmpl +38 -38
  106. crawlo/templates/project/pipelines.py.tmpl +35 -36
  107. crawlo/templates/project/settings.py.tmpl +110 -157
  108. crawlo/templates/project/settings_distributed.py.tmpl +156 -161
  109. crawlo/templates/project/settings_gentle.py.tmpl +170 -171
  110. crawlo/templates/project/settings_high_performance.py.tmpl +171 -172
  111. crawlo/templates/project/settings_minimal.py.tmpl +99 -77
  112. crawlo/templates/project/settings_simple.py.tmpl +168 -169
  113. crawlo/templates/project/spiders/__init__.py.tmpl +9 -9
  114. crawlo/templates/run.py.tmpl +23 -30
  115. crawlo/templates/spider/spider.py.tmpl +33 -144
  116. crawlo/templates/spiders_init.py.tmpl +5 -10
  117. crawlo/tools/__init__.py +86 -189
  118. crawlo/tools/date_tools.py +289 -289
  119. crawlo/tools/distributed_coordinator.py +384 -384
  120. crawlo/tools/scenario_adapter.py +262 -262
  121. crawlo/tools/text_cleaner.py +232 -232
  122. crawlo/utils/__init__.py +50 -50
  123. crawlo/utils/batch_processor.py +276 -259
  124. crawlo/utils/config_manager.py +442 -0
  125. crawlo/utils/controlled_spider_mixin.py +439 -439
  126. crawlo/utils/db_helper.py +250 -244
  127. crawlo/utils/error_handler.py +410 -410
  128. crawlo/utils/fingerprint.py +121 -121
  129. crawlo/utils/func_tools.py +82 -82
  130. crawlo/utils/large_scale_helper.py +344 -344
  131. crawlo/utils/leak_detector.py +335 -0
  132. crawlo/utils/log.py +79 -79
  133. crawlo/utils/misc.py +81 -81
  134. crawlo/utils/mongo_connection_pool.py +157 -0
  135. crawlo/utils/mysql_connection_pool.py +197 -0
  136. crawlo/utils/performance_monitor.py +285 -285
  137. crawlo/utils/queue_helper.py +175 -175
  138. crawlo/utils/redis_checker.py +91 -0
  139. crawlo/utils/redis_connection_pool.py +578 -388
  140. crawlo/utils/redis_key_validator.py +198 -198
  141. crawlo/utils/request.py +278 -256
  142. crawlo/utils/request_serializer.py +225 -225
  143. crawlo/utils/resource_manager.py +337 -0
  144. crawlo/utils/selector_helper.py +137 -137
  145. crawlo/utils/singleton.py +70 -0
  146. crawlo/utils/spider_loader.py +201 -201
  147. crawlo/utils/text_helper.py +94 -94
  148. crawlo/utils/{url.py → url_utils.py} +39 -39
  149. crawlo-1.4.7.dist-info/METADATA +689 -0
  150. crawlo-1.4.7.dist-info/RECORD +347 -0
  151. examples/__init__.py +7 -7
  152. tests/__init__.py +7 -7
  153. tests/advanced_tools_example.py +217 -275
  154. tests/authenticated_proxy_example.py +110 -106
  155. tests/baidu_performance_test.py +108 -108
  156. tests/baidu_test.py +59 -59
  157. tests/bug_check_test.py +250 -250
  158. tests/cleaners_example.py +160 -160
  159. tests/comprehensive_framework_test.py +212 -212
  160. tests/comprehensive_test.py +81 -81
  161. tests/comprehensive_testing_summary.md +186 -186
  162. tests/config_validation_demo.py +142 -142
  163. tests/controlled_spider_example.py +205 -205
  164. tests/date_tools_example.py +180 -180
  165. tests/debug_configure.py +69 -69
  166. tests/debug_framework_logger.py +84 -84
  167. tests/debug_log_config.py +126 -126
  168. tests/debug_log_levels.py +63 -63
  169. tests/debug_pipelines.py +66 -66
  170. tests/detailed_log_test.py +233 -233
  171. tests/direct_selector_helper_test.py +96 -96
  172. tests/distributed_dedup_test.py +467 -0
  173. tests/distributed_test.py +66 -66
  174. tests/distributed_test_debug.py +76 -76
  175. tests/dynamic_loading_example.py +523 -523
  176. tests/dynamic_loading_test.py +104 -104
  177. tests/error_handling_example.py +171 -171
  178. tests/explain_mysql_update_behavior.py +77 -0
  179. tests/final_comprehensive_test.py +151 -151
  180. tests/final_log_test.py +260 -260
  181. tests/final_validation_test.py +182 -182
  182. tests/fix_log_test.py +142 -142
  183. tests/framework_performance_test.py +202 -202
  184. tests/log_buffering_test.py +111 -111
  185. tests/log_generation_timing_test.py +153 -153
  186. tests/monitor_redis_dedup.sh +72 -0
  187. tests/ofweek_scrapy/ofweek_scrapy/items.py +12 -12
  188. tests/ofweek_scrapy/ofweek_scrapy/middlewares.py +100 -100
  189. tests/ofweek_scrapy/ofweek_scrapy/pipelines.py +13 -13
  190. tests/ofweek_scrapy/ofweek_scrapy/settings.py +84 -84
  191. tests/ofweek_scrapy/scrapy.cfg +11 -11
  192. tests/optimized_performance_test.py +211 -211
  193. tests/performance_comparison.py +244 -244
  194. tests/queue_blocking_test.py +113 -113
  195. tests/queue_test.py +89 -89
  196. tests/redis_key_validation_demo.py +130 -130
  197. tests/request_params_example.py +150 -150
  198. tests/response_improvements_example.py +144 -144
  199. tests/scrapy_comparison/ofweek_scrapy.py +138 -138
  200. tests/scrapy_comparison/scrapy_test.py +133 -133
  201. tests/simple_cli_test.py +55 -0
  202. tests/simple_command_test.py +119 -119
  203. tests/simple_crawlo_test.py +126 -126
  204. tests/simple_follow_test.py +38 -38
  205. tests/simple_log_test2.py +137 -137
  206. tests/simple_optimization_test.py +128 -128
  207. tests/simple_queue_type_test.py +41 -41
  208. tests/simple_response_selector_test.py +94 -94
  209. tests/simple_selector_helper_test.py +154 -154
  210. tests/simple_selector_test.py +207 -207
  211. tests/simple_spider_test.py +49 -49
  212. tests/simple_url_test.py +73 -73
  213. tests/simulate_mysql_update_test.py +140 -0
  214. tests/spider_log_timing_test.py +177 -177
  215. tests/test_advanced_tools.py +148 -148
  216. tests/test_all_commands.py +230 -230
  217. tests/test_all_pipeline_fingerprints.py +133 -133
  218. tests/test_all_redis_key_configs.py +145 -145
  219. tests/test_asyncmy_usage.py +57 -0
  220. tests/test_batch_processor.py +178 -178
  221. tests/test_cleaners.py +54 -54
  222. tests/test_cli_arguments.py +119 -0
  223. tests/test_component_factory.py +174 -174
  224. tests/test_config_consistency.py +80 -80
  225. tests/test_config_merge.py +152 -152
  226. tests/test_config_validator.py +182 -182
  227. tests/test_controlled_spider_mixin.py +79 -79
  228. tests/test_crawler_process_import.py +38 -38
  229. tests/test_crawler_process_spider_modules.py +47 -47
  230. tests/test_crawlo_proxy_integration.py +114 -108
  231. tests/test_date_tools.py +123 -123
  232. tests/test_dedup_fix.py +220 -220
  233. tests/test_dedup_pipeline_consistency.py +124 -124
  234. tests/test_default_header_middleware.py +313 -313
  235. tests/test_distributed.py +65 -65
  236. tests/test_double_crawlo_fix.py +204 -204
  237. tests/test_double_crawlo_fix_simple.py +124 -124
  238. tests/test_download_delay_middleware.py +221 -221
  239. tests/test_downloader_proxy_compatibility.py +272 -268
  240. tests/test_edge_cases.py +305 -305
  241. tests/test_encoding_core.py +56 -56
  242. tests/test_encoding_detection.py +126 -126
  243. tests/test_enhanced_error_handler.py +270 -270
  244. tests/test_enhanced_error_handler_comprehensive.py +245 -245
  245. tests/test_error_handler_compatibility.py +112 -112
  246. tests/test_factories.py +252 -252
  247. tests/test_factory_compatibility.py +196 -196
  248. tests/test_final_validation.py +153 -153
  249. tests/test_fingerprint_consistency.py +135 -135
  250. tests/test_fingerprint_simple.py +51 -51
  251. tests/test_get_component_logger.py +83 -83
  252. tests/test_hash_performance.py +99 -99
  253. tests/test_integration.py +169 -169
  254. tests/test_item_dedup_redis_key.py +122 -122
  255. tests/test_large_scale_helper.py +235 -235
  256. tests/test_logging_enhancements.py +374 -374
  257. tests/test_logging_final.py +184 -184
  258. tests/test_logging_integration.py +312 -312
  259. tests/test_logging_system.py +282 -282
  260. tests/test_middleware_debug.py +141 -141
  261. tests/test_mode_consistency.py +51 -51
  262. tests/test_multi_directory.py +67 -67
  263. tests/test_multiple_spider_modules.py +80 -80
  264. tests/test_mysql_pipeline_config.py +165 -0
  265. tests/test_mysql_pipeline_error.py +99 -0
  266. tests/test_mysql_pipeline_init_log.py +83 -0
  267. tests/test_mysql_pipeline_integration.py +133 -0
  268. tests/test_mysql_pipeline_refactor.py +144 -0
  269. tests/test_mysql_pipeline_refactor_simple.py +86 -0
  270. tests/test_mysql_pipeline_robustness.py +196 -0
  271. tests/test_mysql_pipeline_types.py +89 -0
  272. tests/test_mysql_update_columns.py +94 -0
  273. tests/test_offsite_middleware.py +244 -244
  274. tests/test_offsite_middleware_simple.py +203 -203
  275. tests/test_optimized_selector_naming.py +100 -100
  276. tests/test_parsel.py +29 -29
  277. tests/test_performance.py +327 -327
  278. tests/test_performance_monitor.py +115 -115
  279. tests/test_pipeline_fingerprint_consistency.py +86 -86
  280. tests/test_priority_behavior.py +211 -211
  281. tests/test_priority_consistency.py +151 -151
  282. tests/test_priority_consistency_fixed.py +249 -249
  283. tests/test_proxy_health_check.py +32 -32
  284. tests/test_proxy_middleware.py +217 -121
  285. tests/test_proxy_middleware_enhanced.py +212 -216
  286. tests/test_proxy_middleware_integration.py +142 -137
  287. tests/test_proxy_middleware_refactored.py +207 -184
  288. tests/test_proxy_only.py +84 -0
  289. tests/test_proxy_providers.py +56 -56
  290. tests/test_proxy_stats.py +19 -19
  291. tests/test_proxy_strategies.py +59 -59
  292. tests/test_proxy_with_downloader.py +153 -0
  293. tests/test_queue_empty_check.py +41 -41
  294. tests/test_queue_manager_double_crawlo.py +173 -173
  295. tests/test_queue_manager_redis_key.py +179 -179
  296. tests/test_queue_naming.py +154 -154
  297. tests/test_queue_type.py +106 -106
  298. tests/test_queue_type_redis_config_consistency.py +130 -130
  299. tests/test_random_headers_default.py +322 -322
  300. tests/test_random_headers_necessity.py +308 -308
  301. tests/test_random_user_agent.py +72 -72
  302. tests/test_redis_config.py +28 -28
  303. tests/test_redis_connection_pool.py +294 -294
  304. tests/test_redis_key_naming.py +181 -181
  305. tests/test_redis_key_validator.py +123 -123
  306. tests/test_redis_queue.py +224 -224
  307. tests/test_redis_queue_name_fix.py +175 -175
  308. tests/test_redis_queue_type_fallback.py +129 -129
  309. tests/test_request_ignore_middleware.py +182 -182
  310. tests/test_request_params.py +111 -111
  311. tests/test_request_serialization.py +70 -70
  312. tests/test_response_code_middleware.py +349 -349
  313. tests/test_response_filter_middleware.py +427 -427
  314. tests/test_response_follow.py +104 -104
  315. tests/test_response_improvements.py +152 -152
  316. tests/test_response_selector_methods.py +92 -92
  317. tests/test_response_url_methods.py +70 -70
  318. tests/test_response_urljoin.py +86 -86
  319. tests/test_retry_middleware.py +333 -333
  320. tests/test_retry_middleware_realistic.py +273 -273
  321. tests/test_scheduler.py +252 -252
  322. tests/test_scheduler_config_update.py +133 -133
  323. tests/test_scrapy_style_encoding.py +112 -112
  324. tests/test_selector_helper.py +100 -100
  325. tests/test_selector_optimizations.py +146 -146
  326. tests/test_simple_response.py +61 -61
  327. tests/test_spider_loader.py +49 -49
  328. tests/test_spider_loader_comprehensive.py +69 -69
  329. tests/test_spider_modules.py +84 -84
  330. tests/test_spiders/test_spider.py +9 -9
  331. tests/test_telecom_spider_redis_key.py +205 -205
  332. tests/test_template_content.py +87 -87
  333. tests/test_template_redis_key.py +134 -134
  334. tests/test_tools.py +159 -159
  335. tests/test_user_agent_randomness.py +176 -176
  336. tests/test_user_agents.py +96 -96
  337. tests/untested_features_report.md +138 -138
  338. tests/verify_debug.py +51 -51
  339. tests/verify_distributed.py +117 -117
  340. tests/verify_log_fix.py +111 -111
  341. tests/verify_mysql_warnings.py +110 -0
  342. crawlo/logging/async_handler.py +0 -181
  343. crawlo/logging/monitor.py +0 -153
  344. crawlo/logging/sampler.py +0 -167
  345. crawlo/middleware/simple_proxy.py +0 -65
  346. crawlo/tools/authenticated_proxy.py +0 -241
  347. crawlo/tools/data_formatter.py +0 -226
  348. crawlo/tools/data_validator.py +0 -181
  349. crawlo/tools/encoding_converter.py +0 -127
  350. crawlo/tools/network_diagnostic.py +0 -365
  351. crawlo/tools/request_tools.py +0 -83
  352. crawlo/tools/retry_mechanism.py +0 -224
  353. crawlo/utils/env_config.py +0 -143
  354. crawlo/utils/large_scale_config.py +0 -287
  355. crawlo/utils/system.py +0 -11
  356. crawlo/utils/tools.py +0 -5
  357. crawlo-1.4.5.dist-info/METADATA +0 -329
  358. crawlo-1.4.5.dist-info/RECORD +0 -347
  359. tests/env_config_example.py +0 -134
  360. tests/ofweek_scrapy/ofweek_scrapy/spiders/ofweek_spider.py +0 -162
  361. tests/test_authenticated_proxy.py +0 -142
  362. tests/test_comprehensive.py +0 -147
  363. tests/test_dynamic_downloaders_proxy.py +0 -125
  364. tests/test_dynamic_proxy.py +0 -93
  365. tests/test_dynamic_proxy_config.py +0 -147
  366. tests/test_dynamic_proxy_real.py +0 -110
  367. tests/test_env_config.py +0 -122
  368. tests/test_framework_env_usage.py +0 -104
  369. tests/test_large_scale_config.py +0 -113
  370. tests/test_proxy_api.py +0 -265
  371. tests/test_real_scenario_proxy.py +0 -196
  372. tests/tools_example.py +0 -261
  373. {crawlo-1.4.5.dist-info → crawlo-1.4.7.dist-info}/WHEEL +0 -0
  374. {crawlo-1.4.5.dist-info → crawlo-1.4.7.dist-info}/entry_points.txt +0 -0
  375. {crawlo-1.4.5.dist-info → crawlo-1.4.7.dist-info}/top_level.txt +0 -0
@@ -1,665 +1,570 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- """
4
- HTTP Response 封装模块
5
- =====================
6
- 提供功能丰富的HTTP响应封装,支持:
7
- - 智能编码检测和解码
8
- - XPath/CSS 选择器
9
- - JSON 解析和缓存
10
- - 正则表达式支持
11
- - Cookie 处理
12
- """
13
- import re
14
- from http.cookies import SimpleCookie
15
- from typing import Dict, Any, List, Optional, Tuple
16
- from urllib.parse import urljoin as _urljoin, urlparse as _urlparse, urlsplit as _urlsplit, parse_qs as _parse_qs, \
17
- urlencode as _urlencode, quote as _quote, unquote as _unquote, urldefrag as _urldefrag
18
-
19
- import ujson
20
- from parsel import Selector, SelectorList
21
-
22
- # 尝试导入 w3lib 编码检测函数
23
- try:
24
- from w3lib.encoding import (
25
- html_body_declared_encoding,
26
- html_to_unicode,
27
- http_content_type_encoding,
28
- read_bom,
29
- resolve_encoding,
30
- )
31
- W3LIB_AVAILABLE = True
32
- except ImportError:
33
- W3LIB_AVAILABLE = False
34
-
35
- from crawlo.exceptions import DecodeError
36
- from crawlo.utils import (
37
- extract_text,
38
- extract_texts,
39
- extract_attr,
40
- extract_attrs,
41
- is_xpath
42
- )
43
-
44
-
45
- def memoize_method_noargs(func):
46
- """
47
- 装饰器,用于缓存无参数方法的结果
48
- """
49
- cache_attr = f'_cache_{func.__name__}'
50
-
51
- def wrapper(self):
52
- if not hasattr(self, cache_attr):
53
- setattr(self, cache_attr, func(self))
54
- return getattr(self, cache_attr)
55
-
56
- return wrapper
57
-
58
-
59
- class Response:
60
- """
61
- HTTP响应的封装,提供数据解析的便捷方法。
62
-
63
- 功能特性:
64
- - 智能编码检测和缓存
65
- - 懒加载 Selector 实例
66
- - JSON 解析和缓存
67
- - 多类型数据提取
68
- """
69
-
70
- # 默认编码
71
- _DEFAULT_ENCODING = "ascii"
72
-
73
- def __init__(
74
- self,
75
- url: str,
76
- *,
77
- headers: Dict[str, Any] = None,
78
- body: bytes = b"",
79
- method: str = 'GET',
80
- request: 'Request' = None, # 使用字符串注解避免循环导入
81
- status_code: int = 200,
82
- ):
83
- # 基本属性
84
- self.url = url
85
- self.headers = headers or {}
86
- self.body = body
87
- self.method = method.upper()
88
- self.request = request
89
- self.status_code = status_code
90
-
91
- # 编码处理
92
- self.encoding = self._determine_encoding()
93
-
94
- # 缓存属性
95
- self._text_cache = None
96
- self._json_cache = None
97
- self._selector_instance = None
98
-
99
- # 状态标记
100
- self._is_success = 200 <= status_code < 300
101
- self._is_redirect = 300 <= status_code < 400
102
- self._is_client_error = 400 <= status_code < 500
103
- self._is_server_error = status_code >= 500
104
-
105
- # ==================== 编码检测相关方法 ====================
106
-
107
- def _determine_encoding(self) -> str:
108
- """
109
- 智能检测响应编码(参考 Scrapy 设计)
110
-
111
- 编码检测优先级:
112
- 1. Request 中指定的编码
113
- 2. BOM 字节顺序标记
114
- 3. HTTP Content-Type 头部
115
- 4. HTML meta 标签声明
116
- 5. 内容自动检测
117
- 6. 默认编码 (utf-8)
118
- """
119
- # 1. 优先使用声明的编码
120
- declared_encoding = self._declared_encoding()
121
- if declared_encoding:
122
- return declared_encoding
123
-
124
- # 2. 使用 w3lib 进行编码检测(如果可用)
125
- if W3LIB_AVAILABLE:
126
- return self._body_inferred_encoding()
127
- else:
128
- # 如果 w3lib 不可用,使用原有的检测逻辑
129
- # 从 Content-Type 头中检测
130
- content_type = self.headers.get("content-type", "") or self.headers.get("Content-Type", "")
131
- if content_type:
132
- charset_match = re.search(r"charset=([\w-]+)", content_type, re.I)
133
- if charset_match:
134
- return charset_match.group(1).lower()
135
-
136
- # 从 HTML meta 标签中检测(仅对HTML内容)
137
- if b'<html' in self.body[:1024].lower():
138
- # 查找 <meta charset="xxx"> 或 <meta http-equiv="Content-Type" content="...charset=xxx">
139
- html_start = self.body[:4096] # 只检查前4KB
140
- try:
141
- html_text = html_start.decode('ascii', errors='ignore')
142
- # <meta charset="utf-8">
143
- charset_match = re.search(r'<meta[^>]+charset=["\']?([\w-]+)', html_text, re.I)
144
- if charset_match:
145
- return charset_match.group(1).lower()
146
-
147
- # <meta http-equiv="Content-Type" content="text/html; charset=utf-8">
148
- content_match = re.search(r'<meta[^>]+content=["\'][^"\'>]*charset=([\w-]+)', html_text, re.I)
149
- if content_match:
150
- return content_match.group(1).lower()
151
- except Exception:
152
- pass
153
-
154
- # 3. 默认使用 utf-8
155
- return 'utf-8'
156
-
157
- def _declared_encoding(self) -> Optional[str]:
158
- """
159
- 获取声明的编码
160
- 优先级:Request编码 > BOM > HTTP头部 > HTML meta标签
161
- """
162
- # 1. Request 中指定的编码
163
- if self.request and getattr(self.request, 'encoding', None):
164
- return self.request.encoding
165
-
166
- # 2. BOM 字节顺序标记
167
- bom_encoding = self._bom_encoding()
168
- if bom_encoding:
169
- return bom_encoding
170
-
171
- # 3. HTTP Content-Type 头部
172
- headers_encoding = self._headers_encoding()
173
- if headers_encoding:
174
- return headers_encoding
175
-
176
- # 4. HTML meta 标签声明编码
177
- body_declared_encoding = self._body_declared_encoding()
178
- if body_declared_encoding:
179
- return body_declared_encoding
180
-
181
- return None
182
-
183
- @memoize_method_noargs
184
- def _bom_encoding(self) -> Optional[str]:
185
- """BOM 字节顺序标记编码检测"""
186
- if not W3LIB_AVAILABLE:
187
- return None
188
- return read_bom(self.body)[0]
189
-
190
- @memoize_method_noargs
191
- def _headers_encoding(self) -> Optional[str]:
192
- """HTTP Content-Type 头部编码检测"""
193
- if not W3LIB_AVAILABLE:
194
- return None
195
- content_type = self.headers.get(b"Content-Type", b"") or self.headers.get("content-type", b"")
196
- if isinstance(content_type, bytes):
197
- content_type = content_type.decode('latin-1')
198
- return http_content_type_encoding(content_type)
199
-
200
- @memoize_method_noargs
201
- def _body_declared_encoding(self) -> Optional[str]:
202
- """HTML meta 标签声明编码检测"""
203
- if not W3LIB_AVAILABLE:
204
- return None
205
- return html_body_declared_encoding(self.body)
206
-
207
- @memoize_method_noargs
208
- def _body_inferred_encoding(self) -> str:
209
- """内容自动检测编码"""
210
- if not W3LIB_AVAILABLE:
211
- # 回退到简单的自动检测
212
- for enc in (self._DEFAULT_ENCODING, "utf-8", "cp1252"):
213
- try:
214
- self.body.decode(enc)
215
- except UnicodeError:
216
- continue
217
- return enc
218
- return self._DEFAULT_ENCODING
219
-
220
- content_type = self.headers.get(b"Content-Type", b"") or self.headers.get("content-type", b"")
221
- if isinstance(content_type, bytes):
222
- content_type = content_type.decode('latin-1')
223
-
224
- # 使用 w3lib 的 html_to_unicode 函数进行编码检测
225
- benc, _ = html_to_unicode(
226
- content_type,
227
- self.body,
228
- auto_detect_fun=self._auto_detect_fun,
229
- default_encoding=self._DEFAULT_ENCODING,
230
- )
231
- return benc
232
-
233
- def _auto_detect_fun(self, text: bytes) -> Optional[str]:
234
- """自动检测编码的回调函数"""
235
- if not W3LIB_AVAILABLE:
236
- return None
237
- for enc in (self._DEFAULT_ENCODING, "utf-8", "cp1252"):
238
- try:
239
- text.decode(enc)
240
- except UnicodeError:
241
- continue
242
- return resolve_encoding(enc)
243
- return None
244
-
245
- @property
246
- def text(self) -> str:
247
- """将响应体(body)以正确的编码解码为字符串,并缓存结果。"""
248
- if self._text_cache is not None:
249
- return self._text_cache
250
-
251
- if not self.body:
252
- self._text_cache = ""
253
- return self._text_cache
254
-
255
- # 如果可用,使用 w3lib 进行更准确的解码
256
- if W3LIB_AVAILABLE:
257
- try:
258
- content_type = self.headers.get(b"Content-Type", b"") or self.headers.get("content-type", b"")
259
- if isinstance(content_type, bytes):
260
- content_type = content_type.decode('latin-1')
261
-
262
- # 使用 w3lib 的 html_to_unicode 函数
263
- _, self._text_cache = html_to_unicode(
264
- content_type,
265
- self.body,
266
- default_encoding=self.encoding
267
- )
268
- return self._text_cache
269
- except Exception:
270
- # 如果 w3lib 解码失败,回退到原有方法
271
- pass
272
-
273
- # 尝试多种编码
274
- encodings_to_try = [self.encoding]
275
- if self.encoding != 'utf-8':
276
- encodings_to_try.append('utf-8')
277
- if 'gbk' not in encodings_to_try:
278
- encodings_to_try.append('gbk')
279
- if 'gb2312' not in encodings_to_try:
280
- encodings_to_try.append('gb2312')
281
- encodings_to_try.append('latin1') # 最后的回退选项
282
-
283
- for encoding in encodings_to_try:
284
- if not encoding:
285
- continue
286
- try:
287
- self._text_cache = self.body.decode(encoding)
288
- return self._text_cache
289
- except (UnicodeDecodeError, LookupError):
290
- continue
291
-
292
- # 所有编码都失败,使用容错解码
293
- try:
294
- self._text_cache = self.body.decode('utf-8', errors='replace')
295
- return self._text_cache
296
- except Exception as e:
297
- raise DecodeError(f"Failed to decode response from {self.url}: {e}")
298
-
299
- # ==================== 状态检查相关属性 ====================
300
-
301
- @property
302
- def is_success(self) -> bool:
303
- """检查响应是否成功 (2xx)"""
304
- return self._is_success
305
-
306
- @property
307
- def is_redirect(self) -> bool:
308
- """检查响应是否为重定向 (3xx)"""
309
- return self._is_redirect
310
-
311
- @property
312
- def is_client_error(self) -> bool:
313
- """检查响应是否为客户端错误 (4xx)"""
314
- return self._is_client_error
315
-
316
- @property
317
- def is_server_error(self) -> bool:
318
- """检查响应是否为服务器错误 (5xx)"""
319
- return self._is_server_error
320
-
321
- # ==================== 响应头相关属性 ====================
322
-
323
- @property
324
- def content_type(self) -> str:
325
- """获取响应的 Content-Type"""
326
- return self.headers.get('content-type', '') or self.headers.get('Content-Type', '')
327
-
328
- @property
329
- def content_length(self) -> Optional[int]:
330
- """获取响应的 Content-Length"""
331
- length = self.headers.get('content-length') or self.headers.get('Content-Length')
332
- return int(length) if length else None
333
-
334
- # ==================== JSON处理方法 ====================
335
-
336
- def json(self, default: Any = None) -> Any:
337
- """将响应文本解析为 JSON 对象。"""
338
- if self._json_cache is not None:
339
- return self._json_cache
340
-
341
- try:
342
- self._json_cache = ujson.loads(self.text)
343
- return self._json_cache
344
- except (ujson.JSONDecodeError, ValueError) as e:
345
- if default is not None:
346
- return default
347
- raise DecodeError(f"Failed to parse JSON from {self.url}: {e}")
348
-
349
- # ==================== URL处理方法 ====================
350
-
351
- def urljoin(self, url: str) -> str:
352
- """拼接 URL,自动处理相对路径。"""
353
- return _urljoin(self.url, url)
354
-
355
- def urlparse(self, url: str = None) -> Tuple:
356
- """
357
- 解析 URL 为组件元组 (scheme, netloc, path, params, query, fragment)
358
-
359
- Args:
360
- url (str, optional): 要解析的URL,默认为响应的URL
361
-
362
- Returns:
363
- tuple: URL组件元组
364
- """
365
- target_url = url if url is not None else self.url
366
- return _urlparse(target_url)
367
-
368
- def urlsplit(self, url: str = None) -> Tuple:
369
- """
370
- 解析 URL 为组件元组 (scheme, netloc, path, query, fragment)
371
-
372
- Args:
373
- url (str, optional): 要解析的URL,默认为响应的URL
374
-
375
- Returns:
376
- tuple: URL组件元组(不包含params)
377
- """
378
- target_url = url if url is not None else self.url
379
- return _urlsplit(target_url)
380
-
381
- def parse_qs(self, query_string: str = None, keep_blank_values: bool = False) -> Dict[str, List[str]]:
382
- """
383
- 解析查询字符串为字典
384
-
385
- Args:
386
- query_string (str, optional): 查询字符串,默认从URL中提取
387
- keep_blank_values (bool): 是否保留空值
388
-
389
- Returns:
390
- dict: 查询参数字典
391
- """
392
- if query_string is None:
393
- # 从URL中提取查询字符串
394
- parsed = _urlparse(self.url)
395
- query_string = parsed.query
396
-
397
- return _parse_qs(query_string, keep_blank_values=keep_blank_values)
398
-
399
- def urlencode(self, query: Dict[str, Any]) -> str:
400
- """
401
- 将字典编码为查询字符串
402
-
403
- Args:
404
- query (dict): 要编码的查询参数字典
405
-
406
- Returns:
407
- str: 编码后的查询字符串
408
- """
409
- return _urlencode(query)
410
-
411
- def quote(self, string: str, safe: str = '/') -> str:
412
- """
413
- URL 编码
414
-
415
- Args:
416
- string (str): 要编码的字符串
417
- safe (str): 不编码的字符,默认为 '/'
418
-
419
- Returns:
420
- str: URL编码后的字符串
421
- """
422
- return _quote(string, safe=safe)
423
-
424
- def unquote(self, string: str) -> str:
425
- """
426
- URL 解码
427
-
428
- Args:
429
- string (str): 要解码的字符串
430
-
431
- Returns:
432
- str: URL解码后的字符串
433
- """
434
- return _unquote(string)
435
-
436
- def urldefrag(self, url: str = None) -> Tuple[str, str]:
437
- """
438
- 移除 URL 中的片段标识符
439
-
440
- Args:
441
- url (str, optional): 要处理的URL,默认为响应的URL
442
-
443
- Returns:
444
- tuple: (去除片段的URL, 片段)
445
- """
446
- target_url = url if url is not None else self.url
447
- defrag_result = _urldefrag(target_url)
448
- return (defrag_result.url, defrag_result.fragment)
449
-
450
- # ==================== 选择器相关方法 ====================
451
-
452
- @property
453
- def _selector(self) -> Selector:
454
- """懒加载 Selector 实例"""
455
- if self._selector_instance is None:
456
- self._selector_instance = Selector(self.text)
457
- return self._selector_instance
458
-
459
- def xpath(self, query: str) -> SelectorList:
460
- """使用 XPath 选择器查询文档。"""
461
- return self._selector.xpath(query)
462
-
463
- def css(self, query: str) -> SelectorList:
464
- """使用 CSS 选择器查询文档。"""
465
- return self._selector.css(query)
466
-
467
- def _is_xpath(self, query: str) -> bool:
468
- """判断查询语句是否为XPath"""
469
- return is_xpath(query)
470
-
471
- def _extract_text(self, elements: SelectorList, join_str: str = " ") -> str:
472
- """
473
- 从元素列表中提取文本并拼接
474
-
475
- :param elements: SelectorList元素列表
476
- :param join_str: 文本拼接分隔符
477
- :return: 拼接后的文本
478
- """
479
- return extract_text(elements, join_str)
480
-
481
- def extract_text(self, xpath_or_css: str, join_str: str = " ", default: str = '') -> str:
482
- """
483
- 提取单个元素的文本内容,支持CSS和XPath选择器
484
-
485
- 参数:
486
- xpath_or_css: XPath或CSS选择器
487
- join_str: 文本拼接分隔符(默认为空格)
488
- default: 默认返回值,当未找到元素时返回
489
-
490
- 返回:
491
- 拼接后的纯文本字符串
492
-
493
- 示例:
494
- # 使用CSS选择器
495
- title = response.extract_text('title')
496
- content = response.extract_text('.content p', join_str=' ')
497
-
498
- # 使用XPath选择器
499
- title = response.extract_text('//title')
500
- content = response.extract_text('//div[@class="content"]//p', join_str=' ')
501
- """
502
- try:
503
- elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
504
- if not elements:
505
- return default
506
- return self._extract_text(elements, join_str)
507
- except Exception:
508
- return default
509
-
510
- def extract_texts(self, xpath_or_css: str, join_str: str = " ", default: List[str] = None) -> List[str]:
511
- """
512
- 提取多个元素的文本内容列表,支持CSS和XPath选择器
513
-
514
- 参数:
515
- xpath_or_css: XPath或CSS选择器
516
- join_str: 单个节点内文本拼接分隔符
517
- default: 默认返回值,当未找到元素时返回
518
-
519
- 返回:
520
- 纯文本列表(每个元素对应一个节点的文本)
521
-
522
- 示例:
523
- # 使用CSS选择器
524
- paragraphs = response.extract_texts('.content p')
525
- titles = response.extract_texts('h1, h2, h3')
526
-
527
- # 使用XPath选择器
528
- paragraphs = response.extract_texts('//div[@class="content"]//p')
529
- titles = response.extract_texts('//h1 | //h2 | //h3')
530
- """
531
- if default is None:
532
- default = []
533
-
534
- try:
535
- elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
536
- if not elements:
537
- return default
538
-
539
- result = extract_texts(elements, join_str)
540
- return result if result else default
541
- except Exception:
542
- return default
543
-
544
- def extract_attr(self, xpath_or_css: str, attr_name: str, default: Any = None) -> Any:
545
- """
546
- 提取单个元素的属性值,支持CSS和XPath选择器
547
-
548
- 参数:
549
- xpath_or_css: XPath或CSS选择器
550
- attr_name: 属性名称
551
- default: 默认返回值
552
-
553
- 返回:
554
- 属性值或默认值
555
-
556
- 示例:
557
- # 使用CSS选择器
558
- link_href = response.extract_attr('a', 'href')
559
- img_src = response.extract_attr('.image', 'src')
560
-
561
- # 使用XPath选择器
562
- link_href = response.extract_attr('//a[@class="link"]', 'href')
563
- img_src = response.extract_attr('//img[@alt="example"]', 'src')
564
- """
565
- try:
566
- elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
567
- if not elements:
568
- return default
569
- return extract_attr(elements, attr_name, default)
570
- except Exception:
571
- return default
572
-
573
- def extract_attrs(self, xpath_or_css: str, attr_name: str, default: List[Any] = None) -> List[Any]:
574
- """
575
- 提取多个元素的属性值列表,支持CSS和XPath选择器
576
-
577
- 参数:
578
- xpath_or_css: XPath或CSS选择器
579
- attr_name: 属性名称
580
- default: 默认返回值
581
-
582
- 返回:
583
- 属性值列表
584
-
585
- 示例:
586
- # 使用CSS选择器
587
- links = response.extract_attrs('a', 'href')
588
- images = response.extract_attrs('img', 'src')
589
-
590
- # 使用XPath选择器
591
- links = response.extract_attrs('//a[@class="link"]', 'href')
592
- images = response.extract_attrs('//img[@alt]', 'src')
593
- """
594
- if default is None:
595
- default = []
596
-
597
- try:
598
- elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
599
- if not elements:
600
- return default
601
-
602
- result = extract_attrs(elements, attr_name)
603
- return result if result else default
604
- except Exception:
605
- return default
606
-
607
- # ==================== 正则表达式相关方法 ====================
608
-
609
- def re_search(self, pattern: str, flags: int = re.DOTALL) -> Optional[re.Match]:
610
- """在响应文本上执行正则表达式搜索。"""
611
- if not isinstance(pattern, str):
612
- raise TypeError("Pattern must be a string")
613
- return re.search(pattern, self.text, flags=flags)
614
-
615
- def re_findall(self, pattern: str, flags: int = re.DOTALL) -> List[Any]:
616
- """在响应文本上执行正则表达式查找。"""
617
- if not isinstance(pattern, str):
618
- raise TypeError("Pattern must be a string")
619
- return re.findall(pattern, self.text, flags=flags)
620
-
621
- # ==================== Cookie处理方法 ====================
622
-
623
- def get_cookies(self) -> Dict[str, str]:
624
- """从响应头中解析并返回Cookies。"""
625
- cookie_header = self.headers.get("Set-Cookie", "")
626
- if isinstance(cookie_header, list):
627
- cookie_header = ", ".join(cookie_header)
628
- cookies = SimpleCookie()
629
- cookies.load(cookie_header)
630
- return {key: morsel.value for key, morsel in cookies.items()}
631
-
632
- # ==================== 请求相关方法 ====================
633
-
634
- def follow(self, url: str, callback=None, **kwargs):
635
- """
636
- 创建一个跟随链接的请求。
637
-
638
- Args:
639
- url: 要跟随的URL(可以是相对URL)
640
- callback: 回调函数
641
- **kwargs: 传递给Request的其他参数
642
-
643
- Returns:
644
- Request: 新的请求对象
645
- """
646
- # 延迟导入Request以避免循环导入
647
- from ..network.request import Request
648
-
649
- # 使用urljoin处理相对URL
650
- absolute_url = self.urljoin(url)
651
-
652
- # 创建新的请求
653
- return Request(
654
- url=absolute_url,
655
- callback=callback or getattr(self.request, 'callback', None),
656
- **kwargs
657
- )
658
-
659
- @property
660
- def meta(self) -> Dict:
661
- """获取关联的 Request 对象的 meta 字典。"""
662
- return self.request.meta if self.request else {}
663
-
664
- def __str__(self):
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ """
4
+ HTTP Response 封装模块
5
+ =====================
6
+ 提供功能丰富的HTTP响应封装,支持:
7
+ - 智能编码检测和解码
8
+ - XPath/CSS 选择器
9
+ - JSON 解析和缓存
10
+ - 正则表达式支持
11
+ - Cookie 处理
12
+ """
13
+ import re
14
+ from http.cookies import SimpleCookie
15
+ from typing import Dict, Any, List, Optional, Tuple
16
+ from urllib.parse import urljoin as _urljoin, urlparse as _urlparse, urlsplit as _urlsplit, parse_qs as _parse_qs, \
17
+ urlencode as _urlencode, quote as _quote, unquote as _unquote, urldefrag as _urldefrag
18
+
19
+ import ujson
20
+ from parsel import Selector, SelectorList
21
+
22
+ # 尝试导入 w3lib 编码检测函数
23
+ try:
24
+ from w3lib.encoding import (
25
+ html_body_declared_encoding,
26
+ html_to_unicode,
27
+ http_content_type_encoding,
28
+ read_bom,
29
+ resolve_encoding,
30
+ )
31
+ W3LIB_AVAILABLE = True
32
+ except ImportError:
33
+ W3LIB_AVAILABLE = False
34
+
35
+ from crawlo.exceptions import DecodeError
36
+ from crawlo.utils import (
37
+ extract_text,
38
+ extract_texts,
39
+ extract_attr,
40
+ extract_attrs,
41
+ is_xpath
42
+ )
43
+
44
+
45
+ def memoize_method_noargs(func):
46
+ """
47
+ 装饰器,用于缓存无参数方法的结果
48
+ """
49
+ cache_attr = f'_cache_{func.__name__}'
50
+
51
+ def wrapper(self):
52
+ if not hasattr(self, cache_attr):
53
+ setattr(self, cache_attr, func(self))
54
+ return getattr(self, cache_attr)
55
+
56
+ return wrapper
57
+
58
+
59
+ class Response:
60
+ """
61
+ HTTP响应的封装,提供数据解析的便捷方法。
62
+
63
+ 功能特性:
64
+ - 智能编码检测和缓存
65
+ - 懒加载 Selector 实例
66
+ - JSON 解析和缓存
67
+ - 多类型数据提取
68
+ """
69
+
70
+ # 默认编码
71
+ _DEFAULT_ENCODING = "ascii"
72
+
73
+ def __init__(
74
+ self,
75
+ url: str,
76
+ *,
77
+ headers: Dict[str, Any] = None,
78
+ body: bytes = b"",
79
+ method: str = 'GET',
80
+ request: 'Request' = None, # 使用字符串注解避免循环导入
81
+ status_code: int = 200,
82
+ ):
83
+ # 基本属性
84
+ self.url = url
85
+ self.headers = headers or {}
86
+ self.body = body
87
+ self.method = method.upper()
88
+ self.request = request
89
+ self.status_code = status_code
90
+
91
+ # 编码处理
92
+ self.encoding = self._determine_encoding()
93
+
94
+ # 缓存属性
95
+ self._text_cache = None
96
+ self._json_cache = None
97
+ self._selector_instance = None
98
+
99
+ # 状态标记
100
+ self._is_success = 200 <= status_code < 300
101
+ self._is_redirect = 300 <= status_code < 400
102
+ self._is_client_error = 400 <= status_code < 500
103
+ self._is_server_error = status_code >= 500
104
+
105
+ # ==================== 编码检测相关方法 ====================
106
+
107
+ def _determine_encoding(self) -> str:
108
+ """
109
+ 智能检测响应编码
110
+
111
+ 编码检测优先级:
112
+ 1. Request 中指定的编码
113
+ 2. BOM 字节顺序标记
114
+ 3. HTTP Content-Type 头部
115
+ 4. HTML meta 标签声明
116
+ 5. 内容自动检测
117
+ 6. 默认编码 (utf-8)
118
+ """
119
+ # 1. 优先使用声明的编码
120
+ declared_encoding = self._declared_encoding()
121
+ if declared_encoding:
122
+ return declared_encoding
123
+
124
+ # 2. 使用 w3lib 进行编码检测(如果可用)
125
+ if W3LIB_AVAILABLE:
126
+ return self._body_inferred_encoding()
127
+ else:
128
+ # 如果 w3lib 不可用,使用原有的检测逻辑
129
+ # 从 Content-Type 头中检测
130
+ content_type = self.headers.get("content-type", "") or self.headers.get("Content-Type", "")
131
+ if content_type:
132
+ charset_match = re.search(r"charset=([\w-]+)", content_type, re.I)
133
+ if charset_match:
134
+ return charset_match.group(1).lower()
135
+
136
+ # 从 HTML meta 标签中检测(仅对HTML内容)
137
+ if b'<html' in self.body[:1024].lower():
138
+ # 查找 <meta charset="xxx"> 或 <meta http-equiv="Content-Type" content="...charset=xxx">
139
+ html_start = self.body[:4096] # 只检查前4KB
140
+ try:
141
+ html_text = html_start.decode('ascii', errors='ignore')
142
+ # <meta charset="utf-8">
143
+ charset_match = re.search(r'<meta[^>]+charset=["\']?([\w-]+)', html_text, re.I)
144
+ if charset_match:
145
+ return charset_match.group(1).lower()
146
+
147
+ # <meta http-equiv="Content-Type" content="text/html; charset=utf-8">
148
+ content_match = re.search(r'<meta[^>]+content=["\'][^"\'>]*charset=([\w-]+)', html_text, re.I)
149
+ if content_match:
150
+ return content_match.group(1).lower()
151
+ except Exception:
152
+ pass
153
+
154
+ # 3. 默认使用 utf-8
155
+ return 'utf-8'
156
+
157
+ def _declared_encoding(self) -> Optional[str]:
158
+ """
159
+ 获取声明的编码
160
+ 优先级:Request编码 > BOM > HTTP头部 > HTML meta标签
161
+ """
162
+ # 1. Request 中指定的编码
163
+ if self.request and getattr(self.request, 'encoding', None):
164
+ return self.request.encoding
165
+
166
+ # 2. BOM 字节顺序标记
167
+ bom_encoding = self._bom_encoding()
168
+ if bom_encoding:
169
+ return bom_encoding
170
+
171
+ # 3. HTTP Content-Type 头部
172
+ headers_encoding = self._headers_encoding()
173
+ if headers_encoding:
174
+ return headers_encoding
175
+
176
+ # 4. HTML meta 标签声明编码
177
+ body_declared_encoding = self._body_declared_encoding()
178
+ if body_declared_encoding:
179
+ return body_declared_encoding
180
+
181
+ return None
182
+
183
+ @memoize_method_noargs
184
+ def _bom_encoding(self) -> Optional[str]:
185
+ """BOM 字节顺序标记编码检测"""
186
+ if not W3LIB_AVAILABLE:
187
+ return None
188
+ return read_bom(self.body)[0]
189
+
190
+ @memoize_method_noargs
191
+ def _headers_encoding(self) -> Optional[str]:
192
+ """HTTP Content-Type 头部编码检测"""
193
+ if not W3LIB_AVAILABLE:
194
+ return None
195
+ content_type = self.headers.get(b"Content-Type", b"") or self.headers.get("content-type", b"")
196
+ if isinstance(content_type, bytes):
197
+ content_type = content_type.decode('latin-1')
198
+ return http_content_type_encoding(content_type)
199
+
200
+ @memoize_method_noargs
201
+ def _body_declared_encoding(self) -> Optional[str]:
202
+ """HTML meta 标签声明编码检测"""
203
+ if not W3LIB_AVAILABLE:
204
+ return None
205
+ return html_body_declared_encoding(self.body)
206
+
207
+ @memoize_method_noargs
208
+ def _body_inferred_encoding(self) -> str:
209
+ """内容自动检测编码"""
210
+ if not W3LIB_AVAILABLE:
211
+ # 回退到简单的自动检测
212
+ for enc in (self._DEFAULT_ENCODING, "utf-8", "cp1252"):
213
+ try:
214
+ self.body.decode(enc)
215
+ except UnicodeError:
216
+ continue
217
+ return enc
218
+ return self._DEFAULT_ENCODING
219
+
220
+ content_type = self.headers.get(b"Content-Type", b"") or self.headers.get("content-type", b"")
221
+ if isinstance(content_type, bytes):
222
+ content_type = content_type.decode('latin-1')
223
+
224
+ # 使用 w3lib 的 html_to_unicode 函数进行编码检测
225
+ benc, _ = html_to_unicode(
226
+ content_type,
227
+ self.body,
228
+ auto_detect_fun=self._auto_detect_fun,
229
+ default_encoding=self._DEFAULT_ENCODING,
230
+ )
231
+ return benc
232
+
233
+ def _auto_detect_fun(self, text: bytes) -> Optional[str]:
234
+ """自动检测编码的回调函数"""
235
+ if not W3LIB_AVAILABLE:
236
+ return None
237
+ for enc in (self._DEFAULT_ENCODING, "utf-8", "cp1252"):
238
+ try:
239
+ text.decode(enc)
240
+ except UnicodeError:
241
+ continue
242
+ return resolve_encoding(enc)
243
+ return None
244
+
245
+ @property
246
+ def text(self) -> str:
247
+ """将响应体(body)以正确的编码解码为字符串,并缓存结果。"""
248
+ if self._text_cache is not None:
249
+ return self._text_cache
250
+
251
+ if not self.body:
252
+ self._text_cache = ""
253
+ return self._text_cache
254
+
255
+ # 如果可用,使用 w3lib 进行更准确的解码
256
+ if W3LIB_AVAILABLE:
257
+ try:
258
+ content_type = self.headers.get(b"Content-Type", b"") or self.headers.get("content-type", b"")
259
+ if isinstance(content_type, bytes):
260
+ content_type = content_type.decode('latin-1')
261
+
262
+ # 使用 w3lib 的 html_to_unicode 函数
263
+ _, self._text_cache = html_to_unicode(
264
+ content_type,
265
+ self.body,
266
+ default_encoding=self.encoding
267
+ )
268
+ return self._text_cache
269
+ except Exception:
270
+ # 如果 w3lib 解码失败,回退到原有方法
271
+ pass
272
+
273
+ # 尝试多种编码
274
+ encodings_to_try = [self.encoding]
275
+ if self.encoding != 'utf-8':
276
+ encodings_to_try.append('utf-8')
277
+ if 'gbk' not in encodings_to_try:
278
+ encodings_to_try.append('gbk')
279
+ if 'gb2312' not in encodings_to_try:
280
+ encodings_to_try.append('gb2312')
281
+ encodings_to_try.append('latin1') # 最后的回退选项
282
+
283
+ for encoding in encodings_to_try:
284
+ if not encoding:
285
+ continue
286
+ try:
287
+ self._text_cache = self.body.decode(encoding)
288
+ return self._text_cache
289
+ except (UnicodeDecodeError, LookupError):
290
+ continue
291
+
292
+ # 所有编码都失败,使用容错解码
293
+ try:
294
+ self._text_cache = self.body.decode('utf-8', errors='replace')
295
+ return self._text_cache
296
+ except Exception as e:
297
+ raise DecodeError(f"Failed to decode response from {self.url}: {e}")
298
+
299
+ # ==================== 状态检查相关属性 ====================
300
+
301
+ @property
302
+ def is_success(self) -> bool:
303
+ """检查响应是否成功 (2xx)"""
304
+ return self._is_success
305
+
306
+ @property
307
+ def is_redirect(self) -> bool:
308
+ """检查响应是否为重定向 (3xx)"""
309
+ return self._is_redirect
310
+
311
+ @property
312
+ def is_client_error(self) -> bool:
313
+ """检查响应是否为客户端错误 (4xx)"""
314
+ return self._is_client_error
315
+
316
+ @property
317
+ def is_server_error(self) -> bool:
318
+ """检查响应是否为服务器错误 (5xx)"""
319
+ return self._is_server_error
320
+
321
+ # ==================== 响应头相关属性 ====================
322
+
323
+ @property
324
+ def content_type(self) -> str:
325
+ """获取响应的 Content-Type"""
326
+ return self.headers.get('content-type', '') or self.headers.get('Content-Type', '')
327
+
328
+ @property
329
+ def content_length(self) -> Optional[int]:
330
+ """获取响应的 Content-Length"""
331
+ length = self.headers.get('content-length') or self.headers.get('Content-Length')
332
+ return int(length) if length else None
333
+
334
+ # ==================== JSON处理方法 ====================
335
+
336
+ def json(self, default: Any = None) -> Any:
337
+ """将响应文本解析为 JSON 对象。"""
338
+ if self._json_cache is not None:
339
+ return self._json_cache
340
+
341
+ try:
342
+ self._json_cache = ujson.loads(self.text)
343
+ return self._json_cache
344
+ except (ujson.JSONDecodeError, ValueError) as e:
345
+ if default is not None:
346
+ return default
347
+ raise DecodeError(f"Failed to parse JSON from {self.url}: {e}")
348
+
349
+ # ==================== URL处理方法 ====================
350
+
351
+ def urljoin(self, url: str) -> str:
352
+ """拼接 URL,自动处理相对路径。"""
353
+ return _urljoin(self.url, url)
354
+
355
+ # ==================== 选择器相关方法 ====================
356
+
357
+ @property
358
+ def _selector(self) -> Selector:
359
+ """懒加载 Selector 实例"""
360
+ if self._selector_instance is None:
361
+ self._selector_instance = Selector(self.text)
362
+ return self._selector_instance
363
+
364
+ def xpath(self, query: str) -> SelectorList:
365
+ """使用 XPath 选择器查询文档。"""
366
+ return self._selector.xpath(query)
367
+
368
+ def css(self, query: str) -> SelectorList:
369
+ """使用 CSS 选择器查询文档。"""
370
+ return self._selector.css(query)
371
+
372
+ def _is_xpath(self, query: str) -> bool:
373
+ """判断查询语句是否为XPath"""
374
+ return is_xpath(query)
375
+
376
+ def _extract_text(self, elements: SelectorList, join_str: str = " ") -> str:
377
+ """
378
+ 从元素列表中提取文本并拼接
379
+
380
+ :param elements: SelectorList元素列表
381
+ :param join_str: 文本拼接分隔符
382
+ :return: 拼接后的文本
383
+ """
384
+ return extract_text(elements, join_str)
385
+
386
+ def extract_text(self, xpath_or_css: str, join_str: str = " ", default: str = '') -> str:
387
+ """
388
+ 提取单个元素的文本内容,支持CSS和XPath选择器
389
+
390
+ 参数:
391
+ xpath_or_css: XPath或CSS选择器
392
+ join_str: 文本拼接分隔符(默认为空格)
393
+ default: 默认返回值,当未找到元素时返回
394
+
395
+ 返回:
396
+ 拼接后的纯文本字符串
397
+
398
+ 示例:
399
+ # 使用CSS选择器
400
+ title = response.extract_text('title')
401
+ content = response.extract_text('.content p', join_str=' ')
402
+
403
+ # 使用XPath选择器
404
+ title = response.extract_text('//title')
405
+ content = response.extract_text('//div[@class="content"]//p', join_str=' ')
406
+ """
407
+ try:
408
+ elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
409
+ if not elements:
410
+ return default
411
+ return self._extract_text(elements, join_str)
412
+ except Exception:
413
+ return default
414
+
415
+ def extract_texts(self, xpath_or_css: str, join_str: str = " ", default: List[str] = None) -> List[str]:
416
+ """
417
+ 提取多个元素的文本内容列表,支持CSS和XPath选择器
418
+
419
+ 参数:
420
+ xpath_or_css: XPath或CSS选择器
421
+ join_str: 单个节点内文本拼接分隔符
422
+ default: 默认返回值,当未找到元素时返回
423
+
424
+ 返回:
425
+ 纯文本列表(每个元素对应一个节点的文本)
426
+
427
+ 示例:
428
+ # 使用CSS选择器
429
+ paragraphs = response.extract_texts('.content p')
430
+ titles = response.extract_texts('h1, h2, h3')
431
+
432
+ # 使用XPath选择器
433
+ paragraphs = response.extract_texts('//div[@class="content"]//p')
434
+ titles = response.extract_texts('//h1 | //h2 | //h3')
435
+ """
436
+ if default is None:
437
+ default = []
438
+
439
+ try:
440
+ elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
441
+ if not elements:
442
+ return default
443
+
444
+ result = extract_texts(elements, join_str)
445
+ return result if result else default
446
+ except Exception:
447
+ return default
448
+
449
+ def extract_attr(self, xpath_or_css: str, attr_name: str, default: Any = None) -> Any:
450
+ """
451
+ 提取单个元素的属性值,支持CSS和XPath选择器
452
+
453
+ 参数:
454
+ xpath_or_css: XPath或CSS选择器
455
+ attr_name: 属性名称
456
+ default: 默认返回值
457
+
458
+ 返回:
459
+ 属性值或默认值
460
+
461
+ 示例:
462
+ # 使用CSS选择器
463
+ link_href = response.extract_attr('a', 'href')
464
+ img_src = response.extract_attr('.image', 'src')
465
+
466
+ # 使用XPath选择器
467
+ link_href = response.extract_attr('//a[@class="link"]', 'href')
468
+ img_src = response.extract_attr('//img[@alt="example"]', 'src')
469
+ """
470
+ try:
471
+ elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
472
+ if not elements:
473
+ return default
474
+ return extract_attr(elements, attr_name, default)
475
+ except Exception:
476
+ return default
477
+
478
+ def extract_attrs(self, xpath_or_css: str, attr_name: str, default: List[Any] = None) -> List[Any]:
479
+ """
480
+ 提取多个元素的属性值列表,支持CSS和XPath选择器
481
+
482
+ 参数:
483
+ xpath_or_css: XPath或CSS选择器
484
+ attr_name: 属性名称
485
+ default: 默认返回值
486
+
487
+ 返回:
488
+ 属性值列表
489
+
490
+ 示例:
491
+ # 使用CSS选择器
492
+ links = response.extract_attrs('a', 'href')
493
+ images = response.extract_attrs('img', 'src')
494
+
495
+ # 使用XPath选择器
496
+ links = response.extract_attrs('//a[@class="link"]', 'href')
497
+ images = response.extract_attrs('//img[@alt]', 'src')
498
+ """
499
+ if default is None:
500
+ default = []
501
+
502
+ try:
503
+ elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
504
+ if not elements:
505
+ return default
506
+
507
+ result = extract_attrs(elements, attr_name)
508
+ return result if result else default
509
+ except Exception:
510
+ return default
511
+
512
+ # ==================== 正则表达式相关方法 ====================
513
+
514
+ def re_search(self, pattern: str, flags: int = re.DOTALL) -> Optional[re.Match]:
515
+ """在响应文本上执行正则表达式搜索。"""
516
+ if not isinstance(pattern, str):
517
+ raise TypeError("Pattern must be a string")
518
+ return re.search(pattern, self.text, flags=flags)
519
+
520
+ def re_findall(self, pattern: str, flags: int = re.DOTALL) -> List[Any]:
521
+ """在响应文本上执行正则表达式查找。"""
522
+ if not isinstance(pattern, str):
523
+ raise TypeError("Pattern must be a string")
524
+ return re.findall(pattern, self.text, flags=flags)
525
+
526
+ # ==================== Cookie处理方法 ====================
527
+
528
+ def get_cookies(self) -> Dict[str, str]:
529
+ """从响应头中解析并返回Cookies。"""
530
+ cookie_header = self.headers.get("Set-Cookie", "")
531
+ if isinstance(cookie_header, list):
532
+ cookie_header = ", ".join(cookie_header)
533
+ cookies = SimpleCookie()
534
+ cookies.load(cookie_header)
535
+ return {key: morsel.value for key, morsel in cookies.items()}
536
+
537
+ # ==================== 请求相关方法 ====================
538
+
539
+ def follow(self, url: str, callback=None, **kwargs):
540
+ """
541
+ 创建一个跟随链接的请求。
542
+
543
+ Args:
544
+ url: 要跟随的URL(可以是相对URL)
545
+ callback: 回调函数
546
+ **kwargs: 传递给Request的其他参数
547
+
548
+ Returns:
549
+ Request: 新的请求对象
550
+ """
551
+ # 延迟导入Request以避免循环导入
552
+ from ..network.request import Request
553
+
554
+ # 使用urljoin处理相对URL
555
+ absolute_url = self.urljoin(url)
556
+
557
+ # 创建新的请求
558
+ return Request(
559
+ url=absolute_url,
560
+ callback=callback or getattr(self.request, 'callback', None),
561
+ **kwargs
562
+ )
563
+
564
+ @property
565
+ def meta(self) -> Dict:
566
+ """获取关联的 Request 对象的 meta 字典。"""
567
+ return self.request.meta if self.request else {}
568
+
569
+ def __str__(self):
665
570
  return f"<{self.status_code} {self.url}>"