Update Dockerfile

Dockerfile

@@ -8,8 +8,8 @@ RUN apk update && \
     apk add python3 py3-pip git && \
     pip3 install --break-system-packages --upgrade pip
 
-# 安装 Python 包
-RUN pip3 install --break-system-packages watchdog huggingface_hub aiohttp
+# 安装 Python 包，包括 pytz 用于处理时区
+RUN pip3 install --break-system-packages watchdog huggingface_hub aiohttp pytz
 
 # 预创建 Gitea 所需目录并设置权限
 RUN mkdir -p /data/gitea/conf /data/gitea/log /data/gitea/git /data/git /data/ssh && \
@@ -23,14 +23,33 @@ import shutil
 from huggingface_hub import snapshot_download
 import logging
 import base64
+from datetime import datetime
+import pytz
 
-# 配置日志
+# 配置日志，使用北京时间
+beijing_tz = pytz.timezone('Asia/Shanghai')
 logging.basicConfig(
     level=logging.INFO,
-    format='%(asctime)s - %(levelname)s - %(message)s'
+    format='%(asctime)s - %(levelname)s - %(message)s',
+    datefmt='%Y-%m-%d %H:%M:%S %Z'
 )
 logger = logging.getLogger(__name__)
 
+# 自定义日志格式化器以使用北京时间
+class BeijingTimeFormatter(logging.Formatter):
+    def formatTime(self, record, datefmt=None):
+        dt = datetime.fromtimestamp(record.created, tz=beijing_tz)
+        if datefmt:
+            return dt.strftime(datefmt)
+        return dt.strftime("%Y-%m-%d %H:%M:%S %Z")
+
+# 应用自定义格式化器
+for handler in logging.getLogger().handlers:
+    handler.setFormatter(BeijingTimeFormatter(
+        fmt='%(asctime)s - %(levelname)s - %(message)s',
+        datefmt='%Y-%m-%d %H:%M:%S %Z'
+    ))
+
 def pull_from_hf_hub(repo_id, data_directory="/data"):
     """从 Hugging Face Hub 数据集仓库拉取数据替换 /data 目录"""
     # 硬编码的 base64 编码的 HF_TOKEN
@@ -109,7 +128,7 @@ if __name__ == "__main__":
         exit(1)
 EOF
 
-# 复制上传脚本，移除检测到事件日志
+# 复制上传脚本，移除检测到事件日志，修改日志为北京时间，添加崩溃重启功能
 COPY --chown=git:git <<'EOF' /uploadhf.py
 import os
 import time
@@ -123,14 +142,33 @@ from watchdog.observers import Observer
 from watchdog.events import FileSystemEventHandler
 from huggingface_hub import HfApi
 import base64
+from datetime import datetime
+import pytz
 
-# 配置日志
+# 配置日志，使用北京时间
+beijing_tz = pytz.timezone('Asia/Shanghai')
 logging.basicConfig(
     level=logging.INFO,
-    format='%(asctime)s - %(levelname)s - %(message)s'
+    format='%(asctime)s - %(levelname)s - %(message)s',
+    datefmt='%Y-%m-%d %H:%M:%S %Z'
 )
 logger = logging.getLogger(__name__)
 
+# 自定义日志格式化器以使用北京时间
+class BeijingTimeFormatter(logging.Formatter):
+    def formatTime(self, record, datefmt=None):
+        dt = datetime.fromtimestamp(record.created, tz=beijing_tz)
+        if datefmt:
+            return dt.strftime(datefmt)
+        return dt.strftime("%Y-%m-%d %H:%M:%S %Z")
+
+# 应用自定义格式化器
+for handler in logging.getLogger().handlers:
+    handler.setFormatter(BeijingTimeFormatter(
+        fmt='%(asctime)s - %(levelname)s - %(message)s',
+        datefmt='%Y-%m-%d %H:%M:%S %Z'
+    ))
+
 class DataDirectoryHandler(FileSystemEventHandler):
     """处理 /data 目录文件变化的监控器"""
     
@@ -252,7 +290,7 @@ def start_gitea_server(port=7860):
     return gitea_thread
 
 async def main():
