Update frp.py

frp.py

@@ -5,23 +5,29 @@ import subprocess
 def install_frpc():
     print("正在安装 frpc...")
 
-    # 下载 frpc 二进制文件（更新为 Hugging Face 镜像）
-    try:
-        frpc_url = "https://hf-mirror.com/HOPStudio/modelscope/resolve/main/frp_0.61.0_linux_amd64.tar.gz?download=true"
-        download_dir = "/root/frp"
-        os.makedirs(download_dir, exist_ok=True)
+    # 本地解压路径
+    local_path = "/root/modelscope"
+    frpc_tar_gz = os.path.join(local_path, "frp_0.61.0_linux_amd64.tar.gz")
 
-        # 使用 aria2 下载并解压 frpc
-        subprocess.run(["aria2c", "--dir", download_dir, frpc_url], check=True)
+    try:
+        # 确保文件存在
+        if not os.path.exists(frpc_tar_gz):
+            print(f"错误：文件 {frpc_tar_gz} 不存在。")
+            exit(1)
 
         # 解压下载的 tar.gz 文件
-        subprocess.run(["tar", "-xvzf", os.path.join(download_dir, "frp_0.61.0_linux_amd64.tar.gz"), "-C", download_dir], check=True)
-        subprocess.run(["rm", os.path.join(download_dir, "frp_0.61.0_linux_amd64.tar.gz")], check=True)
+        subprocess.run(["tar", "-xvzf", frpc_tar_gz, "-C", local_path], check=True)
+        print("frpc 解压完成。")
+
+        # 移除 tar.gz 文件
+        os.remove(frpc_tar_gz)
+        print("已删除 tar.gz 文件。")
 
-        # 移动 frpc 到合适的位置
-        subprocess.run(["mv", os.path.join(download_dir, "frp_0.61.0_linux_amd64", "frpc"), "/root/"], check=True)
+        # 移动 frpc 到 /root 目录
+        frpc_extracted_dir = os.path.join(local_path, "frp_0.61.0_linux_amd64")
+        subprocess.run(["mv", os.path.join(frpc_extracted_dir, "frpc"), "/root/"], check=True)
+        print("frpc 已成功移动到 /root 目录。")
 
-        print("frpc 安装完成。")
     except subprocess.CalledProcessError as e:
         print(f"安装 frpc 失败: {e}")
         exit(1)
@@ -33,13 +39,11 @@ def create_frpc_toml():
 [common]
 server_addr = 182.43.67.222
 server_port = 7000
-
 [jl]
 type = tcp
 local_ip = 127.0.0.1
 local_port = 8888
 remote_port = 8888
-
 [comfyui]
 type = tcp
 local_ip = 127.0.0.1
@@ -57,16 +61,13 @@ remote_port = 8188
 def create_frp_service_script():
     script_path = "/root/frp_service.sh"
     script_content = '''#!/bin/bash
-
 # 设置 frp 的路径和配置文件路径
 FRP_PATH="/root/frpc"
 FRP_CONFIG="/root/frpc.toml"  # 使用新的 TOML 配置文件
-
 # 启动 frp 并保持运行
 while true; do
     # 启动 frp
     $FRP_PATH -c $FRP_CONFIG
-
     # 如果 frp 崩溃或关闭，则重新启动
     echo "frp 连接已断开，正在重新启动..."
     sleep 1  # 等待 1 秒再重启
@@ -87,7 +88,6 @@ def create_systemd_service():
     service_content = '''[Unit]
 Description=frp connection service
 After=network.target
-
 [Service]
 ExecStart=/root/frp_service.sh
 Restart=always
@@ -96,7 +96,6 @@ WorkingDirectory=/root
 StandardOutput=journal
 StandardError=journal
 SyslogIdentifier=frp
-
 [Install]
 WantedBy=multi-user.target
 '''
@@ -126,4 +125,4 @@ def main():
     enable_and_start_service()   # 启动并启用服务
 
 if __name__ == "__main__":
-    main()
+    main()