From a14ed60b832466e9a8e7e2dafbd1b37b5a7a81d0 Mon Sep 17 00:00:00 2001
From: huangzhenpc <huangzhenpc@git586vip.cn>
Date: Sat, 1 Mar 2025 11:19:58 +0800
Subject: [PATCH] huanx

---
 app/utils.py | 21 +++++++--------------
 1 file changed, 7 insertions(+), 14 deletions(-)

diff --git a/app/utils.py b/app/utils.py
index bce71b7..fa78532 100644
--- a/app/utils.py
+++ b/app/utils.py
@@ -260,12 +260,10 @@ def extract_code_from_body(body):
 
 
 def add_allowed_domain(domain):
-    """添加允许的域名并记录创建时间"""
+    """添加允许的域名"""
     try:
-        timestamp = datetime.now().isoformat()
-        redis_client.rpush('allowed_domains_list', domain)  # 添加到列表
-        redis_client.hset(f'domain:{domain}', 'created_at', timestamp)  # 记录创建时间
-        logger.info(f'Added allowed domain: {domain} with timestamp: {timestamp}')
+        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}')
 
@@ -280,15 +278,10 @@ def remove_allowed_domain(domain):
 
 
 def get_allowed_domains():
-    """获取当前允许的域名及其创建时间"""
+    """获取当前允许的域名列表"""
     try:
-        domains = redis_client.lrange('allowed_domains_list', 0, -1)
-        domain_info = {}
-        for domain in domains:
-            domain = domain.decode()
-            created_at = redis_client.hget(f'domain:{domain}', 'created_at')
-            domain_info[domain] = created_at.decode() if created_at else None
-        return domain_info
+        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 {} 
\ No newline at end of file
+        return [] 
\ No newline at end of file