Update frp.py

frp.py

@@ -1,11 +1,34 @@
 import os
 import subprocess
 
+# 1. 安装 frpc
+def install_frpc():
+    print("正在安装 frpc...")
+
+    # 下载 frpc 二进制文件（更新为 Hugging Face 镜像）
+    try:
+        frpc_url = "https://huggingface.co/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)
+
+        # 使用 aria2 下载并解压 frpc
+        subprocess.run(["aria2c", "--dir", download_dir, frpc_url], check=True)
+
+        # 解压下载的 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)
+
+        # 移动 frpc 到合适的位置
+        subprocess.run(["mv", os.path.join(download_dir, "frp_0.61.0_linux_amd64", "frpc"), "/root/"], check=True)
+
+        print("frpc 安装完成。")
+    except subprocess.CalledProcessError as e:
+        print(f"安装 frpc 失败: {e}")
+        exit(1)
+
+# 2. 创建 frpc.toml 配置文件
 def create_frpc_toml():
-    # frpc.toml 文件路径
     toml_path = "/root/frpc.toml"
-    
-    # frpc.toml 配置内容
     toml_content = """
 [common]
 server_addr = 182.43.67.222
@@ -30,11 +53,9 @@ remote_port = 8188
 
     print(f"frpc.toml 配置文件已创建：{toml_path}")
 
+# 3. 创建启动脚本
 def create_frp_service_script():
-    # 脚本路径
     script_path = "/root/frp_service.sh"
-    
-    # 创建并写入启动脚本
     script_content = '''#!/bin/bash
 
 # 设置 frp 的路径和配置文件路径
@@ -60,11 +81,9 @@ done
     os.chmod(script_path, 0o755)
     print(f"脚本已创建并设置为可执行：{script_path}")
 
+# 4. 创建 systemd 服务文件
 def create_systemd_service():
-    # systemd 服务文件路径
     service_path = "/etc/systemd/system/frp.service"
-    
-    # 创建并写入 systemd 服务文件
     service_content = '''[Unit]
 Description=frp connection service
 After=network.target
@@ -88,23 +107,23 @@ WantedBy=multi-user.target
 
     print(f"systemd 服务文件已创建：{service_path}")
 
+# 5. 启动并启用服务
 def enable_and_start_service():
-    # 启用并启动服务
     try:
-        # 使用 sudo 来执行 systemctl 命令
         subprocess.run(["sudo", "systemctl", "daemon-reload"], check=True)
         subprocess.run(["sudo", "systemctl", "enable", "frp.service"], check=True)
         subprocess.run(["sudo", "systemctl", "start", "frp.service"], check=True)
-        print("服务已启用并启动。")
+        print("frp 服务已启用并启动。")
     except subprocess.CalledProcessError as e:
         print(f"服务启动失败: {e}")
 
+# 主函数
 def main():
-    create_frpc_toml()
-    create_frp_service_script()
-    create_systemd_service()
-    enable_and_start_service()
+    install_frpc()          # 安装 frpc
+    create_frpc_toml()      # 创建 frpc.toml 配置文件
+    create_frp_service_script()  # 创建启动脚本
+    create_systemd_service()     # 创建 systemd 服务
+    enable_and_start_service()   # 启动并启用服务
 
 if __name__ == "__main__":
-    main()
-
+    main()