crawlo 1.1.4__py3-none-any.whl → 1.1.6__py3-none-any.whl

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

Potentially problematic release.


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

Files changed (190) 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 +292 -285
  13. crawlo/commands/startproject.py +419 -196
  14. crawlo/commands/stats.py +188 -188
  15. crawlo/commands/utils.py +186 -186
  16. crawlo/config.py +312 -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 +281 -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 +212 -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 +61 -56
  70. crawlo/pipelines/redis_dedup_pipeline.py +167 -162
  71. crawlo/project.py +188 -153
  72. crawlo/queue/pqueue.py +37 -37
  73. crawlo/queue/queue_manager.py +334 -307
  74. crawlo/queue/redis_priority_queue.py +299 -209
  75. crawlo/settings/__init__.py +7 -7
  76. crawlo/settings/default_settings.py +219 -278
  77. crawlo/settings/setting_manager.py +123 -100
  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.6.dist-info}/METADATA +401 -403
  126. crawlo-1.1.6.dist-info/RECORD +189 -0
  127. examples/__init__.py +7 -7
  128. tests/DOUBLE_CRAWLO_PREFIX_FIX_REPORT.md +82 -0
  129. tests/__init__.py +7 -7
  130. tests/advanced_tools_example.py +276 -0
  131. tests/authenticated_proxy_example.py +237 -0
  132. tests/cleaners_example.py +161 -0
  133. tests/config_validation_demo.py +103 -0
  134. {examples → tests}/controlled_spider_example.py +205 -205
  135. tests/date_tools_example.py +181 -0
  136. tests/dynamic_loading_example.py +524 -0
  137. tests/dynamic_loading_test.py +105 -0
  138. tests/env_config_example.py +134 -0
  139. tests/error_handling_example.py +172 -0
  140. tests/redis_key_validation_demo.py +131 -0
  141. tests/response_improvements_example.py +145 -0
  142. tests/test_advanced_tools.py +149 -0
  143. tests/test_all_redis_key_configs.py +146 -0
  144. tests/test_authenticated_proxy.py +142 -0
  145. tests/test_cleaners.py +55 -0
  146. tests/test_comprehensive.py +147 -0
  147. tests/test_config_validator.py +194 -0
  148. tests/test_date_tools.py +124 -0
  149. tests/test_double_crawlo_fix.py +208 -0
  150. tests/test_double_crawlo_fix_simple.py +125 -0
  151. tests/test_dynamic_downloaders_proxy.py +125 -0
  152. tests/test_dynamic_proxy.py +93 -0
  153. tests/test_dynamic_proxy_config.py +147 -0
  154. tests/test_dynamic_proxy_real.py +110 -0
  155. tests/test_edge_cases.py +304 -0
  156. tests/test_enhanced_error_handler.py +271 -0
  157. tests/test_env_config.py +122 -0
  158. tests/test_error_handler_compatibility.py +113 -0
  159. tests/test_final_validation.py +153 -153
  160. tests/test_framework_env_usage.py +104 -0
  161. tests/test_integration.py +357 -0
  162. tests/test_item_dedup_redis_key.py +123 -0
  163. tests/test_parsel.py +30 -0
  164. tests/test_performance.py +328 -0
  165. tests/test_proxy_health_check.py +32 -32
  166. tests/test_proxy_middleware_integration.py +136 -136
  167. tests/test_proxy_providers.py +56 -56
  168. tests/test_proxy_stats.py +19 -19
  169. tests/test_proxy_strategies.py +59 -59
  170. tests/test_queue_manager_double_crawlo.py +231 -0
  171. tests/test_queue_manager_redis_key.py +177 -0
  172. tests/test_redis_config.py +28 -28
  173. tests/test_redis_connection_pool.py +295 -0
  174. tests/test_redis_key_naming.py +182 -0
  175. tests/test_redis_key_validator.py +124 -0
  176. tests/test_redis_queue.py +224 -224
  177. tests/test_request_serialization.py +70 -70
  178. tests/test_response_improvements.py +153 -0
  179. tests/test_scheduler.py +241 -241
  180. tests/test_simple_response.py +62 -0
  181. tests/test_telecom_spider_redis_key.py +206 -0
  182. tests/test_template_content.py +88 -0
  183. tests/test_template_redis_key.py +135 -0
  184. tests/test_tools.py +154 -0
  185. tests/tools_example.py +258 -0
  186. crawlo/core/enhanced_engine.py +0 -190
  187. crawlo-1.1.4.dist-info/RECORD +0 -117
  188. {crawlo-1.1.4.dist-info → crawlo-1.1.6.dist-info}/WHEEL +0 -0
  189. {crawlo-1.1.4.dist-info → crawlo-1.1.6.dist-info}/entry_points.txt +0 -0
  190. {crawlo-1.1.4.dist-info → crawlo-1.1.6.dist-info}/top_level.txt +0 -0
