4
1
Fork 1

Compare commits

...

No commits in common. "4f7b6de5a659795630de36dfbcc4b052b386289f" and "main" have entirely different histories.

17 changed files with 613 additions and 61 deletions

4
.gitignore vendored
View File

@ -1,4 +0,0 @@
python_venv/
.idea
__pycache__
probe.py

1
Bot.py
View File

@ -1 +0,0 @@
# -*- coding: utf-8 -*-

View File

@ -1,54 +0,0 @@
# -*- coding: utf-8 -*-
from requests import get as req_get
from requests import Response
from abc import ABCMeta, abstractmethod
class DirectoryGetter(object):
__metaclass__ = ABCMeta
@abstractmethod
def getDirs(self) -> dict:
"""
Returns a list key:value - Directory name : directory path
:return: Dictionary - Directory name : directory path
"""
@abstractmethod
def CheckAvailability(self) -> bool:
"""
Checks if the directory parsing method is available
:return: boolean check result
"""
class Jellyfin(DirectoryGetter):
def __init__(self, url: str, api_key: str) -> None:
self.url = url
self.api_key = api_key
def CheckAvailability(self) -> bool:
if self.__get("Library/VirtualFolders").status_code == 200:
return True
else:
raise Exception("Error connecting to JellyfinAPI")
def __get(self, api_path: str) -> Response:
request = req_get(f"{self.url}/{api_path}?api_key={self.api_key}")
status = request.status_code
match status:
case 200:
return request
case 401:
raise Exception("Error 401: Authorization error, check api key")
case 403:
raise Exception("Error 403: Forbidden")
case 404:
raise Exception("Error 404: Not found")
def getDirs(self) -> dict:
folders_list = self.__get("Library/VirtualFolders").json()
out = {}
for folder in folders_list:
out[folder["Name"]] = folder["Locations"][0]
return out

10
Dockerfile Normal file
View File

