crawlo 1.4.1__py3-none-any.whl → 1.4.3__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of crawlo might be problematic. Click here for more details.

Files changed (326) hide show
  1. crawlo/__init__.py +93 -93
  2. crawlo/__version__.py +1 -1
  3. crawlo/cli.py +75 -75
  4. crawlo/commands/__init__.py +14 -14
  5. crawlo/commands/check.py +594 -594
  6. crawlo/commands/genspider.py +151 -151
  7. crawlo/commands/help.py +138 -138
  8. crawlo/commands/list.py +155 -155
  9. crawlo/commands/run.py +341 -341
  10. crawlo/commands/startproject.py +436 -436
  11. crawlo/commands/stats.py +187 -187
  12. crawlo/commands/utils.py +196 -196
  13. crawlo/config.py +312 -312
  14. crawlo/config_validator.py +277 -277
  15. crawlo/core/__init__.py +52 -52
  16. crawlo/core/engine.py +438 -439
  17. crawlo/core/processor.py +47 -47
  18. crawlo/core/scheduler.py +291 -257
  19. crawlo/crawler.py +650 -650
  20. crawlo/data/__init__.py +5 -5
  21. crawlo/data/user_agents.py +194 -194
  22. crawlo/downloader/__init__.py +273 -273
  23. crawlo/downloader/aiohttp_downloader.py +233 -228
  24. crawlo/downloader/cffi_downloader.py +245 -245
  25. crawlo/downloader/httpx_downloader.py +259 -259
  26. crawlo/downloader/hybrid_downloader.py +212 -212
  27. crawlo/downloader/playwright_downloader.py +402 -402
  28. crawlo/downloader/selenium_downloader.py +472 -472
  29. crawlo/event.py +11 -11
  30. crawlo/exceptions.py +81 -81
  31. crawlo/extension/__init__.py +63 -63
  32. crawlo/extension/health_check.py +141 -141
  33. crawlo/extension/log_interval.py +94 -94
  34. crawlo/extension/log_stats.py +70 -70
  35. crawlo/extension/logging_extension.py +61 -61
  36. crawlo/extension/memory_monitor.py +104 -104
  37. crawlo/extension/performance_profiler.py +133 -133
  38. crawlo/extension/request_recorder.py +107 -107
  39. crawlo/factories/__init__.py +27 -27
  40. crawlo/factories/base.py +68 -68
  41. crawlo/factories/crawler.py +103 -103
  42. crawlo/factories/registry.py +84 -84
  43. crawlo/filters/__init__.py +154 -154
  44. crawlo/filters/aioredis_filter.py +257 -257
  45. crawlo/filters/memory_filter.py +269 -269
  46. crawlo/framework.py +292 -292
  47. crawlo/initialization/__init__.py +44 -44
  48. crawlo/initialization/built_in.py +425 -425
  49. crawlo/initialization/context.py +141 -141
  50. crawlo/initialization/core.py +193 -193
  51. crawlo/initialization/phases.py +148 -148
  52. crawlo/initialization/registry.py +145 -145
  53. crawlo/items/__init__.py +23 -23
  54. crawlo/items/base.py +23 -23
  55. crawlo/items/fields.py +52 -52
  56. crawlo/items/items.py +104 -104
  57. crawlo/logging/__init__.py +45 -37
  58. crawlo/logging/async_handler.py +181 -0
  59. crawlo/logging/config.py +196 -96
  60. crawlo/logging/factory.py +171 -128
  61. crawlo/logging/manager.py +111 -111
  62. crawlo/logging/monitor.py +153 -0
  63. crawlo/logging/sampler.py +167 -0
  64. crawlo/middleware/__init__.py +21 -21
  65. crawlo/middleware/default_header.py +132 -132
  66. crawlo/middleware/download_delay.py +104 -104
  67. crawlo/middleware/middleware_manager.py +135 -135
  68. crawlo/middleware/offsite.py +123 -123
  69. crawlo/middleware/proxy.py +386 -386
  70. crawlo/middleware/request_ignore.py +86 -86
  71. crawlo/middleware/response_code.py +150 -150
  72. crawlo/middleware/response_filter.py +136 -136
  73. crawlo/middleware/retry.py +124 -124
  74. crawlo/middleware/simple_proxy.py +65 -65
  75. crawlo/mode_manager.py +219 -219
  76. crawlo/network/__init__.py +21 -21
  77. crawlo/network/request.py +379 -379
  78. crawlo/network/response.py +359 -359
  79. crawlo/pipelines/__init__.py +21 -21
  80. crawlo/pipelines/bloom_dedup_pipeline.py +146 -146
  81. crawlo/pipelines/console_pipeline.py +39 -39
  82. crawlo/pipelines/csv_pipeline.py +316 -316
  83. crawlo/pipelines/database_dedup_pipeline.py +197 -197
  84. crawlo/pipelines/json_pipeline.py +218 -218
  85. crawlo/pipelines/memory_dedup_pipeline.py +105 -105
  86. crawlo/pipelines/mongo_pipeline.py +131 -131
  87. crawlo/pipelines/mysql_pipeline.py +325 -325
  88. crawlo/pipelines/pipeline_manager.py +100 -84
  89. crawlo/pipelines/redis_dedup_pipeline.py +156 -156
  90. crawlo/project.py +349 -338
  91. crawlo/queue/pqueue.py +42 -42
  92. crawlo/queue/queue_manager.py +526 -522
  93. crawlo/queue/redis_priority_queue.py +370 -367
  94. crawlo/settings/__init__.py +7 -7
  95. crawlo/settings/default_settings.py +284 -284
  96. crawlo/settings/setting_manager.py +219 -219
  97. crawlo/spider/__init__.py +657 -657
  98. crawlo/stats_collector.py +73 -73
  99. crawlo/subscriber.py +129 -129
  100. crawlo/task_manager.py +138 -138
  101. crawlo/templates/crawlo.cfg.tmpl +10 -10
  102. crawlo/templates/project/__init__.py.tmpl +3 -3
  103. crawlo/templates/project/items.py.tmpl +17 -17
  104. crawlo/templates/project/middlewares.py.tmpl +118 -118
  105. crawlo/templates/project/pipelines.py.tmpl +96 -96
  106. crawlo/templates/project/settings.py.tmpl +170 -170
  107. crawlo/templates/project/settings_distributed.py.tmpl +169 -169
  108. crawlo/templates/project/settings_gentle.py.tmpl +166 -166
  109. crawlo/templates/project/settings_high_performance.py.tmpl +167 -167
  110. crawlo/templates/project/settings_minimal.py.tmpl +65 -65
  111. crawlo/templates/project/settings_simple.py.tmpl +164 -164
  112. crawlo/templates/project/spiders/__init__.py.tmpl +9 -9
  113. crawlo/templates/run.py.tmpl +34 -34
  114. crawlo/templates/spider/spider.py.tmpl +143 -143
  115. crawlo/templates/spiders_init.py.tmpl +9 -9
  116. crawlo/tools/__init__.py +200 -200
  117. crawlo/tools/anti_crawler.py +268 -268
  118. crawlo/tools/authenticated_proxy.py +240 -240
  119. crawlo/tools/data_formatter.py +225 -225
  120. crawlo/tools/data_validator.py +180 -180
  121. crawlo/tools/date_tools.py +289 -289
  122. crawlo/tools/distributed_coordinator.py +384 -384
  123. crawlo/tools/encoding_converter.py +127 -127
  124. crawlo/tools/network_diagnostic.py +364 -364
  125. crawlo/tools/request_tools.py +82 -82
  126. crawlo/tools/retry_mechanism.py +224 -224
  127. crawlo/tools/scenario_adapter.py +262 -262
  128. crawlo/tools/text_cleaner.py +232 -232
  129. crawlo/utils/__init__.py +34 -34
  130. crawlo/utils/batch_processor.py +259 -259
  131. crawlo/utils/class_loader.py +25 -25
  132. crawlo/utils/controlled_spider_mixin.py +439 -439
  133. crawlo/utils/db_helper.py +343 -343
  134. crawlo/utils/enhanced_error_handler.py +356 -356
  135. crawlo/utils/env_config.py +142 -142
  136. crawlo/utils/error_handler.py +165 -165
  137. crawlo/utils/fingerprint.py +122 -122
  138. crawlo/utils/func_tools.py +82 -82
  139. crawlo/utils/large_scale_config.py +286 -286
  140. crawlo/utils/large_scale_helper.py +344 -344
  141. crawlo/utils/log.py +79 -79
  142. crawlo/utils/performance_monitor.py +285 -285
  143. crawlo/utils/queue_helper.py +175 -175
  144. crawlo/utils/redis_connection_pool.py +388 -388
  145. crawlo/utils/redis_key_validator.py +198 -198
  146. crawlo/utils/request.py +267 -267
  147. crawlo/utils/request_serializer.py +225 -225
  148. crawlo/utils/spider_loader.py +61 -61
  149. crawlo/utils/system.py +11 -11
  150. crawlo/utils/tools.py +4 -4
  151. crawlo/utils/url.py +39 -39
  152. crawlo-1.4.3.dist-info/METADATA +190 -0
  153. crawlo-1.4.3.dist-info/RECORD +326 -0
  154. examples/__init__.py +7 -7
  155. examples/test_project/__init__.py +7 -7
  156. examples/test_project/run.py +34 -34
  157. examples/test_project/test_project/__init__.py +3 -3
  158. examples/test_project/test_project/items.py +17 -17
  159. examples/test_project/test_project/middlewares.py +118 -118
  160. examples/test_project/test_project/pipelines.py +96 -96
  161. examples/test_project/test_project/settings.py +169 -169
  162. examples/test_project/test_project/spiders/__init__.py +9 -9
  163. examples/test_project/test_project/spiders/of_week_dis.py +143 -143
  164. tests/__init__.py +7 -7
  165. tests/advanced_tools_example.py +275 -275
  166. tests/authenticated_proxy_example.py +106 -106
  167. tests/baidu_performance_test.py +108 -108
  168. tests/baidu_test.py +59 -59
  169. tests/cleaners_example.py +160 -160
  170. tests/comprehensive_framework_test.py +212 -212
  171. tests/comprehensive_test.py +81 -81
  172. tests/comprehensive_testing_summary.md +186 -186
  173. tests/config_validation_demo.py +142 -142
  174. tests/controlled_spider_example.py +205 -205
  175. tests/date_tools_example.py +180 -180
  176. tests/debug_configure.py +69 -69
  177. tests/debug_framework_logger.py +84 -84
  178. tests/debug_log_config.py +126 -126
  179. tests/debug_log_levels.py +63 -63
  180. tests/debug_pipelines.py +66 -66
  181. tests/detailed_log_test.py +233 -233
  182. tests/distributed_test.py +66 -66
  183. tests/distributed_test_debug.py +76 -76
  184. tests/dynamic_loading_example.py +523 -523
  185. tests/dynamic_loading_test.py +104 -104
  186. tests/env_config_example.py +133 -133
  187. tests/error_handling_example.py +171 -171
  188. tests/final_comprehensive_test.py +151 -151
  189. tests/final_log_test.py +260 -260
  190. tests/final_validation_test.py +182 -182
  191. tests/fix_log_test.py +142 -142
  192. tests/framework_performance_test.py +202 -202
  193. tests/log_buffering_test.py +111 -111
  194. tests/log_generation_timing_test.py +153 -153
  195. tests/optimized_performance_test.py +211 -211
  196. tests/performance_comparison.py +245 -245
  197. tests/queue_blocking_test.py +113 -113
  198. tests/queue_test.py +89 -89
  199. tests/redis_key_validation_demo.py +130 -130
  200. tests/request_params_example.py +150 -150
  201. tests/response_improvements_example.py +144 -144
  202. tests/scrapy_comparison/ofweek_scrapy.py +138 -138
  203. tests/scrapy_comparison/scrapy_test.py +133 -133
  204. tests/simple_command_test.py +119 -119
  205. tests/simple_crawlo_test.py +127 -127
  206. tests/simple_log_test.py +57 -57
  207. tests/simple_log_test2.py +137 -137
  208. tests/simple_optimization_test.py +128 -128
  209. tests/simple_queue_type_test.py +41 -41
  210. tests/simple_spider_test.py +49 -49
  211. tests/simple_test.py +47 -47
  212. tests/spider_log_timing_test.py +177 -177
  213. tests/test_advanced_tools.py +148 -148
  214. tests/test_all_commands.py +230 -230
  215. tests/test_all_pipeline_fingerprints.py +133 -133
  216. tests/test_all_redis_key_configs.py +145 -145
  217. tests/test_authenticated_proxy.py +141 -141
  218. tests/test_batch_processor.py +178 -178
  219. tests/test_cleaners.py +54 -54
  220. tests/test_component_factory.py +174 -174
  221. tests/test_comprehensive.py +146 -146
  222. tests/test_config_consistency.py +80 -80
  223. tests/test_config_merge.py +152 -152
  224. tests/test_config_validator.py +182 -182
  225. tests/test_controlled_spider_mixin.py +79 -79
  226. tests/test_crawlo_proxy_integration.py +108 -108
  227. tests/test_date_tools.py +123 -123
  228. tests/test_dedup_fix.py +220 -220
  229. tests/test_dedup_pipeline_consistency.py +125 -0
  230. tests/test_default_header_middleware.py +313 -313
  231. tests/test_distributed.py +65 -65
  232. tests/test_double_crawlo_fix.py +204 -204
  233. tests/test_double_crawlo_fix_simple.py +124 -124
  234. tests/test_download_delay_middleware.py +221 -221
  235. tests/test_downloader_proxy_compatibility.py +268 -268
  236. tests/test_dynamic_downloaders_proxy.py +124 -124
  237. tests/test_dynamic_proxy.py +92 -92
  238. tests/test_dynamic_proxy_config.py +146 -146
  239. tests/test_dynamic_proxy_real.py +109 -109
  240. tests/test_edge_cases.py +303 -303
  241. tests/test_enhanced_error_handler.py +270 -270
  242. tests/test_enhanced_error_handler_comprehensive.py +245 -245
  243. tests/test_env_config.py +121 -121
  244. tests/test_error_handler_compatibility.py +112 -112
  245. tests/test_factories.py +252 -252
  246. tests/test_final_validation.py +153 -153
  247. tests/test_fingerprint_consistency.py +135 -135
  248. tests/test_fingerprint_simple.py +51 -51
  249. tests/test_framework_env_usage.py +103 -103
  250. tests/test_framework_logger.py +66 -66
  251. tests/test_framework_startup.py +64 -64
  252. tests/test_get_component_logger.py +83 -83
  253. tests/test_hash_performance.py +99 -99
  254. tests/test_integration.py +169 -169
  255. tests/test_item_dedup_redis_key.py +122 -122
  256. tests/test_large_scale_config.py +112 -112
  257. tests/test_large_scale_helper.py +235 -235
  258. tests/test_logging_enhancements.py +375 -0
  259. tests/test_logging_final.py +185 -0
  260. tests/test_logging_integration.py +313 -0
  261. tests/test_logging_system.py +282 -282
  262. tests/test_middleware_debug.py +142 -0
  263. tests/test_mode_change.py +72 -72
  264. tests/test_mode_consistency.py +51 -51
  265. tests/test_offsite_middleware.py +244 -244
  266. tests/test_offsite_middleware_simple.py +203 -203
  267. tests/test_parsel.py +29 -29
  268. tests/test_performance.py +327 -327
  269. tests/test_performance_monitor.py +115 -115
  270. tests/test_pipeline_fingerprint_consistency.py +86 -86
  271. tests/test_priority_behavior.py +212 -0
  272. tests/test_priority_consistency.py +152 -0
  273. tests/test_priority_consistency_fixed.py +250 -0
  274. tests/test_proxy_api.py +264 -264
  275. tests/test_proxy_health_check.py +32 -32
  276. tests/test_proxy_middleware.py +121 -121
  277. tests/test_proxy_middleware_enhanced.py +216 -216
  278. tests/test_proxy_middleware_integration.py +136 -136
  279. tests/test_proxy_middleware_refactored.py +184 -184
  280. tests/test_proxy_providers.py +56 -56
  281. tests/test_proxy_stats.py +19 -19
  282. tests/test_proxy_strategies.py +59 -59
  283. tests/test_queue_empty_check.py +41 -41
  284. tests/test_queue_manager_double_crawlo.py +173 -173
  285. tests/test_queue_manager_redis_key.py +179 -179
  286. tests/test_queue_naming.py +154 -154
  287. tests/test_queue_type.py +106 -106
  288. tests/test_queue_type_redis_config_consistency.py +131 -0
  289. tests/test_random_headers_default.py +323 -0
  290. tests/test_random_headers_necessity.py +309 -0
  291. tests/test_random_user_agent.py +72 -72
  292. tests/test_real_scenario_proxy.py +195 -195
  293. tests/test_redis_config.py +28 -28
  294. tests/test_redis_connection_pool.py +294 -294
  295. tests/test_redis_key_naming.py +181 -181
  296. tests/test_redis_key_validator.py +123 -123
  297. tests/test_redis_queue.py +224 -224
  298. tests/test_redis_queue_name_fix.py +175 -175
  299. tests/test_redis_queue_type_fallback.py +130 -0
  300. tests/test_request_ignore_middleware.py +182 -182
  301. tests/test_request_params.py +111 -111
  302. tests/test_request_serialization.py +70 -70
  303. tests/test_response_code_middleware.py +349 -349
  304. tests/test_response_filter_middleware.py +427 -427
  305. tests/test_response_improvements.py +152 -152
  306. tests/test_retry_middleware.py +334 -242
  307. tests/test_retry_middleware_realistic.py +274 -0
  308. tests/test_scheduler.py +252 -252
  309. tests/test_scheduler_config_update.py +133 -133
  310. tests/test_simple_response.py +61 -61
  311. tests/test_telecom_spider_redis_key.py +205 -205
  312. tests/test_template_content.py +87 -87
  313. tests/test_template_redis_key.py +134 -134
  314. tests/test_tools.py +159 -159
  315. tests/test_user_agent_randomness.py +177 -0
  316. tests/test_user_agents.py +96 -96
  317. tests/tools_example.py +260 -260
  318. tests/untested_features_report.md +138 -138
  319. tests/verify_debug.py +51 -51
  320. tests/verify_distributed.py +117 -117
  321. tests/verify_log_fix.py +111 -111
  322. crawlo-1.4.1.dist-info/METADATA +0 -1199
  323. crawlo-1.4.1.dist-info/RECORD +0 -309
  324. {crawlo-1.4.1.dist-info → crawlo-1.4.3.dist-info}/WHEEL +0 -0
  325. {crawlo-1.4.1.dist-info → crawlo-1.4.3.dist-info}/entry_points.txt +0 -0
  326. {crawlo-1.4.1.dist-info → crawlo-1.4.3.dist-info}/top_level.txt +0 -0
