Add connection check && some little fixes

This commit is contained in:
trueold89 2023-11-03 01:38:12 +03:00
parent d951b22056
commit e0345df54e
Signed by: trueold89
GPG Key ID: C122E85DD49E6B30
5 changed files with 42 additions and 12 deletions

View File

@ -1,11 +1,11 @@
#!/usr/bin/python3
# -- coding: utf-8 --
import func, telebot, os, log
import func, telebot, os, log, sys
from db import PATH
from lang import LANG as msg
TOKEN = os.environ["TOKEN"]
TOKEN = os.environ.get('TOKEN','None')
bot = telebot.TeleBot(TOKEN)
folder_list = []
dir = None
@ -159,6 +159,20 @@ def unknown(message):
else:
bot.reply_to(message,str(msg.get('adeny')))
# func.qbt()
log.start()
bot.polling()
def run():
if os.path.exists(PATH) == False:
os.mkdir(PATH)
log.start()
try:
func.qbt()
except:
log.errqbt()
sys.exit(1)
try:
bot.polling()
except:
log.errtelebot()
sys.exit(1)
if __name__ == "__main__":
run()

View File

@ -1,7 +1,7 @@
#!/usr/bin/python3
# -- coding: utf-8 --
import db, os, log
import db, os, log, subprocess
from db import *
from lang import LANG as msg
@ -12,10 +12,12 @@ def qbt():
commands = [
f"qbt settings set url {url}",
f"qbt settings set username {username}",
f"echo {password} | qbt settings set password --no-warn"
f"echo {password} | qbt settings set password --no-warn",
f"qbt server info "
]
for command in commands:
os.system(f"bash -c '{command}'")
output = subprocess.run(command, shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
def u_auth(id,passwd):
list = []

View File

@ -27,12 +27,14 @@ RU = {
'adeny': 'Этот бот запривачен, гнида, блять',
# Logs
'l_create': "Log Файл '{}' создан",
'l_start': 'Бот готов к работе',
'l_start': 'Запуск бота...',
'l_auth': "Пользователь '{}' успешно авторизировался",
'l_add': "Пользователь '{}' добавил папку '{}' по пути '{}'",
'l_rm': "Пользователь '{}' удалил папку '{}'",
'l_file': "Пользователь '{}' добавил в очередь файл '{}'",
'l_magnet': "Пользователь '{}' добавил в очередь ссылку '{}'"
'l_magnet': "Пользователь '{}' добавил в очередь ссылку '{}'",
'l_errqbt': "Ошибка подключения к qBitTorrent",
'l_errtele': "Ошибка подключения к Telegram API, проверьте ваш токен",
}
# English
@ -57,12 +59,14 @@ ENG = {
'adeny': "You do not have access, first authorize '/login <password>'",
# Logs
'l_create': "Log File '{}' created",
'l_start': 'Bot is up and running',
'l_start': 'Start bot polling...',
'l_auth': "User '{}' successfully authorized",
'l_add': "User '{}' added a folder '{}' with the path '{}'",
'l_rm': "User '{}' deleted '{}' folder",
'l_file': "User '{}' added file '{}' to the queue",
'l_magnet': "User '{}' added the link '{}' to the queue"
'l_magnet': "User '{}' added the link '{}' to the queue",
'l_errqbt': "Error connecting to qBitTorrent",
'l_errtele': "Error connecting to Telegram API, check your token"
}
for i in langs:

View File

@ -53,3 +53,13 @@ def addmagnet(id,link):
log = DEFAULT + str(msg.get('l_magnet').format(id,link))
file(log)
print(log)
def errqbt():
log = DEFAULT + str(msg.get('l_errqbt'))
file(log)
print(log)
def errtelebot():
log = DEFAULT + str(msg.get('l_errtele'))
file(log)
print(log)

View File

@ -12,4 +12,4 @@ services:
QURL: "<http://<YOUR_QBIT_SERVER_IP_HERE>:<PORT>"
QUSER: "<YOUR_QBIT_USERNAME>"
QPASS: "<YOUR_QBIT_PASSWORD>"
LANG: "RU"
LANG: "<YOUR_LANG>"