@ -0,0 +1,10 @@
FROM alpine:latest
COPY bot /opt/bot
RUN apk update && apk add tzdata bash python3 py-pip wget icu-libs krb5-libs libgcc libintl libssl1.1 libstdc++ zlib && pip install telebot
RUN wget https://github.com/fedarovich/qbittorrent-cli/releases/download/v1.7.22315.1/qbt-linux-alpine-x64-1.7.22315.1.tar.gz && \
mkdir /opt/qbt && \
tar -zxf qbt-linux-alpine-x64-1.7.22315.1.tar.gz -C /opt/qbt && \
chmod a+x /opt/qbt/* && \
ln -sf /opt/qbt/qbt /bin/qbt && ln -sf /opt/bot/bot.py /bin/bot
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
ENTRYPOINT ["/bin/bot"]

View File

@ -1 +0,0 @@
# -*- coding: utf-8 -*-

132
Readme.md Normal file
View File

@ -0,0 +1,132 @@
# qBitDownload Bot
![](https://cloud.orudo.ru/apps/files_sharing/publicpreview/pgxm2mKT5KHEHFE?file=/&fileId=23795&x=1920&y=1200&a=true&etag=430e9d84364f13b79e42991fede6609a)
## Telegram bot designed to remotely add downloads to the queue on qBitTorrent server
| [**git.orudo.ru**](https://git.orudo.ru/OrudoCA/qBitDownload-Bot) | [**GitHub**](https://github.com/OrudoCA/qBitDownload-Bot) | [**DockerHub**](https://hub.docker.com/r/orudoca/qbitdownload-bot) |
| ---------------- | ---------- | ------------- |
| [![](https://cloud.orudo.ru/apps/files_sharing/publicpreview/AmggNTQWgR6KkyB?file=/&fileId=23836&x=1920&y=1200&a=true&etag=0ef9694cea6e4d85c05aef9be72b927a)](https://git.orudo.ru/OrudoCA/qBitDownload-Bot) | [![](https://cloud.orudo.ru/apps/files_sharing/publicpreview/ip5qtGcwKHMPMAG?file=/&fileId=23819&x=1920&y=1200&a=true&etag=c540068d990ac47217a31f7450afc0ee)](https://github.com/OrudoCA/qBitDownload-Bot) |[![](https://cloud.orudo.ru/apps/files_sharing/publicpreview/7AEeEAzHYikFd5B?file=/&fileId=23806&x=1920&y=1200&a=true&etag=59894ecdfa7aaa6fb832cc4bf99c418d)](https://hub.docker.com/r/orudoca/qbitdownload-bot) |
### Bot requires [**qBitTorrent**](https://www.qbittorrent.org/) server
---
### Current features:
---
- **Authorization by password**
- **Adding downloads to the queue via .torrent files / Magnet-links**
- **Add/Delete download directories**
- **Multiple language support**
---
### Image uses ["fedarovich/qbittorrent-cli"](https://github.com/fedarovich/qbittorrent-cli)
---
[![](https://cloud.orudo.ru/apps/files_sharing/publicpreview/rRcdSnCEaA85tWf?file=/&fileId=23784&x=1920&y=1200&a=true&etag=32928842bc4e76adaba194cdd9ec1351)](https://hub.docker.com/r/orudoca/qbitdownload-bot)
## Deploy with Docker
#### 1. Build image or clone it from [Dockerhub](https://hub.docker.com/r/orudoca/qbitdownload-bot)
**Clone a repository and go to its directory**
```bash
git clone https://git.orudo.ru/OrudoCA/qBitDownload-Bot.git && cd qBitDownload-Bot
```
**Build image**
```bash
docker build -t <IMAGE_NAME> .
```
#### 2. Deploy via docker-cli or docker-compose
**Docker-cli**
```bash
docker run \
--name qbitdl_bot \
--restart=unless-stopped \
-v /path/to/config:/etc/dbot \
-v /path/to/media:/path/to/media \
-e TOKEN="<YOUR_BOT_TOKEN_HERE>" \
-e PASS="change_me" \
-e QURL="http://<YOUR_QBIT_SERVER_IP_HERE>:<PORT>" \
-e QUSER="<YOUR_QBIT_USERNAME>" \
-e QPASS="<YOUR_QBIT_PASSWORD>" \
-e LANG="YOUR_LANG" \
-e TZ="Europe/Moscow" \
-d -it your_image_here
```
##### or
**docker-compose**
```yml
services:
qbitdl_bot:
image: <YOUR_IMAGE_HERE>
tty: true
container_name: qbitdl_bot
volumes:
- /path/to/config:/etc/bot
- /path/to/data/:/path/to/data
restart: 'unless-stopped'
environment:
TOKEN: "<YOUR_BOT_TOKEN_HERE>"
PASS: "change_me"
QURL: "http://<YOUR_QBIT_SERVER_IP_HERE>:<PORT>"
QUSER: "<YOUR_QBIT_USERNAME>"
QPASS: "<YOUR_QBIT_PASSWORD>"
LANG: "<YOUR_LANG>"
TZ: "Europe/Moscow"
```
```bash
docker compose up -d
```
---
![](https://cloud.orudo.ru/apps/files_sharing/publicpreview/ffSABnXQ3cQrLZG?file=/&fileId=23851&x=1920&y=1200&a=true&etag=d2d4704b2ab90afe5edee647a19a5540)
## Run natively:
### 1. Install deps:
- python3
- py3-pip
- [qbittorrent-cli](https://github.com/fedarovich/qbittorrent-cli)
### 2. Install TeleBot lib:
```bash
pip install telebot
```
### 3. Set system ENV
```bash
# Linux
export TOKEN="YOUR_BOT_TOKEN"
export PASS="change_me"
export QURL="http://<YOUR_QBIT_SERVER_IP_HERE>:<PORT>"
export QUSER="<YOUR_QBIT_USERNAME>"
export QPASS="<YOUR_QBIT_PASSWORD>"
export LANG="<YOUR_LANG>"
# Windows PS
set TOKEN="YOUR_BOT_TOKEN"
set PASS="change_me"
set QURL="http://<YOUR_QBIT_SERVER_IP_HERE>:<PORT>"
set QUSER="<YOUR_QBIT_USERNAME>"
set QPASS="<YOUR_QBIT_PASSWORD>"
set LANG="<YOUR_LANG>"
```
#### 3.1 On windows change PATH var in `db.py`
### 4. Run
```bash
python3 bot.py
```
---
### Available languages:
##### - **"ENG"** - English
##### - **"RU"** - Russian

View File

@ -1 +0,0 @@
# -*- coding: utf-8 -*-

View File

183
bot/bot.py Executable file
View File

@ -0,0 +1,183 @@
#!/usr/bin/python3
# -- coding: utf-8 --
import func, telebot, os, log, sys
from db import PATH
from lang import LANG as msg
TOKEN = os.environ.get('TOKEN','None')
bot = telebot.TeleBot(TOKEN)
folder_list = []
dir = None
# Start
@bot.message_handler(commands=['start'])
def welcome(message):
id = message.from_user.id
if func.auth_check(id):
bot.reply_to(message,str(msg.get('type')),reply_markup=home())
else:
bot.reply_to(message,str(msg.get('adeny')))
# Keyboard: Homepage
def home():
keyboard = telebot.types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True)
file = telebot.types.KeyboardButton(str(msg.get('file')))
magnet = telebot.types.KeyboardButton(str(msg.get('magnet')))
keyboard.add(file,magnet)
return keyboard
# Login
@bot.message_handler(commands=['login'])
def login(message):
id = message.from_user.id
name = message.from_user.first_name
passwd = message.text.replace('/login ', '')
f = str(func.u_auth(name,id,passwd))
if f == str(msg.get('sucauth')) or f == str(msg.get('alauth')):
bot.reply_to(message,f,reply_markup=home())
else:
bot.reply_to(message,f)
# Folders
def folder_menu():
folders = func.dirlist()
if len(folders) == 0:
return None
else:
keyboard = telebot.types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True)
i = 0
global folder_list
folder_list = []
lst = []
for folder in folders.keys():
globals()[f'var_{i}'] = telebot.types.KeyboardButton(f'{folder}')
lst.append(f'var_{i}')
folder_list.append(folder)
i += 1
for var in lst:
keyboard.add(globals()[var])
return keyboard
# Folder add
@bot.message_handler(commands=['add'])
def add(message):
id = message.from_user.id
name = message.from_user.first_name
txt = message.text.split(' ', 2)
if len(txt) == 3:
key = txt[1]
path = txt[2]
f = str(func.add_dir(name,id,key,path))
else:
f = str(msg.get('aerr'))
bot.reply_to(message,f,reply_markup=home())
# Folder del
@bot.message_handler(commands=['del'])
def rm(message):
id = message.from_user.id
name = message.from_user.first_name
folder = message.text.replace('/del ', '')
f = func.del_dir(name,id,folder)
bot.reply_to(message,str(f),reply_markup=home())
# Magnet
@bot.message_handler(func=lambda message: message.text == str(msg.get('magnet')))
def magnet(message):
id = message.from_user.id
if func.auth_check(id):
global type
type = 'magnet'
f = folder_menu()
if f == None:
bot.reply_to(message,str(msg.get('cff')))
else:
bot.reply_to(message,str(msg.get('chf')),reply_markup=f)
else:
bot.reply_to(message,str(msg.get('adeny')))
# File
@bot.message_handler(func=lambda message: message.text == str(msg.get('file')))
def file(message):
id = message.from_user.id
if func.auth_check(id):
global type
type = 'file'
f = folder_menu()
if f == None:
bot.reply_to(message,str(msg.get('cff')))
else:
bot.reply_to(message,str(msg.get('chf')),reply_markup=f)
else:
bot.reply_to(message,str(msg.get('adeny')))
# File download
@bot.message_handler(content_types=['document'])
def download(message):
id = message.from_user.id
name = message.from_user.first_name
if func.auth_check(id):
global type, dir, folder_list
if dir != None and type == 'file':
if message.document.file_name.lower().endswith('.torrent'):
file_info = bot.get_file(message.document.file_id)
file_path = file_info.file_path
file = bot.download_file(file_path)
file_name = os.path.join(PATH, message.document.file_name)
with open(file_name, 'wb') as dl:
dl.write(file)
f = str(func.file(name,id,file_name,dir))
dir, type, folder_list = None,None,[]
bot.reply_to(message,f)
else:
bot.reply_to(message,str(msg.get('ntorr')))
bot.reply_to(message,str(msg.get('type')),reply_markup=home())
else:
bot.reply_to(message,str(msg.get('adeny')))
# Dir choose
def dirchoose(message):
global dir
dir = message.text
if type == 'magnet':
bot.reply_to(message,str(msg.get('sendm')))
if type == 'file':
bot.reply_to(message,str(msg.get('sendf')))
# Unknown message
@bot.message_handler(func=lambda message: True)
def unknown(message):
global type, dir, folder_list
id = message.from_user.id
name = message.from_user.first_name
if func.auth_check(id):
txt = message.text
if txt in folder_list:
dirchoose(message)
return None
if dir != None and type == 'magnet':
f = str(func.magnet(name,id,txt,dir))
dir, type, folder_list = None,None,[]
bot.reply_to(message,f)
bot.reply_to(message,str(msg.get('type')),reply_markup=home())
else:
bot.reply_to(message,str(msg.get('adeny')))
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()

29
bot/db.py Normal file
View File

@ -0,0 +1,29 @@
#!/usr/bin/python3
# -- coding: utf-8 --
import pickle, os
PATH = "/etc/bot/"
AUTH_FILE= "auth.pkl"
DIR_FILE = "dir.pkl"
def check(type,FILE):
if type == 'dir':
if os.path.exists(PATH) == False:
os.mkdir(PATH)
return True
elif type == 'obj':
if os.path.exists(f'{PATH}{FILE}'):
return True
else:
return False
def write(obj,FILE):
if check('dir',None):
with open(f'{PATH}{FILE}',"wb") as file:
pickle.dump(obj,file)
def read(FILE):
with open(f'{PATH}{FILE}',"rb") as file:
obj = pickle.load(file)
return obj

103
bot/func.py Normal file
View File

@ -0,0 +1,103 @@
#!/usr/bin/python3
# -- coding: utf-8 --
import db, os, log, subprocess
from db import *
from lang import LANG as msg
def qbt():
url = os.environ['QURL']
username = os.environ['QUSER']
password = os.environ['QPASS']
commands = [
f"qbt settings set url {url}",
f"qbt settings set username {username}",
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(name,id,passwd):
list = []
if db.check('obj',AUTH_FILE):
list = db.read(AUTH_FILE)
if id in list:
return msg.get('alauth')
else:
if passwd == os.environ['PASS']:
list.append(id)
db.write(list,AUTH_FILE)
log.auth(name,id)
return msg.get('sucauth')
else:
return msg.get('wrauth')
def auth_check(id):
if db.check('obj',AUTH_FILE):
list = db.read(AUTH_FILE)
else:
list = []
if id in list:
return True
def add_dir(name,id,dir,path):
if auth_check(id):
if os.path.exists(path) == False:
return str(msg.get('pne')).format(path)
if db.check('obj',DIR_FILE):
dict = db.read(DIR_FILE)
else:
dict = {}
dict.setdefault(dir,path)
db.write(dict,DIR_FILE)
log.add(name,id,dir,path)
return str(msg.get('fsa')).format(dir)
else:
return msg.get('adeny')
def del_dir(name,id,dir):
if auth_check(id):
if db.check('obj',DIR_FILE):
dict = db.read(DIR_FILE)
else:
dict = {}
if dir in dict:
del dict[dir]
db.write(dict,DIR_FILE)
log.rm(name,id,dir)
return str(msg.get('frm')).format(dir)
else:
return str(msg.get('fne')).format(dir)
else:
return msg.get('adeny')
def magnet(name,id,link,dir):
if auth_check(id):
dict = db.read(DIR_FILE)
path = dict[dir]
command = f'''qbt torrent add url "{link}" -f "{path}"'''
os.system(f"bash -c '{command}'")
log.addmagnet(name,id,link)
return msg.get('add')
else:
return msg.get('adeny')
def file(name,id,file,dir):
if auth_check(id):
dict = db.read(DIR_FILE)
path = dict[dir]
command = f'''qbt torrent add file "{file}" -f {path}'''
os.system(f"bash -c '{command}'")
os.remove(file)
log.addfile(name,id,file)
return msg.get('add')
else:
return msg.get('adeny')
def dirlist():
dirs = {}
if db.check('obj',DIR_FILE):
dirs = db.read(DIR_FILE)
return dirs

74
bot/lang.py Normal file
View File

@ -0,0 +1,74 @@
#!/usr/bin/python3
# -- coding: utf-8 --
import os
langs = ['ENG','RU']
# Russian
RU = {
'alauth': 'Вы уже авторизированны',
'sucauth': 'Вы успешно авторизировались',
'wrauth': 'Неверный пароль',
'pne': "Директории '{}' не сушествует на сервере",
'fsa': "Папка '{}' успешно добавлена",
'frm': "Папка '{}' успешно удалена",
'fne': "Папки '{}' не существует",
'add': 'Torrent добавлен в очередь',
'type': 'Выберите тип загрузки:',
'magnet': 'Magnet-ссылка',
'file': 'Файл',
'aerr': 'Неверные аргументы',
'cff': 'Папок не обнаруженно, воспользуйтесь коммандой /add',
'chf': 'Выберите папку:',
'ntorr': 'Неверное расширение файла',
'sendm': 'Отправте Magnet-ссылку',
'sendf': 'Отправте .torrent файл',
'adeny': 'Этот бот запривачен, гнида, блять',
# Logs
'l_create': "Log Файл '{}' создан",
'l_start': 'Запуск бота...',
'l_auth': "Пользователь '{} ({})' успешно авторизировался",
'l_add': "Пользователь '{} ({})' добавил папку '{}' по пути '{}'",
'l_rm': "Пользователь '{} ({})' удалил папку '{}'",
'l_file': "Пользователь '{} ({})' добавил в очередь файл '{}'",
'l_magnet': "Пользователь '{} ({})' добавил в очередь ссылку '{}'",
'l_errqbt': "Ошибка подключения к qBitTorrent",
'l_errtele': "Ошибка подключения к Telegram API, проверьте ваш токен",
}
# English
ENG = {
'alauth': 'You are already authorized',
'sucauth': 'You have successfully logged in',
'wrauth': 'Wrong password',
'pne': "The '{}' directory does not exist on the server",
'fsa': "The '{}' folder has been successfully added",
'frm': "The '{}' folder has been successfully deleted",
'fne': "The '{}' folder does not exist",
'add': 'Torrent has been added to the queue',
'type': 'Select the download type:',
'magnet': 'Magnet',
'file': 'File',
'aerr': 'Wrong arguments',
'cff': 'No folders found, use the /add command',
'chf': 'Select folder:',
'ntorr': 'Incorrect file extension',
'sendm': 'Send Magnet link',
'sendf': 'Send .torrent file',
'adeny': "You do not have access, first authorize '/login <password>'",
# Logs
'l_create': "Log File '{}' created",
'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_errqbt': "Error connecting to qBitTorrent",
'l_errtele': "Error connecting to Telegram API, check your token"
}
for i in langs:
if i == os.environ.get('LANG','ENG'):
LANG = globals()[i]

65
bot/log.py Normal file
View File

@ -0,0 +1,65 @@
#!/usr/bin/python3
# -- coding: utf-8 --
import os, uuid
from datetime import datetime
from lang import LANG as msg
from db import PATH
def dt():
date = datetime.now().date()
time = datetime.now().time()
str = f'{date} | {time.strftime("%H:%M:%S")}'
return str
DEFAULT = ['{} LOG: ','{} ERROR: ']
ID = str(uuid.uuid1())[0:7]
FILE = f'{ID}.txt'
def file(log):
if os.path.exists(f'{PATH}logs') == False:
os.mkdir(f'{PATH}logs')
with open(f'{PATH}logs/{FILE}','a') as logfile:
logfile.write(f'{log}\n')
logfile.close()
def start():
log1 = DEFAULT[0].format(dt()) + str(msg.get('l_create').format(FILE))
log2 = DEFAULT[0].format(dt()) + str(msg.get('l_start'))
file(log2)
print(f'{log1}\n{log2}')
def auth(name,id):
log = DEFAULT[0].format(dt()) + str(msg.get('l_auth').format(name,id))
file(log)
print(log)
def add(name,id,folder,path):
log = DEFAULT[0].format(dt()) + str(msg.get('l_add').format(name,id,folder,path))
file(log)
print(log)
def rm(name,id,folder):
log = DEFAULT[0].format(dt()) + str(msg.get('l_rm').format(name,id,folder))
file(log)
print(log)
def addfile(name,id,filename):
log = DEFAULT[0].format(dt()) + str(msg.get('l_file').format(name,id,filename[9:]))
file(log)
print(log)
def addmagnet(name,id,link):
log = DEFAULT[0].format(dt()) + str(msg.get('l_magnet').format(name,id,link))
file(log)
print(log)
def errqbt():
log = DEFAULT[1].format(dt()) + str(msg.get('l_errqbt'))
file(log)
print(log)
def errtelebot():
log = DEFAULT[1].format(dt()) + str(msg.get('l_errtele'))
file(log)
print(log)

17
docker-compose.yml Normal file
View File

@ -0,0 +1,17 @@
services:
qbitdl_bot:
image: <YOUR_IMAGE_HERE>
tty: true
container_name: qbitdl_bot
volumes:
- /path/to/config:/etc/bot
- /path/to/data/:/path/to/data
restart: 'unless-stopped'
environment:
TOKEN: "<YOUR_BOT_TOKEN_HERE>"
PASS: "change_me"
QURL: "http://<YOUR_QBIT_SERVER_IP_HERE>:<PORT>"
QUSER: "<YOUR_QBIT_USERNAME>"
QPASS: "<YOUR_QBIT_PASSWORD>"
LANG: "<YOUR_LANG>"
TZ: "Europe/Moscow"

View File

View File