-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathbot.py
executable file
·154 lines (126 loc) · 5.83 KB
/
bot.py
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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
#!/usr/bin/python
import config
import telegram
import os
import subprocess
import sys
import shlex
import datetime
from subprocess import Popen, PIPE
from telegram.ext import CommandHandler
from imp import reload
#bot = telegram.Bot(token = config.token)
#Проверка бота
#print(bot.getMe())
from telegram.ext import Updater
updater = Updater(token=config.token)
dispatcher = updater.dispatcher
def run_command(command):
process = subprocess.Popen(shlex.split(command), stdout=subprocess.PIPE)
global textoutput
textoutput = ''
while True:
global output
output = process.stdout.readline()
output = output.decode('utf8')
if output == '' and process.poll() is not None:
break
if output:
print (output.strip())
textoutput = textoutput + '\n' + output.strip()
rc = process.poll()
return rc
def start(bot, update):
bot.sendMessage(chat_id=update.message.chat_id, text="Приветики, меня зовут Centa, я твой покорный бот, жду команды")
def help(bot, update):
reload(config)
bot.sendMessage(chat_id=update.message.chat_id, text='''список доступных команд:
/id - узнать свой Telegram id
/ts3status- состояние сервера TS3
/ts3restart - перезапуск сервера TS3
/ts3start - запуск сервера TS3
/ts3stop - остановка сервера TS3
/ts3update - обновление сервера TS3
/ts3backup - резервная копия сервера TS3
/df - информация о наличии места на жетском диске
''')
#функция команады id
def myid(bot, update):
userid = update.message.from_user.id
bot.sendMessage(chat_id=update.message.chat_id, text=userid)
#функция команады df
def df(bot, update):
reload(config)
user = str(update.message.from_user.id)
if user in config.admin: #если пользовательский id в списке admin то команда выполняется
run_command("df -h")
bot.sendMessage(chat_id=update.message.chat_id, text=textoutput)
#функция команады ts3status
def ts3status(bot, update):
reload(config)
user = str(update.message.from_user.id)
if user in config.admin: #если пользовательский id в списке admin то команда выполняется
ts3status_command = config.ts3dir + "ts3server.sh status"
run_command(ts3status_command)
bot.sendMessage(chat_id=update.message.chat_id, text=textoutput)
#функция команады ts3restart
def ts3restart(bot, update):
reload(config)
user = str(update.message.from_user.id)
if user in config.admin: #если пользовательский id в списке admin то команда выполняется
ts3restart_command = config.ts3dir + "ts3server.sh restart"
run_command(ts3restart_command)
bot.sendMessage(chat_id=update.message.chat_id, text=textoutput)
#функция команады ts3update
def ts3update(bot, update):
reload(config)
user = str(update.message.from_user.id)
if user in config.admin: #если пользовательский id в списке admin то команда выполняется
ts3update_command = config.ts3dir + "ts3server.sh update"
run_command(ts3update_command)
bot.sendMessage(chat_id=update.message.chat_id, text=textoutput)
#функция команады ts3backup
def ts3backup(bot, update):
reload(config)
user = str(update.message.from_user.id)
if user in config.admin: #если пользовательский id в списке admin то команда выполняется
ts3backup_command = config.ts3dir + "ts3server.sh backup"
run_command(ts3backup_command)
bot.sendMessage(chat_id=update.message.chat_id, text=textoutput)
#функция команады ts3stop
def ts3stop(bot, update):
reload(config)
user = str(update.message.from_user.id)
if user in config.admin: #если пользовательский id в списке admin то команда выполняется
ts3stop_command = config.ts3dir + "ts3server.sh stop"
run_command(ts3stop_command)
bot.sendMessage(chat_id=update.message.chat_id, text=textoutput)
#функция команады ts3start
def ts3start(bot, update):
reload(config)
user = str(update.message.from_user.id)
if user in config.admin: #если пользовательский id в списке admin то команда выполняется
ts3start_command = config.ts3dir + "ts3server.sh start"
run_command(ts3start_command)
bot.sendMessage(chat_id=update.message.chat_id, text=textoutput)
start_handler = CommandHandler('start', start)
dispatcher.add_handler(start_handler)
myid_handler = CommandHandler('id', myid)
dispatcher.add_handler(myid_handler)
help_handler = CommandHandler('help', help)
dispatcher.add_handler(help_handler)
df_handler = CommandHandler('df', df)
dispatcher.add_handler(df_handler)
ts3status_handler = CommandHandler('ts3status', ts3status)
dispatcher.add_handler(ts3status_handler)
ts3restart_handler = CommandHandler('ts3restart', ts3restart)
dispatcher.add_handler(ts3restart_handler)
ts3update_handler = CommandHandler('ts3update', ts3update)
dispatcher.add_handler(ts3update_handler)
ts3backup_handler = CommandHandler('ts3backup', ts3backup)
dispatcher.add_handler(ts3backup_handler)
ts3stop_handler = CommandHandler('ts3stop', ts3stop)
dispatcher.add_handler(ts3stop_handler)
ts3start_handler = CommandHandler('ts3start', ts3start)
dispatcher.add_handler(ts3start_handler)
updater.start_polling()