convert domainCheck to regular directory

domainCheck/app/utils/domain_utils.py

@@ -0,0 +1,195 @@
+# -*- coding: UTF-8 -*-
+'''
+@Project :domainScanDemo
+@File    :domain_utils.py
+@IDE     :PyCharm
+@Author  :梦伴
+@Date    :2026/4/9 0:05
+@explain : 域名工具类
+'''
+
+import re
+import tldextract
+from loguru import logger
+from app.utils.status_codes import DETECT_STATUS_LABELS, REGISTER_STATUS_LABELS, USE_STATUS_LABELS
+
+
+def normalize_domain(domain):
+    """
+    标准化域名
+    
+    :param domain: 域名
+    :return: str - 标准化后的域名
+    """
+    try:
+        # 转换为小写
+        domain = domain.lower()
+        
+        # 去除空格
+        domain = domain.strip()
+        
+        # 去除协议
+        domain = re.sub(r'^https?://', '', domain)
+        
+        # 去除路径和查询参数
+        domain = domain.split('/')[0]
+        domain = domain.split('?')[0]
+        
+        # 去除端口
+        domain = domain.split(':')[0]
+        
+        # 只保留主域
+        ext = tldextract.extract(domain)
+        if ext.domain and ext.suffix:
+            domain = f"{ext.domain}.{ext.suffix}"
+        
+        # 验证域名格式
+        if not is_valid_domain(domain):
+            return None
+        
+        return domain
+    except Exception as e:
+        logger.error(f"标准化域名出错: {e}")
+        return None
+
+
+def is_valid_domain(domain):
+    """
+    验证域名格式
+    
+    :param domain: 域名
+    :return: bool - 是否有效
+    """
+    try:
+        # 域名格式正则
+        pattern = r'^[a-zA-Z0-9][a-zA-Z0-9-]{0,61}[a-zA-Z0-9]\.[a-zA-Z]{2,}$'
+        return bool(re.match(pattern, domain))
+    except Exception as e:
+        logger.error(f"验证域名格式出错: {e}")
+        return False
+
+
+def extract_tld(domain):
+    """
+    提取顶级域名
+    
+    :param domain: 域名
+    :return: str - 顶级域名
+    """
+    try:
+        ext = tldextract.extract(domain)
+        return ext.suffix
+    except Exception as e:
+        logger.error(f"提取顶级域名出错: {e}")
+        return ''
+
+
+def extract_domain(domain):
+    """
+    提取主域名
+    
+    :param domain: 域名
+    :return: str - 主域名
+    """
+    try:
+        ext = tldextract.extract(domain)
+        if ext.domain and ext.suffix:
+            return f"{ext.domain}.{ext.suffix}"
+        return domain
+    except Exception as e:
+        logger.error(f"提取主域名出错: {e}")
+        return domain
+
+
+def is_com_or_net(domain):
+    """
+    检查是否为 .com 或 .net 域名
+    
+    :param domain: 域名
+    :return: bool - 是否为 .com 或 .net 域名
+    """
+    try:
+        tld = extract_tld(domain)
+        return tld in ['com', 'net']
+    except Exception as e:
+        logger.error(f"检查域名后缀出错: {e}")
+        return False
+
+
+def generate_domain_variants(domain):
+    """
+    生成域名变体
+    
+    :param domain: 域名
+    :return: list - 域名变体列表
+    """
+    try:
+        variants = []
+        
+        # 原始域名
+        variants.append(domain)
+        
+        # 添加 www
+        if not domain.startswith('www.'):
+            variants.append(f"www.{domain}")
+        
+        # 移除 www
+        if domain.startswith('www.'):
+            variants.append(domain[4:])
+        
+        return variants
+    except Exception as e:
+        logger.error(f"生成域名变体出错: {e}")
+        return [domain]
+
+
+def parse_domain_status(status_code):
+    """
+    解析域名状态码
+    
+    :param status_code: 状态码
+    :return: str - 状态描述
+    """
+    return REGISTER_STATUS_LABELS.get(status_code, '未知')
+
+
+def parse_use_status(status_code):
+    """
+    解析使用状态码
+    
+    :param status_code: 状态码
+    :return: str - 状态描述
+    """
+    return USE_STATUS_LABELS.get(status_code, '未知')
+
+
+def parse_detect_status(status_code):
+    """
+    解析检测状态码
+    
+    :param status_code: 状态码
+    :return: str - 状态描述
+    """
+    return DETECT_STATUS_LABELS.get(status_code, '未知')
+
+
+def parse_source_type(source_type):
+    """
+    解析来源类型
+    
+    :param source_type: 来源类型
+    :return: str - 来源描述
+    """
+    source_map = {
+        1: '聚名一口价',
+        2: '聚名过期删除',
+        3: 'zone file',
+        4: '搜索引擎采集',
+        5: '企业目录采集',
+        6: '手工录入',
+        7: 'TXT 导入',
+        8: '第三方接口',
+        9: '其它'
+    }
+    
+    return source_map.get(source_type, '未知')