@@ -1,52 +1,52 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- """
4
- 简化版指纹一致性测试
5
- ==============
6
- 验证框架中各组件对相同数据生成一致的指纹
7
- """
8
-
9
- import sys
10
- import os
11
-
12
- # 添加项目根目录到Python路径
13
- sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
14
-
15
- from crawlo.utils.fingerprint import FingerprintGenerator
16
-
17
-
18
- def test_fingerprint_consistency():
19
- """测试指纹一致性"""
20
- # 测试数据
21
- test_data = {
22
- "title": "Test Title",
23
- "url": "https://example.com",
24
- "content": "Test content",
25
- "price": 99.99
26
- }
27
-
28
- # 使用指纹生成器生成指纹
29
- fingerprint1 = FingerprintGenerator.data_fingerprint(test_data)
30
- fingerprint2 = FingerprintGenerator.data_fingerprint(test_data)
31
-
32
- # 验证相同数据生成相同指纹
33
- print(f"First fingerprint: {fingerprint1}")
34
- print(f"Second fingerprint: {fingerprint2}")
35
- print(f"指纹一致: {fingerprint1 == fingerprint2}")
36
-
37
- # 测试请求指纹
38
- method = "GET"
39
- url = "https://example.com"
40
- body = b""
41
- headers = {"User-Agent": "test-agent"}
42
-
43
- request_fingerprint1 = FingerprintGenerator.request_fingerprint(method, url, body, headers)
44
- request_fingerprint2 = FingerprintGenerator.request_fingerprint(method, url, body, headers)
45
-
46
- print(f"\nRequest fingerprint 1: {request_fingerprint1}")
47
- print(f"Request fingerprint 2: {request_fingerprint2}")
48
- print(f"请求指纹一致: {request_fingerprint1 == request_fingerprint2}")
49
-
50
-
51
- if __name__ == '__main__':
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ """
4
+ 简化版指纹一致性测试
5
+ ==============
6
+ 验证框架中各组件对相同数据生成一致的指纹
7
+ """
8
+
9
+ import sys
10
+ import os
11
+
12
+ # 添加项目根目录到Python路径
13
+ sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
14
+
15
+ from crawlo.utils.fingerprint import FingerprintGenerator
16
+
17
+
18
+ def test_fingerprint_consistency():
19
+ """测试指纹一致性"""
20
+ # 测试数据
21
+ test_data = {
22
+ "title": "Test Title",
23
+ "url": "https://example.com",
24
+ "content": "Test content",
25
+ "price": 99.99
26
+ }
27
+
28
+ # 使用指纹生成器生成指纹
29
+ fingerprint1 = FingerprintGenerator.data_fingerprint(test_data)
30
+ fingerprint2 = FingerprintGenerator.data_fingerprint(test_data)
31
+
32
+ # 验证相同数据生成相同指纹
33
+ print(f"First fingerprint: {fingerprint1}")
34
+ print(f"Second fingerprint: {fingerprint2}")
35
+ print(f"指纹一致: {fingerprint1 == fingerprint2}")
36
+
37
+ # 测试请求指纹
38
+ method = "GET"
39
+ url = "https://example.com"
40
+ body = b""
41
+ headers = {"User-Agent": "test-agent"}
42
+
43
+ request_fingerprint1 = FingerprintGenerator.request_fingerprint(method, url, body, headers)
44
+ request_fingerprint2 = FingerprintGenerator.request_fingerprint(method, url, body, headers)
45
+
46
+ print(f"\nRequest fingerprint 1: {request_fingerprint1}")
47
+ print(f"Request fingerprint 2: {request_fingerprint2}")
48
+ print(f"请求指纹一致: {request_fingerprint1 == request_fingerprint2}")
49
+
50
+
51
+ if __name__ == '__main__':
52
52
  test_fingerprint_consistency()
