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
crawlo/stats_collector.py CHANGED
@@ -1,59 +1,59 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- """
4
- # @Time : 2025-05-17 09:57
5
- # @Author : crawl-coder
6
- # @Desc : 统计信息收集器
7
- """
8
- from pprint import pformat
9
- from crawlo.utils.log import get_logger
10
-
11
-
12
- class StatsCollector(object):
13
-
14
- def __init__(self, crawler):
15
- self.crawler = crawler
16
- self._dump = self.crawler.settings.get_bool('STATS_DUMP')
17
- self._stats = {}
18
- self.logger = get_logger(self.__class__.__name__, "INFO")
19
-
20
- def inc_value(self, key, count=1, start=0):
21
- self._stats[key] = self._stats.setdefault(key, start) + count
22
-
23
- def get_value(self, key, default=None):
24
- return self._stats.get(key, default)
25
-
26
- def get_stats(self):
27
- return self._stats
28
-
29
- def set_stats(self, stats):
30
- self._stats = stats
31
-
32
- def clear_stats(self):
33
- self._stats.clear()
34
-
35
- def close_spider(self, spider, reason):
36
- self._stats['reason'] = reason
37
-
38
- # 首选:使用 spider.name
39
- # 次选:使用实例的类名
40
- # 最后:使用一个完全未知的占位符
41
- spider_name = (
42
- getattr(spider, 'name', None) or
43
- spider.__class__.__name__ or
44
- '<Unknown>'
45
- )
46
-
47
- self._stats['spider_name'] = spider_name
48
-
49
- if self._dump:
50
- self.logger.info(f'{spider_name} stats: \n{pformat(self._stats)}')
51
-
52
- def __getitem__(self, item):
53
- return self._stats[item]
54
-
55
- def __setitem__(self, key, value):
56
- self._stats[key] = value
57
-
58
- def __delitem__(self, key):
59
- del self._stats[key]
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ """
4
+ # @Time : 2025-05-17 09:57
5
+ # @Author : crawl-coder
6
+ # @Desc : 统计信息收集器
7
+ """
8
+ from pprint import pformat
9
+ from crawlo.utils.log import get_logger
10
+
11
+
12
+ class StatsCollector(object):
13
+
14
+ def __init__(self, crawler):
15
+ self.crawler = crawler
16
+ self._dump = self.crawler.settings.get_bool('STATS_DUMP')
17
+ self._stats = {}
18
+ self.logger = get_logger(self.__class__.__name__, "INFO")
19
+
20
+ def inc_value(self, key, count=1, start=0):
21
+ self._stats[key] = self._stats.setdefault(key, start) + count
22
+
23
+ def get_value(self, key, default=None):
24
+ return self._stats.get(key, default)
25
+
26
+ def get_stats(self):
27
+ return self._stats
28
+
29
+ def set_stats(self, stats):
30
+ self._stats = stats
31
+
32
+ def clear_stats(self):
33
+ self._stats.clear()
34
+
35
+ def close_spider(self, spider, reason):
36
+ self._stats['reason'] = reason
37
+
38
+ # 首选:使用 spider.name
39
+ # 次选:使用实例的类名
40
+ # 最后:使用一个完全未知的占位符
41
+ spider_name = (
42
+ getattr(spider, 'name', None) or
43
+ spider.__class__.__name__ or
44
+ '<Unknown>'
45
+ )
46
+
47
+ self._stats['spider_name'] = spider_name
48
+
49
+ if self._dump:
50
+ self.logger.info(f'{spider_name} stats: \n{pformat(self._stats)}')
51
+
52
+ def __getitem__(self, item):
53
+ return self._stats[item]
54
+
55
+ def __setitem__(self, key, value):
56
+ self._stats[key] = value
57
+
58
+ def __delitem__(self, key):
59
+ del self._stats[key]
crawlo/subscriber.py CHANGED
@@ -1,131 +1,131 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- import asyncio
4
- import weakref
5
- from collections import defaultdict
6
- from inspect import iscoroutinefunction
7
- from typing import Dict, Set, Callable, Coroutine, Any, TypeAlias, List, Tuple
8
-
9
-
10
- class ReceiverTypeError(TypeError):
11
- """当订阅的接收者不是一个协程函数时抛出。"""
12
- pass
13
-
14
-
15
- ReceiverCoroutine: TypeAlias = Callable[..., Coroutine[Any, Any, Any]]
16
-
17
-
18
- class Subscriber:
19
- """
20
- 一个支持异步协程的发布/订阅(Pub/Sub)模式实现。
21
-
22
- 这个类允许你注册(订阅)协程函数来监听特定事件,并在事件发生时
23
- 以并发的方式异步地通知所有订阅者。
24
- """
25
-
26
- def __init__(self):
27
- """初始化一个空的订阅者字典。"""
28
- # 使用弱引用字典避免内存泄漏
29
- self._subscribers: Dict[str, Dict[ReceiverCoroutine, int]] = defaultdict(dict)
30
- # 用于缓存排序后的订阅者列表,提高频繁事件的处理性能
31
- self._sorted_subscribers_cache: Dict[str, List[Tuple[ReceiverCoroutine, int]]] = {}
32
-
33
- def subscribe(self, receiver: ReceiverCoroutine, *, event: str, priority: int = 0) -> None:
34
- """
35
- 订阅一个事件。
36
-
37
- Args:
38
- receiver: 一个协程函数 (例如 async def my_func(...))。
39
- event: 要订阅的事件名称。
40
- priority: 订阅者优先级,数值越小优先级越高,默认为0。
41
-
42
- Raises:
43
- ReceiverTypeError: 如果提供的 `receiver` 不是一个协程函数。
44
- """
45
- if not iscoroutinefunction(receiver):
46
- raise ReceiverTypeError(f"接收者 '{receiver.__qualname__}' 必须是一个协程函数。")
47
-
48
- # 使用弱引用避免内存泄漏
49
- self._subscribers[event][receiver] = priority
50
- # 清除缓存
51
- self._sorted_subscribers_cache.pop(event, None)
52
-
53
- def unsubscribe(self, receiver: ReceiverCoroutine, *, event: str) -> None:
54
- """
55
- 取消订阅一个事件。
56
-
57
- 如果事件或接收者不存在,将静默处理。
58
-
59
- Args:
60
- receiver: 要取消订阅的协程函数。
61
- event: 事件名称。
62
- """
63
- if event in self._subscribers:
64
- self._subscribers[event].pop(receiver, None)
65
- # 清除缓存
66
- self._sorted_subscribers_cache.pop(event, None)
67
-
68
- def _get_sorted_subscribers(self, event: str) -> List[Tuple[ReceiverCoroutine, int]]:
69
- """
70
- 获取按优先级排序的订阅者列表。
71
-
72
- Args:
73
- event: 事件名称。
74
-
75
- Returns:
76
- 按优先级排序的订阅者列表。
77
- """
78
- # 检查缓存
79
- if event in self._sorted_subscribers_cache:
80
- return self._sorted_subscribers_cache[event]
81
-
82
- # 获取有效的订阅者(使用弱引用检查)
83
- valid_subscribers = {}
84
- for receiver, priority in list(self._subscribers[event].items()):
85
- # 检查弱引用是否仍然有效
86
- if isinstance(receiver, Callable):
87
- valid_subscribers[receiver] = priority
88
-
89
- # 更新订阅者字典
90
- self._subscribers[event] = valid_subscribers
91
-
92
- # 按优先级排序(数值小的优先级高)
93
- sorted_subscribers = sorted(valid_subscribers.items(), key=lambda x: x[1])
94
- # 缓存结果
95
- self._sorted_subscribers_cache[event] = sorted_subscribers
96
-
97
- return sorted_subscribers
98
-
99
- async def notify(self, event: str, *args, **kwargs) -> List[Any]:
100
- """
101
- 异步地、并发地通知所有订阅了该事件的接收者。
102
-
103
- 此方法会等待所有订阅者任务完成后再返回,并收集所有结果或异常。
104
- 订阅者按优先级顺序执行,优先级高的先执行。
105
-
106
- Args:
107
- event: 要触发的事件名称。
108
- *args: 传递给接收者的位置参数。
109
- **kwargs: 传递给接收者的关键字参数。
110
-
111
- Returns:
112
- 一个列表,包含每个订阅者任务的返回结果或在执行期间捕获的异常。
113
- """
114
- sorted_subscribers = self._get_sorted_subscribers(event)
115
- if not sorted_subscribers:
116
- return []
117
-
118
- # 为频繁触发的事件重用任务对象以提高性能
119
- tasks = []
120
- for receiver, _ in sorted_subscribers:
121
- try:
122
- # 创建任务并添加到列表
123
- task = asyncio.create_task(receiver(*args, **kwargs))
124
- tasks.append(task)
125
- except Exception as e:
126
- # 如果创建任务失败,记录异常并继续处理其他订阅者
127
- tasks.append(asyncio.Future()) # 添加一个已完成的Future表示错误
128
- tasks[-1].set_exception(e)
129
-
130
- # 并发执行所有任务并返回结果列表(包括异常)
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ import asyncio
4
+ import weakref
5
+ from collections import defaultdict
6
+ from inspect import iscoroutinefunction
7
+ from typing import Dict, Set, Callable, Coroutine, Any, TypeAlias, List, Tuple
8
+
9
+
10
+ class ReceiverTypeError(TypeError):
11
+ """当订阅的接收者不是一个协程函数时抛出。"""
12
+ pass
13
+
14
+
15
+ ReceiverCoroutine: TypeAlias = Callable[..., Coroutine[Any, Any, Any]]
16
+
17
+
18
+ class Subscriber:
19
+ """
20
+ 一个支持异步协程的发布/订阅(Pub/Sub)模式实现。
21
+
22
+ 这个类允许你注册(订阅)协程函数来监听特定事件,并在事件发生时
23
+ 以并发的方式异步地通知所有订阅者。
24
+ """
25
+
26
+ def __init__(self):
27
+ """初始化一个空的订阅者字典。"""
28
+ # 使用弱引用字典避免内存泄漏
29
+ self._subscribers: Dict[str, Dict[ReceiverCoroutine, int]] = defaultdict(dict)
30
+ # 用于缓存排序后的订阅者列表,提高频繁事件的处理性能
31
+ self._sorted_subscribers_cache: Dict[str, List[Tuple[ReceiverCoroutine, int]]] = {}
32
+
33
+ def subscribe(self, receiver: ReceiverCoroutine, *, event: str, priority: int = 0) -> None:
34
+ """
35
+ 订阅一个事件。
36
+
37
+ Args:
38
+ receiver: 一个协程函数 (例如 async def my_func(...))。
39
+ event: 要订阅的事件名称。
40
+ priority: 订阅者优先级,数值越小优先级越高,默认为0。
41
+
42
+ Raises:
43
+ ReceiverTypeError: 如果提供的 `receiver` 不是一个协程函数。
44
+ """
45
+ if not iscoroutinefunction(receiver):
46
+ raise ReceiverTypeError(f"接收者 '{receiver.__qualname__}' 必须是一个协程函数。")
47
+
48
+ # 使用弱引用避免内存泄漏
49
+ self._subscribers[event][receiver] = priority
50
+ # 清除缓存
51
+ self._sorted_subscribers_cache.pop(event, None)
52
+
53
+ def unsubscribe(self, receiver: ReceiverCoroutine, *, event: str) -> None:
54
+ """
55
+ 取消订阅一个事件。
56
+
57
+ 如果事件或接收者不存在,将静默处理。
58
+
59
+ Args:
60
+ receiver: 要取消订阅的协程函数。
61
+ event: 事件名称。
62
+ """
63
+ if event in self._subscribers:
64
+ self._subscribers[event].pop(receiver, None)
65
+ # 清除缓存
66
+ self._sorted_subscribers_cache.pop(event, None)
67
+
68
+ def _get_sorted_subscribers(self, event: str) -> List[Tuple[ReceiverCoroutine, int]]:
69
+ """
70
+ 获取按优先级排序的订阅者列表。
71
+
72
+ Args:
73
+ event: 事件名称。
74
+
75
+ Returns:
76
+ 按优先级排序的订阅者列表。
77
+ """
78
+ # 检查缓存
79
+ if event in self._sorted_subscribers_cache:
80
+ return self._sorted_subscribers_cache[event]
81
+
82
+ # 获取有效的订阅者(使用弱引用检查)
83
+ valid_subscribers = {}
84
+ for receiver, priority in list(self._subscribers[event].items()):
85
+ # 检查弱引用是否仍然有效
86
+ if isinstance(receiver, Callable):
87
+ valid_subscribers[receiver] = priority
88
+
89
+ # 更新订阅者字典
90
+ self._subscribers[event] = valid_subscribers
91
+
92
+ # 按优先级排序(数值小的优先级高)
93
+ sorted_subscribers = sorted(valid_subscribers.items(), key=lambda x: x[1])
94
+ # 缓存结果
95
+ self._sorted_subscribers_cache[event] = sorted_subscribers
96
+
97
+ return sorted_subscribers
98
+
99
+ async def notify(self, event: str, *args, **kwargs) -> List[Any]:
100
+ """
101
+ 异步地、并发地通知所有订阅了该事件的接收者。
102
+
103
+ 此方法会等待所有订阅者任务完成后再返回,并收集所有结果或异常。
104
+ 订阅者按优先级顺序执行,优先级高的先执行。
105
+
106
+ Args:
107
+ event: 要触发的事件名称。
108
+ *args: 传递给接收者的位置参数。
109
+ **kwargs: 传递给接收者的关键字参数。
110
+
111
+ Returns:
112
+ 一个列表,包含每个订阅者任务的返回结果或在执行期间捕获的异常。
113
+ """
114
+ sorted_subscribers = self._get_sorted_subscribers(event)
115
+ if not sorted_subscribers:
116
+ return []
117
+
118
+ # 为频繁触发的事件重用任务对象以提高性能
119
+ tasks = []
120
+ for receiver, _ in sorted_subscribers:
121
+ try:
122
+ # 创建任务并添加到列表
123
+ task = asyncio.create_task(receiver(*args, **kwargs))
124
+ tasks.append(task)
125
+ except Exception as e:
126
+ # 如果创建任务失败,记录异常并继续处理其他订阅者
127
+ tasks.append(asyncio.Future()) # 添加一个已完成的Future表示错误
128
+ tasks[-1].set_exception(e)
129
+
130
+ # 并发执行所有任务并返回结果列表(包括异常)
131
131
  return await asyncio.gather(*tasks, return_exceptions=True)
