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

crawlo/middleware/retry.py

@@ -1,125 +1,125 @@
-#!/usr/bin/python
-# -*- coding:UTF-8 -*-
-from typing import List
-import asyncio
-
-try:
-    from anyio import EndOfStream
-except ImportError:
-    # 如果 anyio 不可用或者 EndOfStream 不存在，创建一个占位符
-    class EndOfStream(Exception):
-        pass
-
-try:
-    from httpcore import ReadError
-except ImportError:
-    class ReadError(Exception):
-        pass
-
-try:
-    from httpx import RemoteProtocolError, ConnectError, ReadTimeout
-except ImportError:
-    class RemoteProtocolError(Exception):
-        pass
-    class ConnectError(Exception):
-        pass
-    class ReadTimeout(Exception):
-        pass
-
-try:
-    from aiohttp.client_exceptions import ClientConnectionError, ClientPayloadError
-    from aiohttp import ClientConnectorError, ClientTimeout, ClientConnectorSSLError, ClientResponseError
-except ImportError:
-    class ClientConnectionError(Exception):
-        pass
-    class ClientPayloadError(Exception):
-        pass
-    class ClientConnectorError(Exception):
-        pass
-    class ClientTimeout(Exception):
-        pass
-    class ClientConnectorSSLError(Exception):
-        pass
-    class ClientResponseError(Exception):
-        pass
-
-from crawlo.utils.log import get_logger
-from crawlo.stats_collector import StatsCollector
-
-_retry_exceptions = [
-    EndOfStream,
-    ReadError,
-    asyncio.TimeoutError,
-    ConnectError,
-    ReadTimeout,
-    ClientConnectorError,
-    ClientResponseError,
-    RemoteProtocolError,
-    ClientTimeout,
-    ClientConnectorSSLError,
-    ClientPayloadError,
-    ClientConnectionError
-]
-
-
-class RetryMiddleware(object):
-
-    def __init__(
-            self,
-            *,
-            retry_http_codes: List,
-            ignore_http_codes: List,
-            max_retry_times: int,
-            retry_exceptions: List,
-            stats: StatsCollector,
-            retry_priority: int
-    ):
-        self.retry_http_codes = retry_http_codes
-        self.ignore_http_codes = ignore_http_codes
-        self.max_retry_times = max_retry_times
-        self.retry_exceptions = tuple(retry_exceptions + _retry_exceptions)
-        self.retry_priority = retry_priority
-        self.stats = stats
-        self.logger = get_logger(self.__class__.__name__)
-
-    @classmethod
-    def create_instance(cls, crawler):
-        o = cls(
-            retry_http_codes=crawler.settings.get_list('RETRY_HTTP_CODES'),
-            ignore_http_codes=crawler.settings.get_list('IGNORE_HTTP_CODES'),
-            max_retry_times=crawler.settings.get_int('MAX_RETRY_TIMES'),
-            retry_exceptions=crawler.settings.get_list('RETRY_EXCEPTIONS'),
-            stats=crawler.stats,
-            retry_priority=crawler.settings.get_int('RETRY_PRIORITY')
-        )
-        return o
-
-    def process_response(self, request, response, spider):
-        if request.meta.get('dont_retry', False):
-            return response
-        if response.status_code in self.ignore_http_codes:
-            return response
-        if response.status_code in self.retry_http_codes:
-            # 重试逻辑
-            reason = f"response code {response.status_code}"
-            return self._retry(request, reason, spider) or response
-        return response
-
-    def process_exception(self, request, exc, spider):
-        if isinstance(exc, self.retry_exceptions) and not request.meta.get('dont_retry', False):
-            return self._retry(request=request, reason=type(exc).__name__, spider=spider)
-
-    def _retry(self, request, reason, spider):
-        retry_times = request.meta.get('retry_times', 0)
-        if retry_times < self.max_retry_times:
-            retry_times += 1
-            self.logger.info(f"{spider} {request} {reason} retrying {retry_times} time...")
-            request.meta['retry_times'] = retry_times
-            # request.dont_retry = True
-            request.meta['dont_retry'] = True
-            request.priority = request.priority + self.retry_priority
-            self.stats.inc_value("retry_count")
-            return request
-        else:
-            self.logger.warning(f"{spider} {request} {reason} retry max {self.max_retry_times} times, give up.")
+#!/usr/bin/python
+# -*- coding:UTF-8 -*-
+from typing import List
+import asyncio
+
+try:
+    from anyio import EndOfStream
+except ImportError:
+    # 如果 anyio 不可用或者 EndOfStream 不存在，创建一个占位符
+    class EndOfStream(Exception):
+        pass
+
+try:
+    from httpcore import ReadError
+except ImportError:
+    class ReadError(Exception):
+        pass
+
+try:
+    from httpx import RemoteProtocolError, ConnectError, ReadTimeout
+except ImportError:
+    class RemoteProtocolError(Exception):
+        pass
+    class ConnectError(Exception):
+        pass
+    class ReadTimeout(Exception):
+        pass
+
+try:
+    from aiohttp.client_exceptions import ClientConnectionError, ClientPayloadError
+    from aiohttp import ClientConnectorError, ClientTimeout, ClientConnectorSSLError, ClientResponseError
+except ImportError:
+    class ClientConnectionError(Exception):
+        pass
+    class ClientPayloadError(Exception):
+        pass
+    class ClientConnectorError(Exception):
+        pass
+    class ClientTimeout(Exception):
+        pass
+    class ClientConnectorSSLError(Exception):
+        pass
+    class ClientResponseError(Exception):
+        pass
+
+from crawlo.utils.log import get_logger
+from crawlo.stats_collector import StatsCollector
+
+_retry_exceptions = [
+    EndOfStream,
+    ReadError,
+    asyncio.TimeoutError,
+    ConnectError,
+    ReadTimeout,
+    ClientConnectorError,
+    ClientResponseError,
+    RemoteProtocolError,
+    ClientTimeout,
+    ClientConnectorSSLError,
+    ClientPayloadError,
+    ClientConnectionError
+]
+
+
+class RetryMiddleware(object):
+
+    def __init__(
+            self,
+            *,
+            retry_http_codes: List,
+            ignore_http_codes: List,
+            max_retry_times: int,
+            retry_exceptions: List,
+            stats: StatsCollector,
+            retry_priority: int
+    ):
+        self.retry_http_codes = retry_http_codes
+        self.ignore_http_codes = ignore_http_codes
+        self.max_retry_times = max_retry_times
+        self.retry_exceptions = tuple(retry_exceptions + _retry_exceptions)
+        self.retry_priority = retry_priority
+        self.stats = stats
+        self.logger = get_logger(self.__class__.__name__)
+
+    @classmethod
+    def create_instance(cls, crawler):
+        o = cls(
+            retry_http_codes=crawler.settings.get_list('RETRY_HTTP_CODES'),
+            ignore_http_codes=crawler.settings.get_list('IGNORE_HTTP_CODES'),
+            max_retry_times=crawler.settings.get_int('MAX_RETRY_TIMES'),
+            retry_exceptions=crawler.settings.get_list('RETRY_EXCEPTIONS'),
+            stats=crawler.stats,
+            retry_priority=crawler.settings.get_int('RETRY_PRIORITY')
+        )
+        return o
+
+    def process_response(self, request, response, spider):
+        if request.meta.get('dont_retry', False):
+            return response
+        if response.status_code in self.ignore_http_codes:
+            return response
+        if response.status_code in self.retry_http_codes:
+            # 重试逻辑
+            reason = f"response code {response.status_code}"
+            return self._retry(request, reason, spider) or response
+        return response
+
+    def process_exception(self, request, exc, spider):
+        if isinstance(exc, self.retry_exceptions) and not request.meta.get('dont_retry', False):
+            return self._retry(request=request, reason=type(exc).__name__, spider=spider)
+
+    def _retry(self, request, reason, spider):
+        retry_times = request.meta.get('retry_times', 0)
+        if retry_times < self.max_retry_times:
+            retry_times += 1
+            self.logger.info(f"{spider} {request} {reason} retrying {retry_times} time...")
+            request.meta['retry_times'] = retry_times
+            # request.dont_retry = True
+            request.meta['dont_retry'] = True
+            request.priority = request.priority + self.retry_priority
+            self.stats.inc_value("retry_count")
+            return request
+        else:
+            self.logger.warning(f"{spider} {request} {reason} retry max {self.max_retry_times} times, give up.")
             return None

