服务脚本

1
import commands
2
import sys
3
import os
4

5
Error = "\033[1;31;40m"
6
Through = "\033[1;32;40m"
7
Tail = "\033[0m"
8

9

10
def fen(x):
11
    status, result = commands.getstatusoutput(
12
        'ps aux|grep fen.py|grep -v grep')
13
    if status == 0:
14
        if x == "k":
15
            return 1
16
        return Through + "运行中" + Tail
17
    else:
18
        if x== "g":
19
            return 1
20
        return Error + "未运行" + Tail
21

22

23
def xunlei(x):
24
    status, result = commands.getstatusoutput(
25
        'ps aux|grep xunlei|grep -v grep')
26
    if status == 0:
27
        if x == "k":
28
            return 1
29
        return Through + "运行中" + Tail
30
    else:
31
        if x== "g":
32
            return 1
33
        return Error + "未运行" + Tail
34

35
def frp(x):
36
    status, result = commands.getstatusoutput(
37
        'ps aux|grep frp|grep -v grep')
38
    if status == 0:
39
        if x == "k":
40
            return 1
41
        return Through + "运行中" + Tail
42
    else:
43
        if x== "g":
44
            return 1
45
        return Error + "未运行" + Tail
46

47
def state():
48
    f = open("/etc/startup-script/state.txt")
49
    for line in f:
50
        line = line.strip('\n')
51
        v = line.split(':')
52
        if v[0] == "fen_state":
53
            return fen_state(v[1])
54
    return Error + "Error" + Tail
55

56

57
def fen_state(stat):
58
    if stat == "1":
59
        return "风扇开启中\n"
60
    else:
61
        return "风扇已关闭\n"
62

63
def help():
64
    print("  _____      _     _   _   ")
65
    print(" |  __ \    | |   | | | |  ")
66
    print(" | |__) |___| |__ | |_| |_ ")
67
    print(" |  _  // _ \ '_ \| __| __|")
68
    print(" | | \ \  __/ | | | |_| |_ ")
69
    print(" |_|  \_\___|_| |_|\__|\__|")
70
    print
71
    print("-h | help    显示帮助")
72
    print("-f           开启温控风扇")
73
    print("-df          关闭温控风扇")
74
    print("-x           开启迅雷远程下载")
75
    print("-dx          关闭迅雷远程下载")
76
    print("-frp         开启frpc")
77
    print("-dfrp        关闭frpc")
78
    print("-ls          显示服务状态")
79

80
def show():
81
    print("cpu温控风扇：" + fen(0))
82
    print("迅雷远程下载：" + xunlei(0))
83
    print("frp:"+frp(0))
84
    print
85
    print("状态：" + state())
86

87
if __name__ == '__main__':
88
    arg=sys.argv
89
    k="k"
90
    g="g"
91
    try:
92
        if arg[1] == "help" and arg[1] == "-h":
93
            help()
94
        elif arg[1] == "-f":
95
            if fen(k) == 1:
96
                os.system('sudo python /etc/startup-script/fen.py &')
97
                print("开启成功")
98
            else:
99
                print("温控风扇已在运行状态")
100
        elif arg[1] == "-df":
101
            if fen(g) ==1:
102
                os.system('sudo killall python /etc/startup-script/fen.py')
103
                print("关闭成功")
104
            else:
105
                print("温控风扇未运行")
106
        elif arg[1] == "-x":
107
            if fen(k) == 1:
108
                os.system('sudo python /etc/startup-script/xunlei/setup.py')
109
                print("开启成功")
110
            else:
111
                print("迅雷远程下载已在运行状态")
112
        elif arg[1] == "-dx":
113
            if fen(g) ==1:
114
                os.system('sudo /etc/startup-script/xunlei/portal -s')
115
                print("关闭成功")
116
            else:
117
                print("迅雷远程下载未运行")
118
        elif arg[1]=="-frp":
119
            if frp(k) ==1:
120
                os.system('sudo screen -dmS frp /etc/startup-script/frp_0.20.0_linux_arm/frpc -c /etc/startup-script/frp_0.20.0_linux_arm/frpc.ini')
121
                print("开启成功")
122
            else:
123
                print("frp已在运行状态")
124
        elif arg[1]=="-dfrp":
125
            if frp(g)==1:
126
                os.system('sudo screen -d frp')
127
                print("关闭成功")
128
            else:
129
                print("frp未运行")
130
        elif arg[1] =="-ls":
131
            show()
132
        else:
133
            help()
134
    except:
135
        help()