crawlo/items/fields.py CHANGED
@@ -1,54 +1,54 @@
1
- #!/usr/bin/python
2
- # -*- coding: UTF-8 -*-
3
- """
4
- Field 类定义
5
- """
6
-
7
- from typing import Any, Optional, Type
8
-
9
-
10
- class Field:
11
- """
12
- 字段定义类,用于定义 Item 的字段属性和验证规则
13
- """
14
- def __init__(
15
- self,
16
- nullable: bool = True,
17
- *,
18
- default: Any = None,
19
- field_type: Optional[Type] = None,
20
- max_length: Optional[int] = None,
21
- description: str = ""
22
- ):
23
- self.nullable = nullable
24
- self.default = default
25
- self.field_type = field_type
26
- self.max_length = max_length
27
- self.description = description
28
-
29
- def validate(self, value: Any, field_name: str = "") -> Any:
30
- """
31
- 验证字段值是否符合规则
32
- """
33
- if value is None or (isinstance(value, str) and value.strip() == ""):
34
- if self.default is not None:
35
- return self.default
36
- elif not self.nullable:
37
- raise ValueError(
38
- f"字段 '{field_name}' 不允许为空。"
39
- )
40
-
41
- if value is not None and not (isinstance(value, str) and value.strip() == ""):
42
- if self.field_type and not isinstance(value, self.field_type):
43
- raise TypeError(
44
- f"字段 '{field_name}' 类型错误:期望类型 {self.field_type}, 得到 {type(value)},值:{value!r}"
45
- )
46
- if self.max_length and len(str(value)) > self.max_length:
47
- raise ValueError(
48
- f"字段 '{field_name}' 长度超限:最大长度 {self.max_length},当前长度 {len(str(value))},值:{value!r}"
49
- )
50
-
51
- return value
52
-
53
- def __repr__(self):
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ Field 类定义
5
+ """
6
+
7
+ from typing import Any, Optional, Type
8
+
9
+
10
+ class Field:
11
+ """
12
+ 字段定义类,用于定义 Item 的字段属性和验证规则
13
+ """
14
+ def __init__(
15
+ self,
16
+ nullable: bool = True,
17
+ *,
18
+ default: Any = None,
19
+ field_type: Optional[Type] = None,
20
+ max_length: Optional[int] = None,
21
+ description: str = ""
22
+ ):
23
+ self.nullable = nullable
24
+ self.default = default
25
+ self.field_type = field_type
26
+ self.max_length = max_length
27
+ self.description = description
28
+
29
+ def validate(self, value: Any, field_name: str = "") -> Any:
30
+ """
31
+ 验证字段值是否符合规则
32
+ """
33
+ if value is None or (isinstance(value, str) and value.strip() == ""):
34
+ if self.default is not None:
35
+ return self.default
36
+ elif not self.nullable:
37
+ raise ValueError(
38
+ f"字段 '{field_name}' 不允许为空。"
39
+ )
40
+
41
+ if value is not None and not (isinstance(value, str) and value.strip() == ""):
42
+ if self.field_type and not isinstance(value, self.field_type):
43
+ raise TypeError(
44
+ f"字段 '{field_name}' 类型错误:期望类型 {self.field_type}, 得到 {type(value)},值:{value!r}"
45
+ )
46
+ if self.max_length and len(str(value)) > self.max_length:
47
+ raise ValueError(
48
+ f"字段 '{field_name}' 长度超限:最大长度 {self.max_length},当前长度 {len(str(value))},值:{value!r}"
49
+ )
50
+
51
+ return value
52
+
53
+ def __repr__(self):
54
54
  return f"<Field nullable={self.nullable} type={self.field_type} default={self.default}>"
crawlo/items/items.py CHANGED
@@ -1,105 +1,105 @@
1
- #!/usr/bin/python
2
- # -*- coding: UTF-8 -*-
3
- """
4
- Item 类定义
5
- """
6
- from copy import deepcopy
7
- from pprint import pformat
8
- from typing import Any, Iterator, Dict
9
- from collections.abc import MutableMapping
10
-
11
- from .base import ItemMeta
12
- from crawlo.exceptions import ItemInitError, ItemAttributeError
13
-
14
-
15
- class Item(MutableMapping, metaclass=ItemMeta):
16
- """
17
- 数据项基类,用于定义结构化数据
18
- """
19
- FIELDS: Dict[str, Any] = {}
20
-
21
- def __init__(self, *args, **kwargs):
22
- if args:
23
- raise ItemInitError(f"{self.__class__.__name__} 不支持位置参数:{args},请使用关键字参数初始化。")
24
-
25
- self._values: Dict[str, Any] = {}
26
-
27
- # 初始化字段,默认值填充
28
- for field_name, field_obj in self.FIELDS.items():
29
- if field_obj.default is not None:
30
- self._values[field_name] = field_obj.default
31
-
32
- # 覆盖默认值或设置新值
33
- for key, value in kwargs.items():
34
- self[key] = value
35
-
36
- def __getitem__(self, item: str) -> Any:
37
- return self._values[item]
38
-
39
- def __setitem__(self, key: str, value: Any) -> None:
40
- if key not in self.FIELDS:
41
- raise KeyError(f"{self.__class__.__name__} 不包含字段:{key}")
42
-
43
- field = self.FIELDS[key]
44
- try:
45
- validated_value = field.validate(value, field_name=key)
46
- self._values[key] = validated_value
47
- except Exception as e:
48
- error_lines = [
49
- "",
50
- "【字段校验失败】",
51
- f"字段名称: {key}",
52
- f"数据类型: {type(value)}",
53
- f"原始值: {repr(value)}",
54
- f"是否允许空值: {field.nullable}",
55
- f"错误原因: {str(e)}",
56
- ""
57
- ]
58
- detailed_error = "\n".join(error_lines)
59
- raise type(e)(detailed_error) from e
60
-
61
- def __delitem__(self, key: str) -> None:
62
- del self._values[key]
63
-
64
- def __setattr__(self, key: str, value: Any) -> None:
65
- if not key.startswith("_"):
66
- raise AttributeError(
67
- f"设置字段值请使用 item[{key!r}] = {value!r}"
68
- )
69
- super().__setattr__(key, value)
70
-
71
- def __getattr__(self, item: str) -> Any:
72
- raise AttributeError(
73
- f"{self.__class__.__name__} 不支持字段:{item}。"
74
- f"请先在 `{self.__class__.__name__}` 中声明该字段,再通过 item[{item!r}] 获取。"
75
- )
76
-
77
- def __getattribute__(self, item: str) -> Any:
78
- try:
79
- field = super().__getattribute__("FIELDS")
80
- if isinstance(field, dict) and item in field:
81
- raise ItemAttributeError(
82
- f"获取字段值请使用 item[{item!r}]"
83
- )
84
- except AttributeError:
85
- pass # 如果 FIELDS 尚未定义,继续执行后续逻辑
86
- return super().__getattribute__(item)
87
-
88
- def __repr__(self) -> str:
89
- return pformat(dict(self))
90
-
91
- __str__ = __repr__
92
-
93
- def __iter__(self) -> Iterator[str]:
94
- return iter(self._values)
95
-
96
- def __len__(self) -> int:
97
- return len(self._values)
98
-
99
- def to_dict(self) -> Dict[str, Any]:
100
- """转换为字典"""
101
- return dict(self)
102
-
103
- def copy(self) -> "Item":
104
- """深拷贝当前 Item"""
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ Item 类定义
5
+ """
6
+ from copy import deepcopy
7
+ from pprint import pformat
8
+ from typing import Any, Iterator, Dict
9
+ from collections.abc import MutableMapping
10
+
11
+ from .base import ItemMeta
12
+ from crawlo.exceptions import ItemInitError, ItemAttributeError
13
+
14
+
15
+ class Item(MutableMapping, metaclass=ItemMeta):
16
+ """
17
+ 数据项基类,用于定义结构化数据
18
+ """
19
+ FIELDS: Dict[str, Any] = {}
20
+
21
+ def __init__(self, *args, **kwargs):
22
+ if args:
23
+ raise ItemInitError(f"{self.__class__.__name__} 不支持位置参数:{args},请使用关键字参数初始化。")
24
+
25
+ self._values: Dict[str, Any] = {}
26
+
27
+ # 初始化字段,默认值填充
28
+ for field_name, field_obj in self.FIELDS.items():
29
+ if field_obj.default is not None:
30
+ self._values[field_name] = field_obj.default
31
+
32
+ # 覆盖默认值或设置新值
33
+ for key, value in kwargs.items():
34
+ self[key] = value
35
+
36
+ def __getitem__(self, item: str) -> Any:
37
+ return self._values[item]
38
+
39
+ def __setitem__(self, key: str, value: Any) -> None:
40
+ if key not in self.FIELDS:
41
+ raise KeyError(f"{self.__class__.__name__} 不包含字段:{key}")
42
+
43
+ field = self.FIELDS[key]
44
+ try:
45
+ validated_value = field.validate(value, field_name=key)
46
+ self._values[key] = validated_value
47
+ except Exception as e:
48
+ error_lines = [
49
+ "",
50
+ "【字段校验失败】",
51
+ f"字段名称: {key}",
52
+ f"数据类型: {type(value)}",
53
+ f"原始值: {repr(value)}",
54
+ f"是否允许空值: {field.nullable}",
55
+ f"错误原因: {str(e)}",
56
+ ""
57
+ ]
58
+ detailed_error = "\n".join(error_lines)
59
+ raise type(e)(detailed_error) from e
60
+
61
+ def __delitem__(self, key: str) -> None:
62
+ del self._values[key]
63
+
64
+ def __setattr__(self, key: str, value: Any) -> None:
65
+ if not key.startswith("_"):
66
+ raise AttributeError(
67
+ f"设置字段值请使用 item[{key!r}] = {value!r}"
68
+ )
69
+ super().__setattr__(key, value)
70
+
71
+ def __getattr__(self, item: str) -> Any:
72
+ raise AttributeError(
73
+ f"{self.__class__.__name__} 不支持字段:{item}。"
74
+ f"请先在 `{self.__class__.__name__}` 中声明该字段,再通过 item[{item!r}] 获取。"
75
+ )
76
+
77
+ def __getattribute__(self, item: str) -> Any:
78
+ try:
79
+ field = super().__getattribute__("FIELDS")
80
+ if isinstance(field, dict) and item in field:
81
+ raise ItemAttributeError(
82
+ f"获取字段值请使用 item[{item!r}]"
83
+ )
84
+ except AttributeError:
85
+ pass # 如果 FIELDS 尚未定义,继续执行后续逻辑
86
+ return super().__getattribute__(item)
87
+
88
+ def __repr__(self) -> str:
89
+ return pformat(dict(self))
90
+
91
+ __str__ = __repr__
92
+
93
+ def __iter__(self) -> Iterator[str]:
94
+ return iter(self._values)
95
+
96
+ def __len__(self) -> int:
97
+ return len(self._values)
98
+
99
+ def to_dict(self) -> Dict[str, Any]:
100
+ """转换为字典"""
101
+ return dict(self)
102
+
103
+ def copy(self) -> "Item":
104
+ """深拷贝当前 Item"""
105
105
  return deepcopy(self)