crawlo/mode_manager.py

@@ -1,201 +1,201 @@
-#!/usr/bin/python
-# -*- coding: UTF-8 -*-
-"""
-运行模式管理器
-==============
-管理 Crawlo 框架的不同运行模式，提供优雅的配置方式。
-
-支持的运行模式：
-1. standalone - 单机模式（默认）
-2. distributed - 分布式模式  
-3. auto - 自动检测模式
-"""
-
-from typing import Dict, Any, Optional
-from enum import Enum
-import os
-from crawlo.utils.log import get_logger
-
-
-class RunMode(Enum):
-    """运行模式枚举"""
-    STANDALONE = "standalone"    # 单机模式
-    DISTRIBUTED = "distributed"  # 分布式模式
-    AUTO = "auto"               # 自动检测模式
-
-
-class ModeManager:
-    """运行模式管理器"""
-    
-    def __init__(self):
-        self.logger = get_logger(self.__class__.__name__)
-    
-    @staticmethod
-    def get_standalone_settings() -> Dict[str, Any]:
-        """获取单机模式配置"""
-        return {
-            'QUEUE_TYPE': 'memory',
-            'FILTER_CLASS': 'crawlo.filters.memory_filter.MemoryFilter',
-            'CONCURRENCY': 8,
-            'MAX_RUNNING_SPIDERS': 1,
-            'DOWNLOAD_DELAY': 1.0,
-            'LOG_LEVEL': 'INFO',
-        }
-    
-    @staticmethod
-    def get_distributed_settings(
-        redis_host: str = '127.0.0.1',
-        redis_port: int = 6379,
-        redis_password: Optional[str] = None,
-        project_name: str = 'crawlo'
-    ) -> Dict[str, Any]:
-        """获取分布式模式配置"""
-        # 构建 Redis URL
-        if redis_password:
-            redis_url = f'redis://:{redis_password}@{redis_host}:{redis_port}/0'
-        else:
-            redis_url = f'redis://{redis_host}:{redis_port}/0'
-        
-        return {
-            'QUEUE_TYPE': 'redis',
-            'FILTER_CLASS': 'crawlo.filters.aioredis_filter.AioRedisFilter',
-            'REDIS_HOST': redis_host,
-            'REDIS_PORT': redis_port,
-            'REDIS_PASSWORD': redis_password,
-            'REDIS_URL': redis_url,
-            'SCHEDULER_QUEUE_NAME': f'{project_name}:requests',
-            'REDIS_KEY': f'{project_name}:fingerprint',
-            'CONCURRENCY': 16,
-            'MAX_RUNNING_SPIDERS': 1,
-            'DOWNLOAD_DELAY': 1.0,
-            'LOG_LEVEL': 'INFO',
-        }
-    
-    @staticmethod
-    def get_auto_settings() -> Dict[str, Any]:
-        """获取自动检测模式配置"""
-        return {
-            'QUEUE_TYPE': 'auto',
-            'FILTER_CLASS': 'crawlo.filters.memory_filter.MemoryFilter',  # 默认内存过滤器
-            'CONCURRENCY': 12,
-            'MAX_RUNNING_SPIDERS': 1,
-            'DOWNLOAD_DELAY': 1.0,
-            'LOG_LEVEL': 'INFO',
-        }
-    
-    def resolve_mode_settings(
-        self, 
-        mode: str = 'standalone',
-        **kwargs
-    ) -> Dict[str, Any]:
-        """
-        解析运行模式并返回对应配置
-        
-        Args:
-            mode: 运行模式 ('standalone', 'distributed', 'auto')
-            **kwargs: 额外配置参数
-            
-        Returns:
-            Dict[str, Any]: 配置字典
-        """
-        mode = RunMode(mode.lower())
-        
-        if mode == RunMode.STANDALONE:
-            self.logger.info("🏠 使用单机模式 - 简单快速，适合开发和中小规模爬取")
-            settings = self.get_standalone_settings()
-            
-        elif mode == RunMode.DISTRIBUTED:
-            self.logger.info("🌐 使用分布式模式 - 支持多节点扩展，适合大规模爬取")
-            settings = self.get_distributed_settings(
-                redis_host=kwargs.get('redis_host', '127.0.0.1'),
-                redis_port=kwargs.get('redis_port', 6379),
-                redis_password=kwargs.get('redis_password'),
-                project_name=kwargs.get('project_name', 'crawlo')
-            )
-            
-        elif mode == RunMode.AUTO:
-            self.logger.info("🤖 使用自动检测模式 - 智能选择最佳运行方式")
-            settings = self.get_auto_settings()
-            
-        else:
-            raise ValueError(f"不支持的运行模式: {mode}")
-        
-        # 合并用户自定义配置
-        user_settings = {k: v for k, v in kwargs.items() 
-                        if k not in ['redis_host', 'redis_port', 'redis_password', 'project_name']}
-        settings.update(user_settings)
-        
-        return settings
-    
-    def from_environment(self) -> Dict[str, Any]:
-        """从环境变量构建配置"""
-        config = {}
-        
-        # 扫描 CRAWLO_ 前缀的环境变量
-        for key, value in os.environ.items():
-            if key.startswith('CRAWLO_'):
-                config_key = key[7:]  # 去掉 'CRAWLO_' 前缀
-                # 简单的类型转换
-                if value.lower() in ('true', 'false'):
-                    config[config_key] = value.lower() == 'true'
-                elif value.isdigit():
-                    config[config_key] = int(value)
-                else:
-                    try:
-                        config[config_key] = float(value)
-                    except ValueError:
-                        config[config_key] = value
-        
-        return config
-
-
-# 便利函数
-def standalone_mode(**kwargs) -> Dict[str, Any]:
-    """快速创建单机模式配置"""
-    return ModeManager().resolve_mode_settings('standalone', **kwargs)
-
-
-def distributed_mode(
-    redis_host: str = '127.0.0.1',
-    redis_port: int = 6379,
-    redis_password: Optional[str] = None,
-    project_name: str = 'crawlo',
-    **kwargs
-) -> Dict[str, Any]:
-    """快速创建分布式模式配置"""
-    return ModeManager().resolve_mode_settings(
-        'distributed',
-        redis_host=redis_host,
-        redis_port=redis_port,
-        redis_password=redis_password,
-        project_name=project_name,
-        **kwargs
-    )
-
-
-def auto_mode(**kwargs) -> Dict[str, Any]:
-    """快速创建自动检测模式配置"""
-    return ModeManager().resolve_mode_settings('auto', **kwargs)
-
-
-# 环境变量支持
-def from_env(default_mode: str = 'standalone') -> Dict[str, Any]:
-    """从环境变量创建配置"""
-    mode = os.getenv('CRAWLO_MODE', default_mode).lower()
-    
-    if mode == 'distributed':
-        return distributed_mode(
-            redis_host=os.getenv('REDIS_HOST', '127.0.0.1'),
-            redis_port=int(os.getenv('REDIS_PORT', 6379)),
-            redis_password=os.getenv('REDIS_PASSWORD'),
-            project_name=os.getenv('PROJECT_NAME', 'crawlo'),
-            CONCURRENCY=int(os.getenv('CONCURRENCY', 16)),
-        )
-    elif mode == 'auto':
-        return auto_mode(
-            CONCURRENCY=int(os.getenv('CONCURRENCY', 12)),
-        )
-    else:  # standalone
-        return standalone_mode(
-            CONCURRENCY=int(os.getenv('CONCURRENCY', 8)),
+#!/usr/bin/python
+# -*- coding: UTF-8 -*-
+"""
+运行模式管理器
+==============
+管理 Crawlo 框架的不同运行模式，提供优雅的配置方式。
+
+支持的运行模式：
+1. standalone - 单机模式（默认）
+2. distributed - 分布式模式  
+3. auto - 自动检测模式
+"""
+
+from typing import Dict, Any, Optional
+from enum import Enum
+import os
+from crawlo.utils.log import get_logger
+
+
+class RunMode(Enum):
+    """运行模式枚举"""
+    STANDALONE = "standalone"    # 单机模式
+    DISTRIBUTED = "distributed"  # 分布式模式
+    AUTO = "auto"               # 自动检测模式
+
+
+class ModeManager:
+    """运行模式管理器"""
+    
+    def __init__(self):
+        self.logger = get_logger(self.__class__.__name__)
+    
+    @staticmethod
+    def get_standalone_settings() -> Dict[str, Any]:
+        """获取单机模式配置"""
+        return {
+            'QUEUE_TYPE': 'memory',
+            'FILTER_CLASS': 'crawlo.filters.memory_filter.MemoryFilter',
+            'CONCURRENCY': 8,
+            'MAX_RUNNING_SPIDERS': 1,
+            'DOWNLOAD_DELAY': 1.0,
+            'LOG_LEVEL': 'INFO',
+        }
+    
+    @staticmethod
+    def get_distributed_settings(
+        redis_host: str = '127.0.0.1',
+        redis_port: int = 6379,
+        redis_password: Optional[str] = None,
+        project_name: str = 'crawlo'
+    ) -> Dict[str, Any]:
+        """获取分布式模式配置"""
+        # 构建 Redis URL
+        if redis_password:
+            redis_url = f'redis://:{redis_password}@{redis_host}:{redis_port}/0'
+        else:
+            redis_url = f'redis://{redis_host}:{redis_port}/0'
+        
+        return {
+            'QUEUE_TYPE': 'redis',
+            'FILTER_CLASS': 'crawlo.filters.aioredis_filter.AioRedisFilter',
+            'REDIS_HOST': redis_host,
+            'REDIS_PORT': redis_port,
+            'REDIS_PASSWORD': redis_password,
+            'REDIS_URL': redis_url,
+            'SCHEDULER_QUEUE_NAME': f'{project_name}:requests',
+            'REDIS_KEY': f'{project_name}:fingerprint',
+            'CONCURRENCY': 16,
+            'MAX_RUNNING_SPIDERS': 1,
+            'DOWNLOAD_DELAY': 1.0,
+            'LOG_LEVEL': 'INFO',
+        }
+    
+    @staticmethod
+    def get_auto_settings() -> Dict[str, Any]:
+        """获取自动检测模式配置"""
+        return {
+            'QUEUE_TYPE': 'auto',
+            'FILTER_CLASS': 'crawlo.filters.memory_filter.MemoryFilter',  # 默认内存过滤器
+            'CONCURRENCY': 12,
+            'MAX_RUNNING_SPIDERS': 1,
+            'DOWNLOAD_DELAY': 1.0,
+            'LOG_LEVEL': 'INFO',
+        }
+    
+    def resolve_mode_settings(
+        self, 
+        mode: str = 'standalone',
+        **kwargs
+    ) -> Dict[str, Any]:
+        """
+        解析运行模式并返回对应配置
+        
+        Args:
+            mode: 运行模式 ('standalone', 'distributed', 'auto')
+            **kwargs: 额外配置参数
+            
+        Returns:
+            Dict[str, Any]: 配置字典
+        """
+        mode = RunMode(mode.lower())
+        
+        if mode == RunMode.STANDALONE:
+            self.logger.info("🏠 使用单机模式 - 简单快速，适合开发和中小规模爬取")
+            settings = self.get_standalone_settings()
+            
+        elif mode == RunMode.DISTRIBUTED:
+            self.logger.info("🌐 使用分布式模式 - 支持多节点扩展，适合大规模爬取")
+            settings = self.get_distributed_settings(
+                redis_host=kwargs.get('redis_host', '127.0.0.1'),
+                redis_port=kwargs.get('redis_port', 6379),
+                redis_password=kwargs.get('redis_password'),
+                project_name=kwargs.get('project_name', 'crawlo')
+            )
+            
+        elif mode == RunMode.AUTO:
+            self.logger.info("🤖 使用自动检测模式 - 智能选择最佳运行方式")
+            settings = self.get_auto_settings()
+            
+        else:
+            raise ValueError(f"不支持的运行模式: {mode}")
+        
+        # 合并用户自定义配置
+        user_settings = {k: v for k, v in kwargs.items() 
+                        if k not in ['redis_host', 'redis_port', 'redis_password', 'project_name']}
+        settings.update(user_settings)
+        
+        return settings
+    
+    def from_environment(self) -> Dict[str, Any]:
+        """从环境变量构建配置"""
+        config = {}
+        
+        # 扫描 CRAWLO_ 前缀的环境变量
+        for key, value in os.environ.items():
+            if key.startswith('CRAWLO_'):
+                config_key = key[7:]  # 去掉 'CRAWLO_' 前缀
+                # 简单的类型转换
+                if value.lower() in ('true', 'false'):
+                    config[config_key] = value.lower() == 'true'
+                elif value.isdigit():
+                    config[config_key] = int(value)
+                else:
+                    try:
+                        config[config_key] = float(value)
+                    except ValueError:
+                        config[config_key] = value
+        
+        return config
+
+
+# 便利函数
+def standalone_mode(**kwargs) -> Dict[str, Any]:
+    """快速创建单机模式配置"""
+    return ModeManager().resolve_mode_settings('standalone', **kwargs)
+
+
+def distributed_mode(
+    redis_host: str = '127.0.0.1',
+    redis_port: int = 6379,
+    redis_password: Optional[str] = None,
+    project_name: str = 'crawlo',
+    **kwargs
+) -> Dict[str, Any]:
+    """快速创建分布式模式配置"""
+    return ModeManager().resolve_mode_settings(
+        'distributed',
+        redis_host=redis_host,
+        redis_port=redis_port,
+        redis_password=redis_password,
+        project_name=project_name,
+        **kwargs
+    )
+
+
+def auto_mode(**kwargs) -> Dict[str, Any]:
+    """快速创建自动检测模式配置"""
+    return ModeManager().resolve_mode_settings('auto', **kwargs)
+
+
+# 环境变量支持
+def from_env(default_mode: str = 'standalone') -> Dict[str, Any]:
+    """从环境变量创建配置"""
+    mode = os.getenv('CRAWLO_MODE', default_mode).lower()
+    
+    if mode == 'distributed':
+        return distributed_mode(
+            redis_host=os.getenv('REDIS_HOST', '127.0.0.1'),
+            redis_port=int(os.getenv('REDIS_PORT', 6379)),
+            redis_password=os.getenv('REDIS_PASSWORD'),
+            project_name=os.getenv('PROJECT_NAME', 'crawlo'),
+            CONCURRENCY=int(os.getenv('CONCURRENCY', 16)),
+        )
+    elif mode == 'auto':
+        return auto_mode(
+            CONCURRENCY=int(os.getenv('CONCURRENCY', 12)),
+        )
+    else:  # standalone
+        return standalone_mode(
+            CONCURRENCY=int(os.getenv('CONCURRENCY', 8)),
         )