crawlo 1.2.6__py3-none-any.whl → 1.2.8__py3-none-any.whl

crawlo/middleware/request_ignore.py

@@ -1,87 +1,87 @@
-#!/usr/bin/python
-# -*- coding:UTF-8 -*-
-"""
-RequestIgnoreMiddleware 中间件
-用于处理和记录被忽略的请求
-"""
-from crawlo.utils.log import get_logger
-from crawlo.exceptions import IgnoreRequestError
-from crawlo.event import ignore_request
-
-
-class RequestIgnoreMiddleware(object):
-    """
-    RequestIgnoreMiddleware 中间件
-    用于处理和记录被忽略的请求，提供详细的统计信息
-    """
-
-    def __init__(self, stats, log_level):
-        """
-        初始化中间件
-        
-        Args:
-            stats: 统计信息收集器
-            log_level: 日志级别
-        """
-        self.logger = get_logger(self.__class__.__name__, log_level)
-        self.stats = stats
-
-    @classmethod
-    def create_instance(cls, crawler):
-        """
-        创建中间件实例
-        
-        Args:
-            crawler: 爬虫实例
-            
-        Returns:
-            RequestIgnoreMiddleware: 中间件实例
-        """
-        o = cls(stats=crawler.stats, log_level=crawler.settings.get('LOG_LEVEL'))
-        crawler.subscriber.subscribe(o.request_ignore, event=ignore_request)
-        return o
-
-    async def request_ignore(self, exc, request, _spider):
-        """
-        处理被忽略的请求事件
-        
-        Args:
-            exc: 异常对象
-            request: 被忽略的请求
-            _spider: 爬虫实例
-        """
-        # 记录被忽略的请求
-        self.logger.info(f'请求被忽略: {request.url}')
-        self.stats.inc_value('request_ignore_count')
-        
-        # 记录忽略原因
-        reason = getattr(exc, 'msg', 'unknown')
-        if reason:
-            self.stats.inc_value(f'request_ignore_count/reason/{reason}')
-            
-        # 记录请求的域名分布
-        try:
-            from urllib.parse import urlparse
-            parsed_url = urlparse(request.url)
-            domain = parsed_url.netloc
-            if domain:
-                self.stats.inc_value(f'request_ignore_count/domain/{domain}')
-        except Exception:
-            self.stats.inc_value('request_ignore_count/domain/invalid_url')
-
-    @staticmethod
-    def process_exception(_request, exc, _spider):
-        """
-        处理异常，识别IgnoreRequestError
-        
-        Args:
-            _request: 请求对象
-            exc: 异常对象
-            _spider: 爬虫实例
-            
-        Returns:
-            bool: 如果是IgnoreRequestError则返回True，否则返回None
-        """
-        if isinstance(exc, IgnoreRequestError):
-            return True
+#!/usr/bin/python
+# -*- coding:UTF-8 -*-
+"""
+RequestIgnoreMiddleware 中间件
+用于处理和记录被忽略的请求
+"""
+from crawlo.utils.log import get_logger
+from crawlo.exceptions import IgnoreRequestError
+from crawlo.event import ignore_request
+
+
+class RequestIgnoreMiddleware(object):
+    """
+    RequestIgnoreMiddleware 中间件
+    用于处理和记录被忽略的请求，提供详细的统计信息
+    """
+
+    def __init__(self, stats, log_level):
+        """
+        初始化中间件
+        
+        Args:
+            stats: 统计信息收集器
+            log_level: 日志级别
+        """
+        self.logger = get_logger(self.__class__.__name__, log_level)
+        self.stats = stats
+
+    @classmethod
+    def create_instance(cls, crawler):
+        """
+        创建中间件实例
+        
+        Args:
+            crawler: 爬虫实例
+            
+        Returns:
+            RequestIgnoreMiddleware: 中间件实例
+        """
+        o = cls(stats=crawler.stats, log_level=crawler.settings.get('LOG_LEVEL'))
+        crawler.subscriber.subscribe(o.request_ignore, event=ignore_request)
+        return o
+
+    async def request_ignore(self, exc, request, _spider):
+        """
+        处理被忽略的请求事件
+        
+        Args:
+            exc: 异常对象
+            request: 被忽略的请求
+            _spider: 爬虫实例
+        """
+        # 记录被忽略的请求
+        self.logger.info(f'请求被忽略: {request.url}')
+        self.stats.inc_value('request_ignore_count')
+        
+        # 记录忽略原因
+        reason = getattr(exc, 'msg', 'unknown')
+        if reason:
+            self.stats.inc_value(f'request_ignore_count/reason/{reason}')
+            
+        # 记录请求的域名分布
+        try:
+            from urllib.parse import urlparse
+            parsed_url = urlparse(request.url)
+            domain = parsed_url.netloc
+            if domain:
+                self.stats.inc_value(f'request_ignore_count/domain/{domain}')
+        except Exception:
+            self.stats.inc_value('request_ignore_count/domain/invalid_url')
+
+    @staticmethod
+    def process_exception(_request, exc, _spider):
+        """
+        处理异常，识别IgnoreRequestError
+        
+        Args:
+            _request: 请求对象
+            exc: 异常对象
+            _spider: 爬虫实例
+            
+        Returns:
+            bool: 如果是IgnoreRequestError则返回True，否则返回None
+        """
+        if isinstance(exc, IgnoreRequestError):
+            return True
         return None