@@ -1,104 +1,104 @@
1
- #!/usr/bin/env python3
2
- # -*- coding: utf-8 -*-
3
- """
4
- 框架环境变量使用测试
5
- 验证整个框架中环境变量的正确使用
6
- """
7
- import sys
8
- import os
9
- import unittest
10
- from unittest.mock import patch, MagicMock
11
-
12
- # 添加项目根目录到Python路径
13
- sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
14
-
15
- from crawlo.utils.env_config import get_env_var, get_redis_config, get_runtime_config
16
- from crawlo.settings.setting_manager import SettingManager
17
- from crawlo.settings import default_settings
18
- from crawlo.mode_manager import from_env
19
-
20
-
21
- class TestFrameworkEnvUsage(unittest.TestCase):
22
- """框架环境变量使用测试"""
23
-
24
- def test_default_settings_env_usage(self):
25
- """测试 default_settings.py 中的环境变量使用"""
26
- # 验证 default_settings.py 不直接使用 os.getenv
27
- import inspect
28
- import crawlo.settings.default_settings as default_settings_module
29
-
30
- source_code = inspect.getsource(default_settings_module)
31
- # 检查是否还有直接使用 os.getenv 的地方
32
- self.assertNotIn('os.getenv', source_code,
33
- "default_settings.py 不应该直接使用 os.getenv")
34
-
35
- # 但应该使用 env_config 工具
36
- self.assertIn('get_redis_config', source_code,
37
- "default_settings.py 应该使用 get_redis_config")
38
- self.assertIn('get_runtime_config', source_code,
39
- "default_settings.py 应该使用 get_runtime_config")
40
-
41
- def test_env_config_tool(self):
42
- """测试环境变量配置工具"""
43
- # 测试获取Redis配置
44
- with patch.dict(os.environ, {
45
- 'REDIS_HOST': 'test.redis.com',
46
- 'REDIS_PORT': '6380',
47
- 'REDIS_PASSWORD': 'test_pass',
48
- 'REDIS_DB': '2'
49
- }):
50
- redis_config = get_redis_config()
51
- self.assertEqual(redis_config['REDIS_HOST'], 'test.redis.com')
52
- self.assertEqual(redis_config['REDIS_PORT'], 6380)
53
- self.assertEqual(redis_config['REDIS_PASSWORD'], 'test_pass')
54
- self.assertEqual(redis_config['REDIS_DB'], 2)
55
-
56
- # 测试获取运行时配置
57
- with patch.dict(os.environ, {
58
- 'PROJECT_NAME': 'test_project',
59
- 'CRAWLO_MODE': 'distributed',
60
- 'CONCURRENCY': '16'
61
- }):
62
- runtime_config = get_runtime_config()
63
- self.assertEqual(runtime_config['PROJECT_NAME'], 'test_project')
64
- self.assertEqual(runtime_config['CRAWLO_MODE'], 'distributed')
65
- self.assertEqual(runtime_config['CONCURRENCY'], 16)
66
-
67
- def test_settings_manager_with_env(self):
68
- """测试设置管理器与环境变量的集成"""
69
- # 设置环境变量
70
- env_vars = {
71
- 'PROJECT_NAME': 'env_test_project',
72
- 'CONCURRENCY': '12',
73
- 'REDIS_HOST': 'env.redis.test',
74
- 'REDIS_PORT': '6381'
75
- }
76
-
77
- with patch.dict(os.environ, env_vars):
78
- # 重新导入 default_settings 模块以获取最新的环境变量
79
- import importlib
80
- import crawlo.settings.default_settings
81
- importlib.reload(crawlo.settings.default_settings)
82
-
83
- # 创建设置管理器
84
- settings = SettingManager()
85
- settings.set_settings(crawlo.settings.default_settings)
86
-
87
- # 验证环境变量被正确使用
88
- redis_config = get_redis_config()
89
- self.assertEqual(settings.get('REDIS_HOST'), redis_config['REDIS_HOST'])
90
-
91
- runtime_config = get_runtime_config()
92
- self.assertEqual(settings.get('PROJECT_NAME'), runtime_config['PROJECT_NAME'])
93
-
94
- def test_mode_manager_env_usage(self):
95
- """测试 mode_manager.py 中的环境变量使用"""
96
- # 验证 from_env 函数现在会抛出异常
97
- with self.assertRaises(RuntimeError) as context:
98
- from_env()
99
-
100
- self.assertIn("环境变量配置已移除", str(context.exception))
101
-
102
-
103
- if __name__ == '__main__':
1
+ #!/usr/bin/env python3
2
+ # -*- coding: utf-8 -*-
3
+ """
4
+ 框架环境变量使用测试
5
+ 验证整个框架中环境变量的正确使用
6
+ """
7
+ import sys
8
+ import os
9
+ import unittest
10
+ from unittest.mock import patch, MagicMock
11
+
12
+ # 添加项目根目录到Python路径
13
+ sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
14
+
15
+ from crawlo.utils.env_config import get_env_var, get_redis_config, get_runtime_config
16
+ from crawlo.settings.setting_manager import SettingManager
17
+ from crawlo.settings import default_settings
18
+ from crawlo.mode_manager import from_env
19
+
20
+
21
+ class TestFrameworkEnvUsage(unittest.TestCase):
22
+ """框架环境变量使用测试"""
23
+
24
+ def test_default_settings_env_usage(self):
25
+ """测试 default_settings.py 中的环境变量使用"""
26
+ # 验证 default_settings.py 不直接使用 os.getenv
27
+ import inspect
28
+ import crawlo.settings.default_settings as default_settings_module
29
+
30
+ source_code = inspect.getsource(default_settings_module)
31
+ # 检查是否还有直接使用 os.getenv 的地方
32
+ self.assertNotIn('os.getenv', source_code,
33
+ "default_settings.py 不应该直接使用 os.getenv")
34
+
35
+ # 但应该使用 env_config 工具
36
+ self.assertIn('get_redis_config', source_code,
37
+ "default_settings.py 应该使用 get_redis_config")
38
+ self.assertIn('get_runtime_config', source_code,
39
+ "default_settings.py 应该使用 get_runtime_config")
40
+
41
+ def test_env_config_tool(self):
42
+ """测试环境变量配置工具"""
43
+ # 测试获取Redis配置
44
+ with patch.dict(os.environ, {
45
+ 'REDIS_HOST': 'test.redis.com',
46
+ 'REDIS_PORT': '6380',
47
+ 'REDIS_PASSWORD': 'test_pass',
48
+ 'REDIS_DB': '2'
49
+ }):
50
+ redis_config = get_redis_config()
51
+ self.assertEqual(redis_config['REDIS_HOST'], 'test.redis.com')
52
+ self.assertEqual(redis_config['REDIS_PORT'], 6380)
53
+ self.assertEqual(redis_config['REDIS_PASSWORD'], 'test_pass')
54
+ self.assertEqual(redis_config['REDIS_DB'], 2)
55
+
56
+ # 测试获取运行时配置
57
+ with patch.dict(os.environ, {
58
+ 'PROJECT_NAME': 'test_project',
59
+ 'CRAWLO_MODE': 'distributed',
60
+ 'CONCURRENCY': '16'
61
+ }):
62
+ runtime_config = get_runtime_config()
63
+ self.assertEqual(runtime_config['PROJECT_NAME'], 'test_project')
64
+ self.assertEqual(runtime_config['CRAWLO_MODE'], 'distributed')
65
+ self.assertEqual(runtime_config['CONCURRENCY'], 16)
66
+
67
+ def test_settings_manager_with_env(self):
68
+ """测试设置管理器与环境变量的集成"""
69
+ # 设置环境变量
70
+ env_vars = {
71
+ 'PROJECT_NAME': 'env_test_project',
72
+ 'CONCURRENCY': '12',
73
+ 'REDIS_HOST': 'env.redis.test',
74
+ 'REDIS_PORT': '6381'
75
+ }
76
+
77
+ with patch.dict(os.environ, env_vars):
78
+ # 重新导入 default_settings 模块以获取最新的环境变量
79
+ import importlib
80
+ import crawlo.settings.default_settings
81
+ importlib.reload(crawlo.settings.default_settings)
82
+
83
+ # 创建设置管理器
84
+ settings = SettingManager()
85
+ settings.set_settings(crawlo.settings.default_settings)
86
+
87
+ # 验证环境变量被正确使用
88
+ redis_config = get_redis_config()
89
+ self.assertEqual(settings.get('REDIS_HOST'), redis_config['REDIS_HOST'])
90
+
91
+ runtime_config = get_runtime_config()
92
+ self.assertEqual(settings.get('PROJECT_NAME'), runtime_config['PROJECT_NAME'])
93
+
94
+ def test_mode_manager_env_usage(self):
95
+ """测试 mode_manager.py 中的环境变量使用"""
96
+ # 验证 from_env 函数现在会抛出异常
97
+ with self.assertRaises(RuntimeError) as context:
98
+ from_env()
99
+
100
+ self.assertIn("环境变量配置已移除", str(context.exception))
101
+
102
+
103
+ if __name__ == '__main__':
104
104
  unittest.main()
