testapiaddyuming

app/utils.py

@@ -258,4 +258,32 @@ def extract_code_from_body(body):
     """从邮件正文中提取验证码"""
     import re
     match = re.search(r'\b(\d{6})\b', body)
-    return match.group(1) if match else None 
+    return match.group(1) if match else None
+
+
+def add_allowed_domain(domain):
+    """添加允许的域名"""
+    try:
+        redis_client.sadd('allowed_domains', domain)
+        logger.info(f'Added allowed domain: {domain}')
+    except Exception as e:
+        logger.error(f'Error adding allowed domain: {e}')
+
+
+def remove_allowed_domain(domain):
+    """删除允许的域名"""
+    try:
+        redis_client.srem('allowed_domains', domain)
+        logger.info(f'Removed allowed domain: {domain}')
+    except Exception as e:
+        logger.error(f'Error removing allowed domain: {e}')
+
+
+def get_allowed_domains():
+    """获取当前允许的域名列表"""
+    try:
+        domains = redis_client.smembers('allowed_domains')
+        return [domain.decode() for domain in domains]
+    except Exception as e:
+        logger.error(f'Error fetching allowed domains: {e}')
+        return []