crawlo/middleware/response_code.py

@@ -1,164 +1,164 @@
-#!/usr/bin/python
-# -*- coding:UTF-8 -*-
-"""
-ResponseCodeMiddleware 中间件
-用于处理HTTP响应状态码，记录统计信息并支持特殊状态码处理
-"""
-from crawlo.utils.log import get_logger
-
-
-class ResponseCodeMiddleware(object):
-    """
-    ResponseCodeMiddleware 中间件
-    用于处理HTTP响应状态码，记录统计信息并支持特殊状态码处理
-    
-    功能特性:
-    - 记录各种HTTP状态码的出现次数
-    - 支持特殊状态码的详细处理
-    - 提供详细的日志信息
-    - 支持状态码分类统计(2xx, 3xx, 4xx, 5xx)
-    """
-
-    def __init__(self, stats, log_level):
-        """
-        初始化中间件
-        
-        Args:
-            stats: 统计信息收集器
-            log_level: 日志级别
-        """
-        self.logger = get_logger(self.__class__.__name__, log_level)
-        self.stats = stats
-
-    @classmethod
-    def create_instance(cls, crawler):
-        """
-        创建中间件实例
-        
-        Args:
-            crawler: 爬虫实例
-            
-        Returns:
-            ResponseCodeMiddleware: 中间件实例
-        """
-        o = cls(stats=crawler.stats, log_level=crawler.settings.get('LOG_LEVEL'))
-        return o
-
-    def _get_status_category(self, status_code):
-        """
-        获取状态码分类
-        
-        Args:
-            status_code (int): HTTP状态码
-            
-        Returns:
-            str: 状态码分类 (2xx, 3xx, 4xx, 5xx, other)
-        """
-        if 200 <= status_code < 300:
-            return "2xx"
-        elif 300 <= status_code < 400:
-            return "3xx"
-        elif 400 <= status_code < 500:
-            return "4xx"
-        elif 500 <= status_code < 600:
-            return "5xx"
-        else:
-            return "other"
-
-    def _is_success_response(self, status_code):
-        """
-        判断是否为成功响应
-        
-        Args:
-            status_code (int): HTTP状态码
-            
-        Returns:
-            bool: 是否为成功响应
-        """
-        return 200 <= status_code < 300
-
-    def _is_redirect_response(self, status_code):
-        """
-        判断是否为重定向响应
-        
-        Args:
-            status_code (int): HTTP状态码
-            
-        Returns:
-            bool: 是否为重定向响应
-        """
-        return 300 <= status_code < 400
-
-    def _is_client_error(self, status_code):
-        """
-        判断是否为客户端错误
-        
-        Args:
-            status_code (int): HTTP状态码
-            
-        Returns:
-            bool: 是否为客户端错误
-        """
-        return 400 <= status_code < 500
-
-    def _is_server_error(self, status_code):
-        """
-        判断是否为服务器错误
-        
-        Args:
-            status_code (int): HTTP状态码
-            
-        Returns:
-            bool: 是否为服务器错误
-        """
-        return 500 <= status_code < 600
-
-    def process_response(self, request, response, spider):
-        """
-        处理响应，记录状态码统计信息
-        
-        Args:
-            request: 请求对象
-            response: 响应对象
-            spider: 爬虫实例
-            
-        Returns:
-            response: 响应对象
-        """
-        status_code = response.status_code
-        
-        # 记录具体状态码统计
-        self.stats.inc_value(f'response_status_code/count/{status_code}')
-        
-        # 记录状态码分类统计
-        category = self._get_status_category(status_code)
-        self.stats.inc_value(f'response_status_code/category/{category}')
-        
-        # 记录成功/失败统计
-        if self._is_success_response(status_code):
-            self.stats.inc_value('response_status_code/success_count')
-        elif self._is_client_error(status_code) or self._is_server_error(status_code):
-            self.stats.inc_value('response_status_code/error_count')
-            
-        # 记录响应大小统计
-        if hasattr(response, 'content_length') and response.content_length:
-            self.stats.inc_value('response_total_bytes', response.content_length)
-        
-        # 记录域名统计
-        try:
-            from urllib.parse import urlparse
-            parsed_url = urlparse(response.url)
-            domain = parsed_url.netloc
-            if domain:
-                self.stats.inc_value(f'response_status_code/domain/{domain}/count/{status_code}')
-                self.stats.inc_value(f'response_status_code/domain/{domain}/category/{category}')
-        except Exception:
-            self.stats.inc_value('response_status_code/domain/invalid_url/count/{status_code}')
-        
-        # 详细日志记录
-        self.logger.debug(
-            f'收到响应: {status_code} {response.url} '
-            f'(分类: {category}, 大小: {getattr(response, "content_length", "unknown")} bytes)'
-        )
-        
+#!/usr/bin/python
+# -*- coding:UTF-8 -*-
+"""
+ResponseCodeMiddleware 中间件
+用于处理HTTP响应状态码，记录统计信息并支持特殊状态码处理
+"""
+from crawlo.utils.log import get_logger
+
+
+class ResponseCodeMiddleware(object):
+    """
+    ResponseCodeMiddleware 中间件
+    用于处理HTTP响应状态码，记录统计信息并支持特殊状态码处理
+    
+    功能特性:
+    - 记录各种HTTP状态码的出现次数
+    - 支持特殊状态码的详细处理
+    - 提供详细的日志信息
+    - 支持状态码分类统计(2xx, 3xx, 4xx, 5xx)
+    """
+
+    def __init__(self, stats, log_level):
+        """
+        初始化中间件
+        
+        Args:
+            stats: 统计信息收集器
+            log_level: 日志级别
+        """
+        self.logger = get_logger(self.__class__.__name__, log_level)
+        self.stats = stats
+
+    @classmethod
+    def create_instance(cls, crawler):
+        """
+        创建中间件实例
+        
+        Args:
+            crawler: 爬虫实例
+            
+        Returns:
+            ResponseCodeMiddleware: 中间件实例
+        """
+        o = cls(stats=crawler.stats, log_level=crawler.settings.get('LOG_LEVEL'))
+        return o
+
+    def _get_status_category(self, status_code):
+        """
+        获取状态码分类
+        
+        Args:
+            status_code (int): HTTP状态码
+            
+        Returns:
+            str: 状态码分类 (2xx, 3xx, 4xx, 5xx, other)
+        """
+        if 200 <= status_code < 300:
+            return "2xx"
+        elif 300 <= status_code < 400:
+            return "3xx"
+        elif 400 <= status_code < 500:
+            return "4xx"
+        elif 500 <= status_code < 600:
+            return "5xx"
+        else:
+            return "other"
+
+    def _is_success_response(self, status_code):
+        """
+        判断是否为成功响应
+        
+        Args:
+            status_code (int): HTTP状态码
+            
+        Returns:
+            bool: 是否为成功响应
+        """
+        return 200 <= status_code < 300
+
+    def _is_redirect_response(self, status_code):
+        """
+        判断是否为重定向响应
+        
+        Args:
+            status_code (int): HTTP状态码
+            
+        Returns:
+            bool: 是否为重定向响应
+        """
+        return 300 <= status_code < 400
+
+    def _is_client_error(self, status_code):
+        """
+        判断是否为客户端错误
+        
+        Args:
+            status_code (int): HTTP状态码
+            
+        Returns:
+            bool: 是否为客户端错误
+        """
+        return 400 <= status_code < 500
+
+    def _is_server_error(self, status_code):
+        """
+        判断是否为服务器错误
+        
+        Args:
+            status_code (int): HTTP状态码
+            
+        Returns:
+            bool: 是否为服务器错误
+        """
+        return 500 <= status_code < 600
+
+    def process_response(self, request, response, spider):
+        """
+        处理响应，记录状态码统计信息
+        
+        Args:
+            request: 请求对象
+            response: 响应对象
+            spider: 爬虫实例
+            
+        Returns:
+            response: 响应对象
+        """
+        status_code = response.status_code
+        
+        # 记录具体状态码统计
+        self.stats.inc_value(f'response_status_code/count/{status_code}')
+        
+        # 记录状态码分类统计
+        category = self._get_status_category(status_code)
+        self.stats.inc_value(f'response_status_code/category/{category}')
+        
+        # 记录成功/失败统计
+        if self._is_success_response(status_code):
+            self.stats.inc_value('response_status_code/success_count')
+        elif self._is_client_error(status_code) or self._is_server_error(status_code):
+            self.stats.inc_value('response_status_code/error_count')
+            
+        # 记录响应大小统计
+        if hasattr(response, 'content_length') and response.content_length:
+            self.stats.inc_value('response_total_bytes', response.content_length)
+        
+        # 记录域名统计
+        try:
+            from urllib.parse import urlparse
+            parsed_url = urlparse(response.url)
+            domain = parsed_url.netloc
+            if domain:
+                self.stats.inc_value(f'response_status_code/domain/{domain}/count/{status_code}')
+                self.stats.inc_value(f'response_status_code/domain/{domain}/category/{category}')
+        except Exception:
+            self.stats.inc_value('response_status_code/domain/invalid_url/count/{status_code}')
+        
+        # 详细日志记录
+        self.logger.debug(
+            f'收到响应: {status_code} {response.url} '
+            f'(分类: {category}, 大小: {getattr(response, "content_length", "unknown")} bytes)'
+        )
+        
         return response