@@ -1,21 +1,21 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- from crawlo import Request, Response
4
-
5
-
6
- class BaseMiddleware(object):
7
- def process_request(self, request, spider) -> None | Request | Response:
8
- # 请求预处理
9
- pass
10
-
11
- def process_response(self, request, response, spider) -> Request | Response:
12
- # 响应预处理
13
- pass
14
-
15
- def process_exception(self, request, exp, spider) -> None | Request | Response:
16
- # 异常预处理
17
- pass
18
-
19
- @classmethod
20
- def create_instance(cls, crawler):
21
- return cls()
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ from crawlo import Request, Response
4
+
5
+
6
+ class BaseMiddleware(object):
7
+ def process_request(self, request, spider) -> None | Request | Response:
8
+ # 请求预处理
9
+ pass
10
+
11
+ def process_response(self, request, response, spider) -> Request | Response:
12
+ # 响应预处理
13
+ pass
14
+
15
+ def process_exception(self, request, exp, spider) -> None | Request | Response:
16
+ # 异常预处理
17
+ pass
18
+
19
+ @classmethod
20
+ def create_instance(cls, crawler):
21
+ return cls()
@@ -1,32 +1,32 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- from crawlo.event import spider_opened
4
-
5
-
6
- class DefaultHeaderMiddleware(object):
7
-
8
- def __init__(self, user_agent, headers, spider):
9
- self.user_agent = user_agent
10
- self.headers = headers
11
- self.spider = spider
12
-
13
- @classmethod
14
- def create_instance(cls, crawler):
15
- o = cls(
16
- user_agent=crawler.settings.get('USER_AGENT'),
17
- headers=crawler.settings.get_dict('DEFAULT_HEADERS'),
18
- spider=crawler.spider
19
- )
20
- crawler.subscriber.subscribe(o.spider_opened, event=spider_opened)
21
- return o
22
-
23
- async def spider_opened(self):
24
- self.user_agent = getattr(self.spider, 'user_agent', self.user_agent)
25
- self.headers = getattr(self.spider, 'headers', self.headers)
26
- if self.user_agent:
27
- self.headers.setdefault('User-Agent', self.user_agent)
28
-
29
- def process_request(self, request, _spider):
30
- if self.headers:
31
- for key, value in self.headers.items():
32
- request.headers.setdefault(key, value)
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ from crawlo.event import spider_opened
4
+
5
+
6
+ class DefaultHeaderMiddleware(object):
7
+
8
+ def __init__(self, user_agent, headers, spider):
9
+ self.user_agent = user_agent
10
+ self.headers = headers
11
+ self.spider = spider
12
+
13
+ @classmethod
14
+ def create_instance(cls, crawler):
15
+ o = cls(
16
+ user_agent=crawler.settings.get('USER_AGENT'),
17
+ headers=crawler.settings.get_dict('DEFAULT_HEADERS'),
18
+ spider=crawler.spider
19
+ )
20
+ crawler.subscriber.subscribe(o.spider_opened, event=spider_opened)
21
+ return o
22
+
23
+ async def spider_opened(self):
24
+ self.user_agent = getattr(self.spider, 'user_agent', self.user_agent)
25
+ self.headers = getattr(self.spider, 'headers', self.headers)
26
+ if self.user_agent:
27
+ self.headers.setdefault('User-Agent', self.user_agent)
28
+
29
+ def process_request(self, request, _spider):
30
+ if self.headers:
31
+ for key, value in self.headers.items():
32
+ request.headers.setdefault(key, value)
@@ -1,28 +1,28 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- from asyncio import sleep
4
- from random import uniform
5
- from crawlo.utils.log import get_logger
6
- from crawlo.exceptions import NotConfiguredError
7
-
8
-
9
- class DownloadDelayMiddleware(object):
10
-
11
- def __init__(self, settings, log_level):
12
- self.delay = settings.get_float("DOWNLOAD_DELAY")
13
- if not self.delay:
14
- raise NotConfiguredError
15
- self.randomness = settings.get_bool("RANDOMNESS")
16
- self.floor, self.upper = settings.get_list("RANDOM_RANGE")
17
- self.logger = get_logger(self.__class__.__name__, log_level)
18
-
19
- @classmethod
20
- def create_instance(cls, crawler):
21
- o = cls(settings=crawler.settings, log_level=crawler.settings.get('LOG_LEVEL'))
22
- return o
23
-
24
- async def process_request(self, _request, _spider):
25
- if self.randomness:
26
- await sleep(uniform(self.delay * self.floor, self.delay * self.upper))
27
- else:
28
- await sleep(self.delay)
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ from asyncio import sleep
4
+ from random import uniform
5
+ from crawlo.utils.log import get_logger
6
+ from crawlo.exceptions import NotConfiguredError
7
+
8
+
9
+ class DownloadDelayMiddleware(object):
10
+
11
+ def __init__(self, settings, log_level):
12
+ self.delay = settings.get_float("DOWNLOAD_DELAY")
13
+ if not self.delay:
14
+ raise NotConfiguredError
15
+ self.randomness = settings.get_bool("RANDOMNESS")
16
+ self.floor, self.upper = settings.get_list("RANDOM_RANGE")
17
+ self.logger = get_logger(self.__class__.__name__, log_level)
18
+
19
+ @classmethod
20
+ def create_instance(cls, crawler):
21
+ o = cls(settings=crawler.settings, log_level=crawler.settings.get('LOG_LEVEL'))
22
+ return o
23
+
24
+ async def process_request(self, _request, _spider):
25
+ if self.randomness:
26
+ await sleep(uniform(self.delay * self.floor, self.delay * self.upper))
27
+ else:
28
+ await sleep(self.delay)