-    """主函数 - 启动 Gitea 和目录监控服务"""
+    """主函数 - 启动 Gitea 和目录监控服务，带崩溃重启功能"""
     # 配置参数
     CONFIG = {
         "data_directory": "/data",
@@ -261,44 +299,56 @@ async def main():
         "gitea_port": 7860
     }
     
-    logger.info("🚀 启动集成服务...")
+    max_retries = 5
+    retry_count = 0
     
-    try:
-        # 先运行 pullhf.py 拉取最新数据集
-        logger.info("运行 pullhf.py 拉取最新数据集")
-        pull_result = subprocess.run(["python3", "/pullhf.py"], check=True)
-        if pull_result.returncode != 0:
-            logger.error("❌ pullhf.py 执行失败，退出")
-            exit(1)
-        
-        # 启动 Gitea 服务器
-        gitea_thread = start_gitea_server(CONFIG["gitea_port"])
-        
-        # 启动目录监控服务
-        observer = start_directory_monitoring(
-            data_directory=CONFIG["data_directory"],
-            repo_id=CONFIG["repo_id"],
-            hf_token=CONFIG["hf_token"]
-        )
-        
-        logger.info("✅ 所有服务已启动完成！")
-        logger.info("📁 目录监控: /data → Hugging Face Hub")
-        logger.info("🌐 Gitea 服务: http://localhost:7860")
-        logger.info("🛑 按 Ctrl+C 停止所有服务")
-        
+    while retry_count < max_retries:
         try:
-            while True:
-                await asyncio.sleep(1)
-        except KeyboardInterrupt:
-            logger.info("正在停止服务...")
+            logger.info(f"🚀 启动集成服务 (尝试 {retry_count + 1}/{max_retries})...")
             
-    except Exception as e:
-        logger.error(f"❌ 启动服务时发生错误: {e}")
-    finally:
-        if 'observer' in locals():
-            observer.stop()
-            observer.join()
-        logger.info("所有服务已停止")
+            # 先运行 pullhf.py 拉取最新数据集
+            logger.info("运行 pullhf.py 拉取最新数据集")
+            pull_result = subprocess.run(["python3", "/pullhf.py"], check=True)
+            if pull_result.returncode != 0:
+                logger.error("❌ pullhf.py 执行失败，退出")
+                exit(1)
+            
+            # 启动 Gitea 服务器
+            gitea_thread = start_gitea_server(CONFIG["gitea_port"])
+            
+            # 启动目录监控服务
+            observer = start_directory_monitoring(
+                data_directory=CONFIG["data_directory"],
+                repo_id=CONFIG["repo_id"],
+                hf_token=CONFIG["hf_token"]
+            )
+            
+            logger.info("✅ 所有服务已启动完成！")
+            logger.info("📁 目录监控: /data → Hugging Face Hub")
+            logger.info("🌐 Gitea 服务: http://localhost:7860")
+            logger.info("🛑 按 Ctrl+C 停止所有服务")
+            
+            try:
+                while True:
+                    await asyncio.sleep(1)
+            except KeyboardInterrupt:
+                logger.info("正在停止服务...")
+                break
+            
+        except Exception as e:
+            retry_count += 1
+            logger.error(f"❌ 服务崩溃 (尝试 {retry_count}/{max_retries}): {e}")
+            if retry_count < max_retries:
+                logger.info(f"将在 5 秒后尝试重启...")
+                await asyncio.sleep(5)
+            else:
+                logger.error(f"❌ 达到最大重试次数 ({max_retries})，退出")
+                exit(1)
+        finally:
+            if 'observer' in locals():
+                observer.stop()
+                observer.join()
+            logger.info("所有服务已停止")
 
 if __name__ == "__main__":
     asyncio.run(main())
@@ -332,5 +382,4 @@ ENV GITEA__database__DB_TYPE=sqlite3 \
 USER git
 
 # 运行 Gitea，显式指定 Spaces 默认端口
-RUN cat /uploadhf.py
 CMD ["/set_gitea_env.sh", "python3", "/uploadhf.py"]