crawlo 1.1.4__py3-none-any.whl → 1.1.5__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 (186) hide show
  1. crawlo/__init__.py +61 -34
  2. crawlo/__version__.py +1 -1
  3. crawlo/cleaners/__init__.py +61 -0
  4. crawlo/cleaners/data_formatter.py +226 -0
  5. crawlo/cleaners/encoding_converter.py +126 -0
  6. crawlo/cleaners/text_cleaner.py +233 -0
  7. crawlo/cli.py +40 -40
  8. crawlo/commands/__init__.py +13 -13
  9. crawlo/commands/check.py +594 -594
  10. crawlo/commands/genspider.py +151 -151
  11. crawlo/commands/list.py +155 -155
  12. crawlo/commands/run.py +285 -285
  13. crawlo/commands/startproject.py +300 -196
  14. crawlo/commands/stats.py +188 -188
  15. crawlo/commands/utils.py +186 -186
  16. crawlo/config.py +309 -279
  17. crawlo/config_validator.py +253 -0
  18. crawlo/core/__init__.py +2 -2
  19. crawlo/core/engine.py +346 -172
  20. crawlo/core/processor.py +40 -40
  21. crawlo/core/scheduler.py +137 -166
  22. crawlo/crawler.py +1027 -1027
  23. crawlo/downloader/__init__.py +266 -242
  24. crawlo/downloader/aiohttp_downloader.py +220 -212
  25. crawlo/downloader/cffi_downloader.py +256 -251
  26. crawlo/downloader/httpx_downloader.py +259 -259
  27. crawlo/downloader/hybrid_downloader.py +214 -0
  28. crawlo/downloader/playwright_downloader.py +403 -0
  29. crawlo/downloader/selenium_downloader.py +473 -0
  30. crawlo/event.py +11 -11
  31. crawlo/exceptions.py +81 -81
  32. crawlo/extension/__init__.py +37 -37
  33. crawlo/extension/health_check.py +141 -141
  34. crawlo/extension/log_interval.py +57 -57
  35. crawlo/extension/log_stats.py +81 -81
  36. crawlo/extension/logging_extension.py +43 -43
  37. crawlo/extension/memory_monitor.py +104 -88
  38. crawlo/extension/performance_profiler.py +133 -117
  39. crawlo/extension/request_recorder.py +107 -107
  40. crawlo/filters/__init__.py +154 -154
  41. crawlo/filters/aioredis_filter.py +280 -242
  42. crawlo/filters/memory_filter.py +269 -269
  43. crawlo/items/__init__.py +23 -23
  44. crawlo/items/base.py +21 -21
  45. crawlo/items/fields.py +53 -53
  46. crawlo/items/items.py +104 -104
  47. crawlo/middleware/__init__.py +21 -21
  48. crawlo/middleware/default_header.py +32 -32
  49. crawlo/middleware/download_delay.py +28 -28
  50. crawlo/middleware/middleware_manager.py +135 -135
  51. crawlo/middleware/proxy.py +272 -248
  52. crawlo/middleware/request_ignore.py +30 -30
  53. crawlo/middleware/response_code.py +18 -18
  54. crawlo/middleware/response_filter.py +26 -26
  55. crawlo/middleware/retry.py +124 -124
  56. crawlo/mode_manager.py +206 -201
  57. crawlo/network/__init__.py +21 -21
  58. crawlo/network/request.py +338 -311
  59. crawlo/network/response.py +360 -271
  60. crawlo/pipelines/__init__.py +21 -21
  61. crawlo/pipelines/bloom_dedup_pipeline.py +156 -156
  62. crawlo/pipelines/console_pipeline.py +39 -39
  63. crawlo/pipelines/csv_pipeline.py +316 -316
  64. crawlo/pipelines/database_dedup_pipeline.py +224 -224
  65. crawlo/pipelines/json_pipeline.py +218 -218
  66. crawlo/pipelines/memory_dedup_pipeline.py +115 -115
  67. crawlo/pipelines/mongo_pipeline.py +131 -131
  68. crawlo/pipelines/mysql_pipeline.py +316 -316
  69. crawlo/pipelines/pipeline_manager.py +56 -56
  70. crawlo/pipelines/redis_dedup_pipeline.py +166 -162
  71. crawlo/project.py +153 -153
  72. crawlo/queue/pqueue.py +37 -37
  73. crawlo/queue/queue_manager.py +320 -307
  74. crawlo/queue/redis_priority_queue.py +277 -209
  75. crawlo/settings/__init__.py +7 -7
  76. crawlo/settings/default_settings.py +216 -278
  77. crawlo/settings/setting_manager.py +99 -99
  78. crawlo/spider/__init__.py +639 -639
  79. crawlo/stats_collector.py +59 -59
  80. crawlo/subscriber.py +130 -130
  81. crawlo/task_manager.py +30 -30
  82. crawlo/templates/crawlo.cfg.tmpl +10 -10
  83. crawlo/templates/project/__init__.py.tmpl +3 -3
  84. crawlo/templates/project/items.py.tmpl +17 -17
  85. crawlo/templates/project/middlewares.py.tmpl +110 -110
  86. crawlo/templates/project/pipelines.py.tmpl +97 -97
  87. crawlo/templates/project/run.py.tmpl +251 -251
  88. crawlo/templates/project/settings.py.tmpl +326 -279
  89. crawlo/templates/project/settings_distributed.py.tmpl +120 -0
  90. crawlo/templates/project/settings_gentle.py.tmpl +95 -0
  91. crawlo/templates/project/settings_high_performance.py.tmpl +152 -0
  92. crawlo/templates/project/settings_simple.py.tmpl +69 -0
  93. crawlo/templates/project/spiders/__init__.py.tmpl +5 -5
  94. crawlo/templates/spider/spider.py.tmpl +141 -141
  95. crawlo/tools/__init__.py +183 -0
  96. crawlo/tools/anti_crawler.py +269 -0
  97. crawlo/tools/authenticated_proxy.py +241 -0
  98. crawlo/tools/data_validator.py +181 -0
  99. crawlo/tools/date_tools.py +36 -0
  100. crawlo/tools/distributed_coordinator.py +387 -0
  101. crawlo/tools/retry_mechanism.py +221 -0
  102. crawlo/tools/scenario_adapter.py +263 -0
  103. crawlo/utils/__init__.py +35 -7
  104. crawlo/utils/batch_processor.py +261 -0
  105. crawlo/utils/controlled_spider_mixin.py +439 -439
  106. crawlo/utils/date_tools.py +290 -233
  107. crawlo/utils/db_helper.py +343 -343
  108. crawlo/utils/enhanced_error_handler.py +360 -0
  109. crawlo/utils/env_config.py +106 -0
  110. crawlo/utils/error_handler.py +126 -0
  111. crawlo/utils/func_tools.py +82 -82
  112. crawlo/utils/large_scale_config.py +286 -286
  113. crawlo/utils/large_scale_helper.py +343 -343
  114. crawlo/utils/log.py +128 -128
  115. crawlo/utils/performance_monitor.py +285 -0
  116. crawlo/utils/queue_helper.py +175 -175
  117. crawlo/utils/redis_connection_pool.py +335 -0
  118. crawlo/utils/redis_key_validator.py +200 -0
  119. crawlo/utils/request.py +267 -267
  120. crawlo/utils/request_serializer.py +219 -219
  121. crawlo/utils/spider_loader.py +62 -62
  122. crawlo/utils/system.py +11 -11
  123. crawlo/utils/tools.py +4 -4
  124. crawlo/utils/url.py +39 -39
  125. {crawlo-1.1.4.dist-info → crawlo-1.1.5.dist-info}/METADATA +401 -403
  126. crawlo-1.1.5.dist-info/RECORD +185 -0
  127. examples/__init__.py +7 -7
  128. tests/__init__.py +7 -7
  129. tests/advanced_tools_example.py +276 -0
  130. tests/authenticated_proxy_example.py +237 -0
  131. tests/cleaners_example.py +161 -0
  132. tests/config_validation_demo.py +103 -0
  133. {examples → tests}/controlled_spider_example.py +205 -205
  134. tests/date_tools_example.py +181 -0
  135. tests/dynamic_loading_example.py +524 -0
  136. tests/dynamic_loading_test.py +105 -0
  137. tests/env_config_example.py +134 -0
  138. tests/error_handling_example.py +172 -0
  139. tests/redis_key_validation_demo.py +131 -0
  140. tests/response_improvements_example.py +145 -0
  141. tests/test_advanced_tools.py +149 -0
  142. tests/test_all_redis_key_configs.py +146 -0
  143. tests/test_authenticated_proxy.py +142 -0
  144. tests/test_cleaners.py +55 -0
  145. tests/test_comprehensive.py +147 -0
  146. tests/test_config_validator.py +194 -0
  147. tests/test_date_tools.py +124 -0
  148. tests/test_dynamic_downloaders_proxy.py +125 -0
  149. tests/test_dynamic_proxy.py +93 -0
  150. tests/test_dynamic_proxy_config.py +147 -0
  151. tests/test_dynamic_proxy_real.py +110 -0
  152. tests/test_edge_cases.py +304 -0
  153. tests/test_enhanced_error_handler.py +271 -0
  154. tests/test_env_config.py +122 -0
  155. tests/test_error_handler_compatibility.py +113 -0
  156. tests/test_final_validation.py +153 -153
  157. tests/test_framework_env_usage.py +104 -0
  158. tests/test_integration.py +357 -0
  159. tests/test_item_dedup_redis_key.py +123 -0
  160. tests/test_parsel.py +30 -0
  161. tests/test_performance.py +328 -0
  162. tests/test_proxy_health_check.py +32 -32
  163. tests/test_proxy_middleware_integration.py +136 -136
  164. tests/test_proxy_providers.py +56 -56
  165. tests/test_proxy_stats.py +19 -19
  166. tests/test_proxy_strategies.py +59 -59
  167. tests/test_queue_manager_redis_key.py +177 -0
  168. tests/test_redis_config.py +28 -28
  169. tests/test_redis_connection_pool.py +295 -0
  170. tests/test_redis_key_naming.py +182 -0
  171. tests/test_redis_key_validator.py +124 -0
  172. tests/test_redis_queue.py +224 -224
  173. tests/test_request_serialization.py +70 -70
  174. tests/test_response_improvements.py +153 -0
  175. tests/test_scheduler.py +241 -241
  176. tests/test_simple_response.py +62 -0
  177. tests/test_telecom_spider_redis_key.py +206 -0
  178. tests/test_template_content.py +88 -0
  179. tests/test_template_redis_key.py +135 -0
  180. tests/test_tools.py +154 -0
  181. tests/tools_example.py +258 -0
  182. crawlo/core/enhanced_engine.py +0 -190
  183. crawlo-1.1.4.dist-info/RECORD +0 -117
  184. {crawlo-1.1.4.dist-info → crawlo-1.1.5.dist-info}/WHEEL +0 -0
  185. {crawlo-1.1.4.dist-info → crawlo-1.1.5.dist-info}/entry_points.txt +0 -0
  186. {crawlo-1.1.4.dist-info → crawlo-1.1.5.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,194 @@
1
+ #!/usr/bin/env python3
2
+ # -*- coding: utf-8 -*-
3
+ """
4
+ 配置验证器测试脚本
5
+ 用于验证配置验证器的功能
6
+ """
7
+ import sys
8
+ import os
9
+ import unittest
10
+
11
+ # 添加项目根目录到路径
12
+ sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
13
+
14
+ from crawlo.config_validator import ConfigValidator, validate_config
15
+
16
+
17
+ class TestConfigValidator(unittest.TestCase):
18
+ """配置验证器测试类"""
19
+
20
+ def setUp(self):
21
+ """测试前准备"""
22
+ self.validator = ConfigValidator()
23
+
24
+ def test_valid_standalone_config(self):
25
+ """测试有效的单机配置"""
26
+ config = {
27
+ 'PROJECT_NAME': 'test_project',
28
+ 'QUEUE_TYPE': 'memory',
29
+ 'CONCURRENCY': 8,
30
+ 'DOWNLOAD_DELAY': 1.0,
31
+ 'DOWNLOAD_TIMEOUT': 30,
32
+ 'CONNECTION_POOL_LIMIT': 50,
33
+ 'SCHEDULER_MAX_QUEUE_SIZE': 2000,
34
+ 'LOG_LEVEL': 'INFO',
35
+ 'MIDDLEWARES': [
36
+ 'crawlo.middleware.request_ignore.RequestIgnoreMiddleware',
37
+ 'crawlo.middleware.download_delay.DownloadDelayMiddleware'
38
+ ],
39
+ 'PIPELINES': [
40
+ 'crawlo.pipelines.console_pipeline.ConsolePipeline'
41
+ ]
42
+ }
43
+
44
+ is_valid, errors, warnings = self.validator.validate(config)
45
+ self.assertTrue(is_valid)
46
+ self.assertEqual(len(errors), 0)
47
+
48
+ def test_valid_distributed_config(self):
49
+ """测试有效的分布式配置"""
50
+ config = {
51
+ 'PROJECT_NAME': 'test_project',
52
+ 'QUEUE_TYPE': 'redis',
53
+ 'CONCURRENCY': 16,
54
+ 'DOWNLOAD_DELAY': 1.0,
55
+ 'DOWNLOAD_TIMEOUT': 30,
56
+ 'CONNECTION_POOL_LIMIT': 50,
57
+ 'SCHEDULER_MAX_QUEUE_SIZE': 2000,
58
+ 'SCHEDULER_QUEUE_NAME': 'crawlo:test_project:queue:requests',
59
+ 'REDIS_HOST': '127.0.0.1',
60
+ 'REDIS_PORT': 6379,
61
+ 'REDIS_URL': 'redis://127.0.0.1:6379/0',
62
+ 'LOG_LEVEL': 'INFO',
63
+ 'MIDDLEWARES': [
64
+ 'crawlo.middleware.request_ignore.RequestIgnoreMiddleware',
65
+ 'crawlo.middleware.download_delay.DownloadDelayMiddleware'
66
+ ],
67
+ 'PIPELINES': [
68
+ 'crawlo.pipelines.console_pipeline.ConsolePipeline'
69
+ ]
70
+ }
71
+
72
+ is_valid, errors, warnings = self.validator.validate(config)
73
+ self.assertTrue(is_valid)
74
+ self.assertEqual(len(errors), 0)
75
+
76
+ def test_invalid_project_name(self):
77
+ """测试无效的项目名称"""
78
+ config = {
79
+ 'PROJECT_NAME': '', # 空字符串
80
+ 'QUEUE_TYPE': 'memory',
81
+ 'CONCURRENCY': 8
82
+ }
83
+
84
+ is_valid, errors, warnings = self.validator.validate(config)
85
+ self.assertFalse(is_valid)
86
+ self.assertIn("PROJECT_NAME 必须是非空字符串", errors)
87
+
88
+ def test_invalid_concurrency(self):
89
+ """测试无效的并发数"""
90
+ config = {
91
+ 'PROJECT_NAME': 'test_project',
92
+ 'QUEUE_TYPE': 'memory',
93
+ 'CONCURRENCY': -1 # 负数
94
+ }
95
+
96
+ is_valid, errors, warnings = self.validator.validate(config)
97
+ self.assertFalse(is_valid)
98
+ self.assertIn("CONCURRENCY 必须是正整数", errors)
99
+
100
+ def test_invalid_queue_type(self):
101
+ """测试无效的队列类型"""
102
+ config = {
103
+ 'PROJECT_NAME': 'test_project',
104
+ 'QUEUE_TYPE': 'invalid_type', # 无效类型
105
+ 'CONCURRENCY': 8
106
+ }
107
+
108
+ is_valid, errors, warnings = self.validator.validate(config)
109
+ self.assertFalse(is_valid)
110
+ self.assertIn("QUEUE_TYPE 必须是以下值之一: ['memory', 'redis', 'auto']", errors)
111
+
112
+ def test_invalid_redis_queue_name(self):
113
+ """测试无效的Redis队列名称"""
114
+ config = {
115
+ 'PROJECT_NAME': 'test_project',
116
+ 'QUEUE_TYPE': 'redis',
117
+ 'CONCURRENCY': 8,
118
+ 'SCHEDULER_QUEUE_NAME': 'invalid_queue_name' # 不符合命名规范
119
+ }
120
+
121
+ is_valid, errors, warnings = self.validator.validate(config)
122
+ self.assertTrue(is_valid) # 队列名称错误是警告,不是错误
123
+ self.assertGreater(len(warnings), 0)
124
+ self.assertTrue(any("Redis队列名称" in warning for warning in warnings))
125
+
126
+ def test_missing_redis_queue_name(self):
127
+ """测试缺少Redis队列名称"""
128
+ config = {
129
+ 'PROJECT_NAME': 'test_project',
130
+ 'QUEUE_TYPE': 'redis',
131
+ 'CONCURRENCY': 8
132
+ # 缺少 SCHEDULER_QUEUE_NAME
133
+ }
134
+
135
+ is_valid, errors, warnings = self.validator.validate(config)
136
+ self.assertFalse(is_valid)
137
+ self.assertIn("使用Redis队列时,SCHEDULER_QUEUE_NAME 不能为空", errors)
138
+
139
+ def test_invalid_redis_port(self):
140
+ """测试无效的Redis端口"""
141
+ config = {
142
+ 'PROJECT_NAME': 'test_project',
143
+ 'QUEUE_TYPE': 'redis',
144
+ 'CONCURRENCY': 8,
145
+ 'SCHEDULER_QUEUE_NAME': 'crawlo:test_project:queue:requests',
146
+ 'REDIS_HOST': '127.0.0.1',
147
+ 'REDIS_PORT': 99999 # 无效端口
148
+ }
149
+
150
+ is_valid, errors, warnings = self.validator.validate(config)
151
+ self.assertFalse(is_valid)
152
+ self.assertIn("REDIS_PORT 必须是1-65535之间的整数", errors)
153
+
154
+ def test_invalid_log_level(self):
155
+ """测试无效的日志级别"""
156
+ config = {
157
+ 'PROJECT_NAME': 'test_project',
158
+ 'QUEUE_TYPE': 'memory',
159
+ 'CONCURRENCY': 8,
160
+ 'LOG_LEVEL': 'INVALID_LEVEL' # 无效日志级别
161
+ }
162
+
163
+ is_valid, errors, warnings = self.validator.validate(config)
164
+ self.assertFalse(is_valid)
165
+ self.assertIn("LOG_LEVEL 必须是以下值之一: ['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL']", errors)
166
+
167
+ def test_convenience_function(self):
168
+ """测试便利函数"""
169
+ config = {
170
+ 'PROJECT_NAME': 'test_project',
171
+ 'QUEUE_TYPE': 'memory',
172
+ 'CONCURRENCY': 8,
173
+ 'LOG_LEVEL': 'INFO'
174
+ }
175
+
176
+ is_valid, errors, warnings = validate_config(config)
177
+ self.assertTrue(is_valid)
178
+ self.assertEqual(len(errors), 0)
179
+
180
+
181
+ def main():
182
+ """主测试函数"""
183
+ print("🚀 开始配置验证器测试...")
184
+ print("=" * 50)
185
+
186
+ # 运行测试
187
+ unittest.main(argv=['first-arg-is-ignored'], exit=False, verbosity=2)
188
+
189
+ print("=" * 50)
190
+ print("✅ 配置验证器测试完成")
191
+
192
+
193
+ if __name__ == "__main__":
194
+ main()
@@ -0,0 +1,124 @@
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ 日期工具测试
5
+ """
6
+ import unittest
7
+ from crawlo.tools import (
8
+ TimeUtils,
9
+ parse_time,
10
+ format_time,
11
+ time_diff,
12
+ to_timestamp,
13
+ to_datetime,
14
+ now,
15
+ to_timezone,
16
+ to_utc,
17
+ to_local
18
+ )
19
+
20
+
21
+ class TestDateTools(unittest.TestCase):
22
+ """日期工具测试类"""
23
+
24
+ def test_date_parsing(self):
25
+ """测试日期解析功能"""
26
+ # 测试标准格式
27
+ dt = parse_time("2025-09-10 14:30:00")
28
+ self.assertIsNotNone(dt)
29
+ self.assertEqual(dt.year, 2025)
30
+ self.assertEqual(dt.month, 9)
31
+ self.assertEqual(dt.day, 10)
32
+ self.assertEqual(dt.hour, 14)
33
+ self.assertEqual(dt.minute, 30)
34
+ self.assertEqual(dt.second, 0)
35
+
36
+ # 测试不同格式
37
+ dt2 = parse_time("September 10, 2025 2:30 PM")
38
+ self.assertIsNotNone(dt2)
39
+ self.assertEqual(dt2.year, 2025)
40
+ self.assertEqual(dt2.month, 9)
41
+ self.assertEqual(dt2.day, 10)
42
+ self.assertEqual(dt2.hour, 14)
43
+ self.assertEqual(dt2.minute, 30)
44
+
45
+ def test_date_formatting(self):
46
+ """测试日期格式化功能"""
47
+ dt = parse_time("2025-09-10 14:30:00")
48
+
49
+ # 测试标准格式
50
+ formatted = format_time(dt, "%Y-%m-%d")
51
+ self.assertEqual(formatted, "2025-09-10")
52
+
53
+ # 测试中文格式
54
+ formatted_cn = format_time(dt, "%Y年%m月%d日")
55
+ self.assertEqual(formatted_cn, "2025年09月10日")
56
+
57
+ def test_time_difference(self):
58
+ """测试时间差计算功能"""
59
+ start_time = "2025-09-10 10:00:00"
60
+ end_time = "2025-09-10 14:30:45"
61
+
62
+ # 测试秒级差值
63
+ diff_seconds = time_diff(start_time, end_time, "seconds")
64
+ self.assertEqual(diff_seconds, 16245) # 4小时30分45秒 = 16245秒
65
+
66
+ # 测试分钟级差值
67
+ diff_minutes = time_diff(start_time, end_time, "minutes")
68
+ self.assertEqual(diff_minutes, 270) # 270分钟
69
+
70
+ # 测试小时级差值
71
+ diff_hours = time_diff(start_time, end_time, "hours")
72
+ self.assertEqual(diff_hours, 4) # 4小时
73
+
74
+ def test_timestamp_conversion(self):
75
+ """测试时间戳转换功能"""
76
+ # 测试转换为时间戳
77
+ dt = parse_time("2025-09-10 14:30:00")
78
+ timestamp = to_timestamp(dt)
79
+ self.assertIsInstance(timestamp, float)
80
+
81
+ # 测试从时间戳转换
82
+ dt_from_ts = to_datetime(timestamp)
83
+ self.assertEqual(dt.year, dt_from_ts.year)
84
+ self.assertEqual(dt.month, dt_from_ts.month)
85
+ self.assertEqual(dt.day, dt_from_ts.day)
86
+ self.assertEqual(dt.hour, dt_from_ts.hour)
87
+ self.assertEqual(dt.minute, dt_from_ts.minute)
88
+ self.assertEqual(dt.second, dt_from_ts.second)
89
+
90
+ def test_timezone_conversion(self):
91
+ """测试时区转换功能"""
92
+ dt = parse_time("2025-09-10 14:30:00")
93
+
94
+ # 测试UTC转换
95
+ utc_time = to_utc(dt)
96
+ self.assertIsNotNone(utc_time)
97
+
98
+ # 测试本地时区转换
99
+ local_time = to_local(dt)
100
+ self.assertIsNotNone(local_time)
101
+
102
+ # 测试指定时区转换
103
+ ny_time = to_timezone(dt, "America/New_York")
104
+ self.assertIsNotNone(ny_time)
105
+
106
+ def test_time_utils_class(self):
107
+ """测试TimeUtils类方法"""
108
+ # 测试日期加减
109
+ base_date = "2025-09-10"
110
+ plus_30_days = TimeUtils.add_days(base_date, 30)
111
+ self.assertEqual(plus_30_days.month, 10)
112
+ self.assertEqual(plus_30_days.day, 10)
113
+
114
+ # 测试月份加减
115
+ plus_3_months = TimeUtils.add_months(base_date, 3)
116
+ self.assertEqual(plus_3_months.month, 12)
117
+
118
+ # 测试闰年判断
119
+ self.assertTrue(TimeUtils.is_leap_year(2024))
120
+ self.assertFalse(TimeUtils.is_leap_year(2025))
121
+
122
+
123
+ if __name__ == '__main__':
124
+ unittest.main()
@@ -0,0 +1,125 @@
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ 测试动态下载器(Selenium和Playwright)的代理功能
5
+ """
6
+
7
+ import asyncio
8
+ from crawlo.tools import AuthenticatedProxy
9
+
10
+
11
+ def test_selenium_proxy_configuration():
12
+ """测试Selenium下载器的代理配置"""
13
+ print("=== 测试Selenium下载器的代理配置 ===")
14
+
15
+ # 代理配置
16
+ proxy_config = {
17
+ "http": "http://dwe20241014:Dwe0101014@182.201.243.186:58111",
18
+ "https": "http://dwe20241014:Dwe0101014@182.201.243.186:58111"
19
+ }
20
+
21
+ # 创建代理对象
22
+ proxy_url = proxy_config["http"]
23
+ proxy = AuthenticatedProxy(proxy_url)
24
+
25
+ print(f"原始代理URL: {proxy_url}")
26
+ print(f"清洁URL: {proxy.clean_url}")
27
+ print(f"认证信息: {proxy.get_auth_credentials()}")
28
+
29
+ # Selenium的代理设置方式
30
+ print(f"\nSelenium代理设置方式:")
31
+ print(f" 1. 在爬虫设置中配置:")
32
+ print(f" settings = {{")
33
+ print(f" 'SELENIUM_PROXY': '{proxy.clean_url}',")
34
+ print(f" }}")
35
+
36
+ # 对于带认证的代理,需要特殊处理
37
+ if proxy.username and proxy.password:
38
+ print(f"\n 2. 带认证代理的处理:")
39
+ print(f" - 用户名: {proxy.username}")
40
+ print(f" - 密码: {proxy.password}")
41
+ print(f" - 认证头: {proxy.get_auth_header()}")
42
+ print(f" - 注意: Selenium需要通过扩展或其他方式处理认证")
43
+
44
+ print("\nSelenium测试完成!")
45
+
46
+
47
+ async def test_playwright_proxy_configuration():
48
+ """测试Playwright下载器的代理配置"""
49
+ print("\n=== 测试Playwright下载器的代理配置 ===")
50
+
51
+ # 代理配置
52
+ proxy_config = {
53
+ "http": "http://dwe20241014:Dwe0101014@182.201.243.186:58111",
54
+ "https": "http://dwe20241014:Dwe0101014@182.201.243.186:58111"
55
+ }
56
+
57
+ # 创建代理对象
58
+ proxy_url = proxy_config["http"]
59
+ proxy = AuthenticatedProxy(proxy_url)
60
+
61
+ print(f"原始代理URL: {proxy_url}")
62
+ print(f"清洁URL: {proxy.clean_url}")
63
+ print(f"认证信息: {proxy.get_auth_credentials()}")
64
+
65
+ # Playwright的代理设置方式
66
+ print(f"\nPlaywright代理设置方式:")
67
+ print(f" 1. 简单代理配置:")
68
+ print(f" settings = {{")
69
+ print(f" 'PLAYWRIGHT_PROXY': '{proxy.clean_url}',")
70
+ print(f" }}")
71
+
72
+ # 对于带认证的代理,Playwright可以直接在代理配置中包含认证信息
73
+ if proxy.username and proxy.password:
74
+ print(f"\n 2. 带认证的代理配置:")
75
+ print(f" settings = {{")
76
+ print(f" 'PLAYWRIGHT_PROXY': {{")
77
+ print(f" 'server': '{proxy.clean_url}',")
78
+ print(f" 'username': '{proxy.username}',")
79
+ print(f" 'password': '{proxy.password}'")
80
+ print(f" }}")
81
+ print(f" }}")
82
+
83
+ print("\nPlaywright测试完成!")
84
+
85
+
86
+ def show_proxy_usage_examples():
87
+ """显示代理使用示例"""
88
+ print("\n=== 代理使用示例 ===")
89
+
90
+ # 代理配置示例
91
+ proxy_examples = [
92
+ "http://username:password@proxy.example.com:8080", # 带认证HTTP代理
93
+ "https://user:pass@secure-proxy.example.com:443", # 带认证HTTPS代理
94
+ "http://proxy.example.com:8080", # 不带认证代理
95
+ "socks5://username:password@socks-proxy.example.com:1080" # SOCKS5代理
96
+ ]
97
+
98
+ for i, proxy_url in enumerate(proxy_examples, 1):
99
+ print(f"\n示例 {i}: {proxy_url}")
100
+ try:
101
+ proxy = AuthenticatedProxy(proxy_url)
102
+ print(f" 清洁URL: {proxy.clean_url}")
103
+ print(f" 用户名: {proxy.username or '无'}")
104
+ print(f" 密码: {proxy.password or '无'}")
105
+ print(f" 是否有效: {proxy.is_valid()}")
106
+ if proxy.username and proxy.password:
107
+ print(f" 认证头: {proxy.get_auth_header()}")
108
+ except Exception as e:
109
+ print(f" 错误: {e}")
110
+
111
+
112
+ async def main():
113
+ """主测试函数"""
114
+ print("开始测试动态下载器的代理功能...\n")
115
+
116
+ # 测试各个下载器
117
+ test_selenium_proxy_configuration()
118
+ await test_playwright_proxy_configuration()
119
+ show_proxy_usage_examples()
120
+
121
+ print("\n所有测试完成!")
122
+
123
+
124
+ if __name__ == "__main__":
125
+ asyncio.run(main())
@@ -0,0 +1,93 @@
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ 测试动态下载器(Selenium和Playwright)的代理功能
5
+ """
6
+
7
+ import asyncio
8
+ from crawlo.network.request import Request
9
+ from crawlo.tools import AuthenticatedProxy
10
+
11
+
12
+ def test_proxy_with_selenium():
13
+ """测试Selenium下载器与认证代理"""
14
+ print("=== 测试Selenium下载器与认证代理 ===")
15
+
16
+ # 代理配置
17
+ proxy_config = {
18
+ "http": "http://dwe20241014:Dwe0101014@182.201.243.186:58111",
19
+ "https": "http://dwe20241014:Dwe0101014@182.201.243.186:58111"
20
+ }
21
+
22
+ # 创建代理对象
23
+ proxy_url = proxy_config["http"]
24
+ proxy = AuthenticatedProxy(proxy_url)
25
+
26
+ print(f"原始代理URL: {proxy_url}")
27
+ print(f"清洁URL: {proxy.clean_url}")
28
+ print(f"认证信息: {proxy.get_auth_credentials()}")
29
+
30
+ # Selenium的代理设置方式
31
+ print(f"Selenium代理设置方式:")
32
+ print(f" 1. 在设置中配置: SELENIUM_PROXY = '{proxy.clean_url}'")
33
+ print(f" 2. 认证信息需要通过其他方式处理")
34
+
35
+ # 对于带认证的代理,Selenium需要特殊处理
36
+ if proxy.username and proxy.password:
37
+ print(f" 3. 认证信息:")
38
+ print(f" 用户名: {proxy.username}")
39
+ print(f" 密码: {proxy.password}")
40
+ print(f" 认证头: {proxy.get_auth_header()}")
41
+
42
+ print("Selenium测试完成!")
43
+
44
+
45
+ async def test_proxy_with_playwright():
46
+ """测试Playwright下载器与认证代理"""
47
+ print("\n=== 测试Playwright下载器与认证代理 ===")
48
+
49
+ # 代理配置
50
+ proxy_config = {
51
+ "http": "http://dwe20241014:Dwe0101014@182.201.243.186:58111",
52
+ "https": "http://dwe20241014:Dwe0101014@182.201.243.186:58111"
53
+ }
54
+
55
+ # 创建代理对象
56
+ proxy_url = proxy_config["http"]
57
+ proxy = AuthenticatedProxy(proxy_url)
58
+
59
+ print(f"原始代理URL: {proxy_url}")
60
+ print(f"清洁URL: {proxy.clean_url}")
61
+ print(f"认证信息: {proxy.get_auth_credentials()}")
62
+
63
+ # Playwright的代理设置方式
64
+ print(f"Playwright代理设置方式:")
65
+ print(f" 1. 在启动浏览器时配置代理:")
66
+ print(f" browser = await playwright.chromium.launch(proxy={{'server': '{proxy.clean_url}'}})")
67
+
68
+ # 对于带认证的代理,Playwright需要在代理配置中包含认证信息
69
+ if proxy.username and proxy.password:
70
+ print(f" 2. 带认证的代理配置:")
71
+ print(f" proxy_config = {{")
72
+ print(f" 'server': '{proxy.clean_url}',")
73
+ print(f" 'username': '{proxy.username}',")
74
+ print(f" 'password': '{proxy.password}'")
75
+ print(f" }}")
76
+ print(f" browser = await playwright.chromium.launch(proxy=proxy_config)")
77
+
78
+ print("Playwright测试完成!")
79
+
80
+
81
+ async def main():
82
+ """主测试函数"""
83
+ print("开始测试动态下载器的代理功能...\n")
84
+
85
+ # 测试各个下载器
86
+ test_proxy_with_selenium()
87
+ await test_proxy_with_playwright()
88
+
89
+ print("\n所有测试完成!")
90
+
91
+
92
+ if __name__ == "__main__":
93
+ asyncio.run(main())
@@ -0,0 +1,147 @@
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ 测试动态下载器(Selenium和Playwright)的代理配置逻辑
5
+ """
6
+
7
+ from crawlo.tools import AuthenticatedProxy
8
+
9
+
10
+ def test_selenium_proxy_logic():
11
+ """测试Selenium下载器的代理配置逻辑"""
12
+ print("=== 测试Selenium下载器的代理配置逻辑 ===")
13
+
14
+ # 代理配置
15
+ proxy_config = {
16
+ "http": "http://dwe20241014:Dwe0101014@182.201.243.186:58111",
17
+ "https": "http://dwe20241014:Dwe0101014@182.201.243.186:58111"
18
+ }
19
+
20
+ # 创建代理对象
21
+ proxy_url = proxy_config["http"]
22
+ proxy = AuthenticatedProxy(proxy_url)
23
+
24
+ print(f"原始代理URL: {proxy_url}")
25
+ print(f"清洁URL: {proxy.clean_url}")
26
+ print(f"认证信息: {proxy.get_auth_credentials()}")
27
+
28
+ # 模拟Selenium的代理配置逻辑
29
+ print(f"\nSelenium代理配置逻辑:")
30
+ print(f" 1. 在爬虫设置中配置:")
31
+ print(f" settings = {{")
32
+ print(f" 'SELENIUM_PROXY': '{proxy.clean_url}',")
33
+ print(f" }}")
34
+
35
+ # 对于带认证的代理,需要特殊处理
36
+ if proxy.username and proxy.password:
37
+ print(f"\n 2. 带认证代理的处理逻辑:")
38
+ print(f" - 用户名: {proxy.username}")
39
+ print(f" - 密码: {proxy.password}")
40
+ print(f" - 认证头: {proxy.get_auth_header()}")
41
+ print(f" - 处理方式: 通过浏览器扩展或手动输入认证信息")
42
+
43
+ print("\nSelenium配置逻辑测试完成!")
44
+
45
+
46
+ def test_playwright_proxy_logic():
47
+ """测试Playwright下载器的代理配置逻辑"""
48
+ print("\n=== 测试Playwright下载器的代理配置逻辑 ===")
49
+
50
+ # 代理配置
51
+ proxy_config = {
52
+ "http": "http://dwe20241014:Dwe0101014@182.201.243.186:58111",
53
+ "https": "http://dwe20241014:Dwe0101014@182.201.243.186:58111"
54
+ }
55
+
56
+ # 创建代理对象
57
+ proxy_url = proxy_config["http"]
58
+ proxy = AuthenticatedProxy(proxy_url)
59
+
60
+ print(f"原始代理URL: {proxy_url}")
61
+ print(f"清洁URL: {proxy.clean_url}")
62
+ print(f"认证信息: {proxy.get_auth_credentials()}")
63
+
64
+ # 模拟Playwright的代理配置逻辑
65
+ print(f"\nPlaywright代理配置逻辑:")
66
+ print(f" 1. 简单代理配置:")
67
+ print(f" settings = {{")
68
+ print(f" 'PLAYWRIGHT_PROXY': '{proxy.clean_url}',")
69
+ print(f" }}")
70
+
71
+ # 对于带认证的代理,Playwright可以直接在代理配置中包含认证信息
72
+ if proxy.username and proxy.password:
73
+ print(f"\n 2. 带认证的代理配置逻辑:")
74
+ print(f" settings = {{")
75
+ print(f" 'PLAYWRIGHT_PROXY': {{")
76
+ print(f" 'server': '{proxy.clean_url}',")
77
+ print(f" 'username': '{proxy.username}',")
78
+ print(f" 'password': '{proxy.password}'")
79
+ print(f" }}")
80
+ print(f" }}")
81
+ print(f" 实现方式: 在启动浏览器时传递proxy参数")
82
+
83
+ print("\nPlaywright配置逻辑测试完成!")
84
+
85
+
86
+ def show_proxy_configuration_examples():
87
+ """显示代理配置示例"""
88
+ print("\n=== 代理配置示例 ===")
89
+
90
+ # 不同类型的代理配置示例
91
+ examples = [
92
+ {
93
+ "name": "带认证HTTP代理",
94
+ "url": "http://username:password@proxy.example.com:8080",
95
+ "selenium_config": "SELENIUM_PROXY = 'http://proxy.example.com:8080'",
96
+ "playwright_config": "PLAYWRIGHT_PROXY = {'server': 'http://proxy.example.com:8080', 'username': 'username', 'password': 'password'}"
97
+ },
98
+ {
99
+ "name": "带认证HTTPS代理",
100
+ "url": "https://user:pass@secure-proxy.example.com:443",
101
+ "selenium_config": "SELENIUM_PROXY = 'https://secure-proxy.example.com:443'",
102
+ "playwright_config": "PLAYWRIGHT_PROXY = {'server': 'https://secure-proxy.example.com:443', 'username': 'user', 'password': 'pass'}"
103
+ },
104
+ {
105
+ "name": "SOCKS5代理",
106
+ "url": "socks5://username:password@socks-proxy.example.com:1080",
107
+ "selenium_config": "SELENIUM_PROXY = 'socks5://socks-proxy.example.com:1080'",
108
+ "playwright_config": "PLAYWRIGHT_PROXY = {'server': 'socks5://socks-proxy.example.com:1080', 'username': 'username', 'password': 'password'}"
109
+ },
110
+ {
111
+ "name": "不带认证代理",
112
+ "url": "http://proxy.example.com:8080",
113
+ "selenium_config": "SELENIUM_PROXY = 'http://proxy.example.com:8080'",
114
+ "playwright_config": "PLAYWRIGHT_PROXY = 'http://proxy.example.com:8080'"
115
+ }
116
+ ]
117
+
118
+ for i, example in enumerate(examples, 1):
119
+ print(f"\n示例 {i}: {example['name']}")
120
+ print(f" 代理URL: {example['url']}")
121
+ proxy = AuthenticatedProxy(example['url'])
122
+ print(f" 清洁URL: {proxy.clean_url}")
123
+ print(f" 用户名: {proxy.username or '无'}")
124
+ print(f" 密码: {proxy.password or '无'}")
125
+ print(f" Selenium配置: {example['selenium_config']}")
126
+ print(f" Playwright配置: {example['playwright_config']}")
127
+
128
+
129
+ def main():
130
+ """主函数"""
131
+ print("开始测试动态下载器的代理配置逻辑...\n")
132
+
133
+ # 测试各个下载器的配置逻辑
134
+ test_selenium_proxy_logic()
135
+ test_playwright_proxy_logic()
136
+ show_proxy_configuration_examples()
137
+
138
+ print("\n所有配置逻辑测试完成!")
139
+ print("\n总结:")
140
+ print("1. Selenium和Playwright都支持代理配置")
141
+ print("2. 带认证的代理需要特殊处理")
142
+ print("3. Playwright对带认证代理的支持更直接")
143
+ print("4. Selenium需要通过扩展或其他方式处理认证")
144
+
145
+
146
+ if __name__ == "__main__":
147
+ main()