@@ -1,67 +1,67 @@
1
- #!/usr/bin/env python
2
- # -*- coding: UTF-8 -*-
3
- """
4
- 测试框架日志系统
5
- """
6
- import sys
7
- import os
8
- sys.path.insert(0, '/')
9
-
10
- from crawlo.initialization import initialize_framework, get_framework_initializer
11
- from crawlo.utils.log import get_logger, LoggerManager
12
-
13
- def test_framework_logger():
14
- print("=== 测试框架日志系统 ===")
15
-
16
- # 1. 初始化框架,传入基本配置
17
- print("1. 初始化框架...")
18
- custom_settings = {
19
- 'LOG_LEVEL': 'INFO',
20
- 'LOG_FILE': 'test_framework.log',
21
- 'PROJECT_NAME': 'test_framework',
22
- 'RUN_MODE': 'standalone'
23
- }
24
- settings = initialize_framework(custom_settings)
25
- print(f" LOG_LEVEL: {settings.get('LOG_LEVEL')}")
26
- print(f" LOG_FILE: {settings.get('LOG_FILE')}")
27
-
28
- # 2. 获取框架初始化管理器
29
- init_manager = get_framework_initializer()
30
- print(f" 框架是否就绪: {init_manager.is_ready}")
31
- print(f" 初始化阶段: {init_manager.phase}")
32
-
33
- # 3. 测试框架logger
34
- framework_logger = init_manager.logger
35
- if framework_logger:
36
- print(f" 框架logger名称: {framework_logger.name}")
37
- print(f" 框架logger级别: {framework_logger.level}")
38
- print(f" 框架logger处理器数量: {len(framework_logger.handlers)}")
39
-
40
- for i, handler in enumerate(framework_logger.handlers):
41
- handler_type = type(handler).__name__
42
- print(f" 处理器{i}: {handler_type}, 级别: {handler.level}")
43
- else:
44
- print(" 框架logger为None!")
45
- framework_logger = get_logger('crawlo.framework')
46
- print(f" 手动创建的框架logger: {framework_logger.name}")
47
-
48
- # 4. 测试日志输出
49
- print("2. 测试日志输出...")
50
- framework_logger.info("Crawlo框架初始化完成")
51
- framework_logger.info("Crawlo Framework Started 1.3.3")
52
- framework_logger.info("使用单机模式 - 简单快速,适合开发和中小规模爬取")
53
- framework_logger.info("Run Mode: standalone")
54
- framework_logger.info("Starting running test_spider")
55
-
56
- # 5. 测试其他logger
57
- print("3. 测试其他组件logger...")
58
- queue_logger = get_logger('QueueManager')
59
- queue_logger.info("Queue initialized successfully Type: memory")
60
-
61
- scheduler_logger = get_logger('Scheduler')
62
- scheduler_logger.info("enabled filters: crawlo.filters.memory_filter.MemoryFilter")
63
-
64
- print("=== 测试完成 ===")
65
-
66
- if __name__ == "__main__":
1
+ #!/usr/bin/env python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ 测试框架日志系统
5
+ """
6
+ import sys
7
+ import os
8
+ sys.path.insert(0, '/')
9
+
10
+ from crawlo.initialization import initialize_framework, get_framework_initializer
11
+ from crawlo.utils.log import get_logger, LoggerManager
12
+
13
+ def test_framework_logger():
14
+ print("=== 测试框架日志系统 ===")
15
+
16
+ # 1. 初始化框架,传入基本配置
17
+ print("1. 初始化框架...")
18
+ custom_settings = {
19
+ 'LOG_LEVEL': 'INFO',
20
+ 'LOG_FILE': 'test_framework.log',
21
+ 'PROJECT_NAME': 'test_framework',
22
+ 'RUN_MODE': 'standalone'
23
+ }
24
+ settings = initialize_framework(custom_settings)
25
+ print(f" LOG_LEVEL: {settings.get('LOG_LEVEL')}")
26
+ print(f" LOG_FILE: {settings.get('LOG_FILE')}")
27
+
28
+ # 2. 获取框架初始化管理器
29
+ init_manager = get_framework_initializer()
30
+ print(f" 框架是否就绪: {init_manager.is_ready}")
31
+ print(f" 初始化阶段: {init_manager.phase}")
32
+
33
+ # 3. 测试框架logger
34
+ framework_logger = init_manager.logger
35
+ if framework_logger:
36
+ print(f" 框架logger名称: {framework_logger.name}")
37
+ print(f" 框架logger级别: {framework_logger.level}")
38
+ print(f" 框架logger处理器数量: {len(framework_logger.handlers)}")
39
+
40
+ for i, handler in enumerate(framework_logger.handlers):
41
+ handler_type = type(handler).__name__
42
+ print(f" 处理器{i}: {handler_type}, 级别: {handler.level}")
43
+ else:
44
+ print(" 框架logger为None!")
45
+ framework_logger = get_logger('crawlo.framework')
46
+ print(f" 手动创建的框架logger: {framework_logger.name}")
47
+
48
+ # 4. 测试日志输出
49
+ print("2. 测试日志输出...")
50
+ framework_logger.info("Crawlo框架初始化完成")
51
+ framework_logger.info("Crawlo Framework Started 1.3.3")
52
+ framework_logger.info("使用单机模式 - 简单快速,适合开发和中小规模爬取")
53
+ framework_logger.info("Run Mode: standalone")
54
+ framework_logger.info("Starting running test_spider")
55
+
56
+ # 5. 测试其他logger
57
+ print("3. 测试其他组件logger...")
58
+ queue_logger = get_logger('QueueManager')
59
+ queue_logger.info("Queue initialized successfully Type: memory")
60
+
61
+ scheduler_logger = get_logger('Scheduler')
62
+ scheduler_logger.info("enabled filters: crawlo.filters.memory_filter.MemoryFilter")
63
+
64
+ print("=== 测试完成 ===")
65
+
66
+ if __name__ == "__main__":
67
67
  test_framework_logger()
@@ -1,65 +1,65 @@
1
- #!/usr/bin/env python
2
- # -*- coding: utf-8 -*-
3
- """
4
- 测试框架启动信息是否正确输出到日志文件
5
- """
6
- import sys
7
- import os
8
- sys.path.insert(0, '/')
9
-
10
- def test_framework_startup():
11
- print("=== 测试框架启动信息输出 ===")
12
-
13
- # 删除旧的日志文件
14
- test_log_file = '/Users/oscar/projects/Crawlo/test_startup.log'
15
- if os.path.exists(test_log_file):
16
- os.remove(test_log_file)
17
-
18
- # 准备测试设置
19
- test_settings = {
20
- 'PROJECT_NAME': 'test_startup',
21
- 'LOG_LEVEL': 'INFO',
22
- 'LOG_FILE': test_log_file,
23
- 'RUN_MODE': 'standalone'
24
- }
25
-
26
- # 初始化框架
27
- from crawlo.initialization import initialize_framework
28
- settings = initialize_framework(test_settings)
29
-
30
- print(f"设置初始化完成: {settings.get('PROJECT_NAME')}")
31
-
32
- # 检查日志文件是否包含框架启动信息
33
- if os.path.exists(test_log_file):
34
- with open(test_log_file, 'r', encoding='utf-8') as f:
35
- content = f.read()
36
- print(f"日志文件内容长度: {len(content)} 字符")
37
-
38
- # 检查关键的框架启动信息
39
- if "Crawlo框架初始化完成" in content:
40
- print("✅ 发现: Crawlo框架初始化完成")
41
- else:
42
- print("❌ 未找到: Crawlo框架初始化完成")
43
-
44
- if "Crawlo Framework Started" in content:
45
- print("✅ 发现: Crawlo Framework Started")
46
- else:
47
- print("❌ 未找到: Crawlo Framework Started")
48
-
49
- if "使用单机模式" in content:
50
- print("✅ 发现: 使用单机模式")
51
- else:
52
- print("❌ 未找到: 使用单机模式")
53
-
54
- print("\n前50行日志内容:")
55
- lines = content.split('\n')[:50]
56
- for i, line in enumerate(lines, 1):
57
- if any(keyword in line for keyword in ["框架", "Framework", "Started"]):
58
- print(f"{i:3d}: {line}")
59
- else:
60
- print("❌ 日志文件未创建")
61
-
62
- print("=== 测试完成 ===")
63
-
64
- if __name__ == "__main__":
1
+ #!/usr/bin/env python
2
+ # -*- coding: utf-8 -*-
3
+ """
4
+ 测试框架启动信息是否正确输出到日志文件
5
+ """
6
+ import sys
7
+ import os
8
+ sys.path.insert(0, '/')
9
+
10
+ def test_framework_startup():
11
+ print("=== 测试框架启动信息输出 ===")
12
+
13
+ # 删除旧的日志文件
14
+ test_log_file = '/Users/oscar/projects/Crawlo/test_startup.log'
15
+ if os.path.exists(test_log_file):
16
+ os.remove(test_log_file)
17
+
18
+ # 准备测试设置
19
+ test_settings = {
20
+ 'PROJECT_NAME': 'test_startup',
21
+ 'LOG_LEVEL': 'INFO',
22
+ 'LOG_FILE': test_log_file,
23
+ 'RUN_MODE': 'standalone'
24
+ }
25
+
26
+ # 初始化框架
27
+ from crawlo.initialization import initialize_framework
28
+ settings = initialize_framework(test_settings)
29
+
30
+ print(f"设置初始化完成: {settings.get('PROJECT_NAME')}")
31
+
32
+ # 检查日志文件是否包含框架启动信息
33
+ if os.path.exists(test_log_file):
34
+ with open(test_log_file, 'r', encoding='utf-8') as f:
35
+ content = f.read()
36
+ print(f"日志文件内容长度: {len(content)} 字符")
37
+
38
+ # 检查关键的框架启动信息
39
+ if "Crawlo框架初始化完成" in content:
40
+ print("✅ 发现: Crawlo框架初始化完成")
41
+ else:
42
+ print("❌ 未找到: Crawlo框架初始化完成")
43
+
44
+ if "Crawlo Framework Started" in content:
45
+ print("✅ 发现: Crawlo Framework Started")
46
+ else:
47
+ print("❌ 未找到: Crawlo Framework Started")
48
+
49
+ if "使用单机模式" in content:
50
+ print("✅ 发现: 使用单机模式")
51
+ else:
52
+ print("❌ 未找到: 使用单机模式")
53
+
54
+ print("\n前50行日志内容:")
55
+ lines = content.split('\n')[:50]
56
+ for i, line in enumerate(lines, 1):
57
+ if any(keyword in line for keyword in ["框架", "Framework", "Started"]):
58
+ print(f"{i:3d}: {line}")
59
+ else:
60
+ print("❌ 日志文件未创建")
61
+
62
+ print("=== 测试完成 ===")
63
+
64
+ if __name__ == "__main__":
65
65
  test_framework_startup()