Update frp.py

frp.py

@@ -1,108 +1,110 @@
-import os
-import subprocess
-
-def create_frpc_toml():
-    # frpc.toml 文件路径
-    toml_path = "/root/frpc.toml"
-    
-    # frpc.toml 配置内容
-    toml_content = """
-[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
-local_port = 8188
-remote_port = 8188
-    """
-
-    # 写入 frpc.toml 文件
-    with open(toml_path, 'w') as file:
-        file.write(toml_content)
-
-    print(f"frpc.toml 配置文件已创建：{toml_path}")
-
-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 秒再重启
-done
-'''
-
-    # 写入脚本文件
-    with open(script_path, 'w') as file:
-        file.write(script_content)
-
-    # 确保脚本具有执行权限
-    os.chmod(script_path, 0o755)
-    print(f"脚本已创建并设置为可执行：{script_path}")
-
-def create_systemd_service():
-    # systemd 服务文件路径
-    service_path = "/etc/systemd/system/frp.service"
-    
-    # 创建并写入 systemd 服务文件
-    service_content = '''[Unit]
-Description=frp connection service
-After=network.target
-
-[Service]
-ExecStart=/root/frp_service.sh
-Restart=always
-User=root
-WorkingDirectory=/root
-StandardOutput=journal
-StandardError=journal
-SyslogIdentifier=frp
-
-[Install]
-WantedBy=multi-user.target
-'''
-
-    # 写入服务文件
-    with open(service_path, 'w') as file:
-        file.write(service_content)
-
-    print(f"systemd 服务文件已创建：{service_path}")
-
-def enable_and_start_service():
-    # 启用并启动服务
-    try:
-        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("服务已启用并启动。")
-    except subprocess.CalledProcessError as e:
-        print(f"服务启动失败: {e}")
-
-def main():
-    create_frpc_toml()
-    create_frp_service_script()
-    create_systemd_service()
-    enable_and_start_service()
-
-if __name__ == "__main__":
-    main()
+import os
+import subprocess
+
+def create_frpc_toml():
+    # frpc.toml 文件路径
+    toml_path = "/root/frpc.toml"
+    
+    # frpc.toml 配置内容
+    toml_content = """
+[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
+local_port = 8188
+remote_port = 8188
+    """
+
+    # 写入 frpc.toml 文件
+    with open(toml_path, 'w') as file:
+        file.write(toml_content)
+
+    print(f"frpc.toml 配置文件已创建：{toml_path}")
+
+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 秒再重启
+done
+'''
+
+    # 写入脚本文件
+    with open(script_path, 'w') as file:
+        file.write(script_content)
+
+    # 确保脚本具有执行权限
+    os.chmod(script_path, 0o755)
+    print(f"脚本已创建并设置为可执行：{script_path}")
+
+def create_systemd_service():
+    # systemd 服务文件路径
+    service_path = "/etc/systemd/system/frp.service"
+    
+    # 创建并写入 systemd 服务文件
+    service_content = '''[Unit]
+Description=frp connection service
+After=network.target
+
+[Service]
+ExecStart=/root/frp_service.sh
+Restart=always
+User=root
+WorkingDirectory=/root
+StandardOutput=journal
+StandardError=journal
+SyslogIdentifier=frp
+
+[Install]
+WantedBy=multi-user.target
+'''
+
+    # 写入服务文件
+    with open(service_path, 'w') as file:
+        file.write(service_content)
+
+    print(f"systemd 服务文件已创建：{service_path}")
+
+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("服务已启用并启动。")
+    except subprocess.CalledProcessError as e:
+        print(f"服务启动失败: {e}")
+
+def main():
+    create_frpc_toml()
+    create_frp_service_script()
+    create_systemd_service()
+    enable_and_start_service()
+
+if __name__ == "__main__":
+    main()
+