crawlo/task_manager.py CHANGED
@@ -1,30 +1,30 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- import asyncio
4
- from asyncio import Task, Future, Semaphore
5
- from typing import Set, Final
6
-
7
-
8
- class TaskManager:
9
-
10
- def __init__(self, total_concurrency: int = 8):
11
- self.current_task: Final[Set] = set()
12
- self.semaphore: Semaphore = Semaphore(total_concurrency)
13
-
14
- async def create_task(self, coroutine) -> Task:
15
- # 等待信号量,控制并发数
16
- await self.semaphore.acquire()
17
-
18
- task = asyncio.create_task(coroutine)
19
- self.current_task.add(task)
20
-
21
- def done_callback(_future: Future) -> None:
22
- self.current_task.remove(task)
23
- self.semaphore.release()
24
-
25
- task.add_done_callback(done_callback)
26
-
27
- return task
28
-
29
- def all_done(self) -> bool:
30
- return len(self.current_task) == 0
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ import asyncio
4
+ from asyncio import Task, Future, Semaphore
5
+ from typing import Set, Final
6
+
7
+
8
+ class TaskManager:
9
+
10
+ def __init__(self, total_concurrency: int = 8):
11
+ self.current_task: Final[Set] = set()
12
+ self.semaphore: Semaphore = Semaphore(total_concurrency)
13
+
14
+ async def create_task(self, coroutine) -> Task:
15
+ # 等待信号量,控制并发数
16
+ await self.semaphore.acquire()
17
+
18
+ task = asyncio.create_task(coroutine)
19
+ self.current_task.add(task)
20
+
21
+ def done_callback(_future: Future) -> None:
22
+ self.current_task.remove(task)
23
+ self.semaphore.release()
24
+
25
+ task.add_done_callback(done_callback)
26
+
27
+ return task
28
+
29
+ def all_done(self) -> bool:
30
+ return len(self.current_task) == 0
@@ -1,11 +1,11 @@
1
- # crawlo.cfg
2
- # 项目的配置文件。
3
-
4
- [settings]
5
- # 指定 settings 模块的导入路径
6
- default = {{project_name}}.settings
7
-
8
- [deploy]
9
- # (可选)用于部署配置
10
- # url = http://localhost:6800/
1
+ # crawlo.cfg
2
+ # 项目的配置文件。
3
+
4
+ [settings]
5
+ # 指定 settings 模块的导入路径
6
+ default = {{project_name}}.settings
7
+
8
+ [deploy]
9
+ # (可选)用于部署配置
10
+ # url = http://localhost:6800/
11
11
  # project = {{project_name}}
@@ -1,4 +1,4 @@
1
- # -*- coding: UTF-8 -*-
2
- """
3
- {{project_name}} 项目包
1
+ # -*- coding: UTF-8 -*-
2
+ """
3
+ {{project_name}} 项目包
4
4
  """
@@ -1,18 +1,18 @@
1
- # -*- coding: UTF-8 -*-
2
- """
3
- {{project_name}}.items
4
- ======================
5
- 定义你抓取的数据结构。
6
- """
7
-
8
- from crawlo.items import Item, Field
9
-
10
-
11
- class ExampleItem(Item):
12
- """
13
- 一个示例数据项。
14
- """
15
- id = Field()
16
- # price = Field()
17
- # description = Field()
1
+ # -*- coding: UTF-8 -*-
2
+ """
3
+ {{project_name}}.items
4
+ ======================
5
+ 定义你抓取的数据结构。
6
+ """
7
+
8
+ from crawlo.items import Item, Field
9
+
10
+
11
+ class ExampleItem(Item):
12
+ """
13
+ 一个示例数据项。
14
+ """
15
+ id = Field()
16
+ # price = Field()
17
+ # description = Field()
18
18
  pass