Compare commits
No commits in common. "main" and "e8acbd89c763f4702535f6333e5a26db3b50c6ff" have entirely different histories.
main
...
e8acbd89c7
|
|
@ -0,0 +1,5 @@
|
||||||
|
venv
|
||||||
|
probe.*
|
||||||
|
__pycache__
|
||||||
|
*.egg-info
|
||||||
|
dist
|
||||||
10
Dockerfile
10
Dockerfile
|
|
@ -1,10 +0,0 @@
|
||||||
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"]
|
|
||||||
|
|
@ -0,0 +1,5 @@
|
||||||
|
# Torrent Uploader Bot
|
||||||
|
|
||||||
|
## A simple Telegram bot that will allow you to upload torrent files / magnet links to a remote Torrent server (qBitTorrent, Transmission, etc).
|
||||||
|
|
||||||
|
***
|
||||||
132
Readme.md
132
Readme.md
|
|
@ -1,132 +0,0 @@
|
||||||
# qBitDownload Bot
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
## 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://git.orudo.ru/OrudoCA/qBitDownload-Bot) | [](https://github.com/OrudoCA/qBitDownload-Bot) |[](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://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
|
|
||||||
```
|
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
## 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
|
|
||||||
183
bot/bot.py
183
bot/bot.py
|
|
@ -1,183 +0,0 @@
|
||||||
#!/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
29
bot/db.py
|
|
@ -1,29 +0,0 @@
|
||||||
#!/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
103
bot/func.py
|
|
@ -1,103 +0,0 @@
|
||||||
#!/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
74
bot/lang.py
|
|
@ -1,74 +0,0 @@
|
||||||
#!/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
65
bot/log.py
|
|
@ -1,65 +0,0 @@
|
||||||
#!/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)
|
|
||||||
|
|
@ -1,17 +0,0 @@
|
||||||
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"
|
|
||||||
|
|
@ -0,0 +1,10 @@
|
||||||
|
aiofiles>=24.1.0
|
||||||
|
aiohappyeyeballs>=2.3.4
|
||||||
|
aiohttp>=3.10.0
|
||||||
|
aiosignal>=1.3.1
|
||||||
|
attrs>=23.2.0
|
||||||
|
frozenlist>=1.4.1
|
||||||
|
idna>=3.7
|
||||||
|
multidict>=6.0.5
|
||||||
|
python-magic>=0.4.27
|
||||||
|
yarl>=1.9.4
|
||||||
|
|
@ -0,0 +1,16 @@
|
||||||
|
from setuptools import setup
|
||||||
|
|
||||||
|
setup(
|
||||||
|
name="TorrentUploaderBot",
|
||||||
|
version="",
|
||||||
|
url="",
|
||||||
|
author="ORUDO",
|
||||||
|
author_email="root@orudo.ru",
|
||||||
|
description="A simple Telegram bot that will allow you to upload torrent files / magnet links to a remote Torrent server (qBitTorrent, Transmission, etc.)",
|
||||||
|
install_requires=[
|
||||||
|
"aiohttp>=3.10.0",
|
||||||
|
"aiofiles>=24.1.0",
|
||||||
|
"aiofiles>=24.1.0"
|
||||||
|
],
|
||||||
|
packages=["tubot", "tubot.static", "tubot.torrent"],
|
||||||
|
)
|
||||||
|
|
@ -0,0 +1,24 @@
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
##########################################
|
||||||
|
# Shared Abstract classes and interfaces #
|
||||||
|
##########################################
|
||||||
|
|
||||||
|
# Imports
|
||||||
|
from abc import ABC, abstractmethod
|
||||||
|
|
||||||
|
|
||||||
|
class IValidatable(ABC):
|
||||||
|
"""
|
||||||
|
Interface initializing a class with a magic method for
|
||||||
|
checking the validity of its objects
|
||||||
|
"""
|
||||||
|
|
||||||
|
@abstractmethod
|
||||||
|
async def __validate__(self) -> bool:
|
||||||
|
"""
|
||||||
|
Checks if the object of the class is valid
|
||||||
|
|
||||||
|
:return: Object validity boolean
|
||||||
|
"""
|
||||||
|
raise NotImplementedError
|
||||||
|
|
@ -0,0 +1,21 @@
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
####################
|
||||||
|
# Static functions #
|
||||||
|
####################
|
||||||
|
|
||||||
|
# Imports
|
||||||
|
from tubot.static.abc import IValidatable
|
||||||
|
|
||||||
|
|
||||||
|
async def validate(obj: IValidatable, msg: str | None = None) -> bool:
|
||||||
|
"""
|
||||||
|
Throws an exception if the object has not been validated
|
||||||
|
|
||||||
|
:return: Object validity boolean
|
||||||
|
"""
|
||||||
|
if await obj.__validate__():
|
||||||
|
return True
|
||||||
|
if msg is None:
|
||||||
|
raise TypeError("Object validation failed")
|
||||||
|
raise TypeError(f"Object validation failed: {msg}")
|
||||||
|
|
@ -0,0 +1,82 @@
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
######################################################
|
||||||
|
# Abstract methods and interfaces for torrent module #
|
||||||
|
######################################################
|
||||||
|
|
||||||
|
from abc import ABC, abstractmethod
|
||||||
|
from tubot.static.abc import IValidatable
|
||||||
|
from tubot.torrent.types import TorrentTypes, ServerTypes
|
||||||
|
|
||||||
|
|
||||||
|
class TorrentObj(IValidatable, ABC):
|
||||||
|
"""
|
||||||
|
Abstract class of torrent object
|
||||||
|
"""
|
||||||
|
|
||||||
|
_ttype: TorrentTypes # Torrent type property
|
||||||
|
dest: str
|
||||||
|
content: str
|
||||||
|
|
||||||
|
@property
|
||||||
|
def torrent_type(self) -> TorrentTypes:
|
||||||
|
"""
|
||||||
|
:return: Torrent type
|
||||||
|
"""
|
||||||
|
if self._ttype is None:
|
||||||
|
raise NotImplementedError("Torrent type not implemented")
|
||||||
|
return self._ttype
|
||||||
|
|
||||||
|
def __init__(self, content: str, destination: str) -> None:
|
||||||
|
"""
|
||||||
|
:param content: Torrent content (link, file path)
|
||||||
|
:param destination: Download directory
|
||||||
|
"""
|
||||||
|
self.content = content
|
||||||
|
self.dest = destination
|
||||||
|
|
||||||
|
|
||||||
|
class TorrentAPI(IValidatable, ABC):
|
||||||
|
"""
|
||||||
|
Abstract class of torrent-server API's
|
||||||
|
"""
|
||||||
|
|
||||||
|
_atype: ServerTypes # Server type propery
|
||||||
|
|
||||||
|
def __init__(self) -> None:
|
||||||
|
if self._atype is None:
|
||||||
|
raise NotImplementedError("Torrent Server type not implemented")
|
||||||
|
|
||||||
|
async def upload(self, torrent: TorrentObj) -> None:
|
||||||
|
"""
|
||||||
|
Adds the torrent to a queue on the server
|
||||||
|
|
||||||
|
:param torrent: TorrenObject type (file, magnet, etc.)
|
||||||
|
"""
|
||||||
|
match torrent.torrent_type:
|
||||||
|
case TorrentTypes.File:
|
||||||
|
await self.upload_file(torrent)
|
||||||
|
case TorrentTypes.Magnet:
|
||||||
|
await self.upload_magnet(torrent)
|
||||||
|
case TorrentTypes.URL:
|
||||||
|
await self.upload_url(torrent)
|
||||||
|
|
||||||
|
@abstractmethod
|
||||||
|
async def upload_file(self, torrent) -> None:
|
||||||
|
raise NotImplementedError
|
||||||
|
|
||||||
|
@abstractmethod
|
||||||
|
async def upload_magnet(self, torrent) -> None:
|
||||||
|
raise NotImplementedError
|
||||||
|
|
||||||
|
@abstractmethod
|
||||||
|
async def upload_url(self, torrent) -> None:
|
||||||
|
raise NotImplementedError
|
||||||
|
|
||||||
|
@property
|
||||||
|
@abstractmethod
|
||||||
|
async def torrent_list(self) -> str:
|
||||||
|
"""
|
||||||
|
Returns PlainString with current torrent queue
|
||||||
|
"""
|
||||||
|
raise NotImplementedError
|
||||||
|
|
@ -0,0 +1,61 @@
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
#################################
|
||||||
|
# Torrent types implementations #
|
||||||
|
#################################
|
||||||
|
|
||||||
|
# Imports
|
||||||
|
from tubot.torrent.types import TorrentTypes
|
||||||
|
from tubot.torrent.abc import TorrentObj
|
||||||
|
from aiofiles import open, ospath
|
||||||
|
from magic import Magic
|
||||||
|
from re import match
|
||||||
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
|
|
||||||
|
class TorrentFile(TorrentObj):
|
||||||
|
"""
|
||||||
|
.torrent file
|
||||||
|
"""
|
||||||
|
|
||||||
|
_ttype = TorrentTypes.File
|
||||||
|
|
||||||
|
async def __validate__(self) -> bool:
|
||||||
|
if await ospath.isfile(self.content):
|
||||||
|
mime = Magic(mime=True).from_file(self.content)
|
||||||
|
if mime == "application/x-bittorrent":
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
async def getbytes(self) -> bytes:
|
||||||
|
async with open(self.content, "rb") as dottorrent:
|
||||||
|
return await dottorrent.read()
|
||||||
|
|
||||||
|
|
||||||
|
class TorrentMagnet(TorrentObj):
|
||||||
|
"""
|
||||||
|
Torrent magnet link
|
||||||
|
"""
|
||||||
|
|
||||||
|
_ttype = TorrentTypes.Magnet
|
||||||
|
|
||||||
|
async def __validate__(self) -> bool:
|
||||||
|
pattern = r"^magnet:\?xt=urn:btih:[a-fA-F0-9]{40}.*$"
|
||||||
|
if match(pattern, self.content):
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
class TorrentURL(TorrentObj):
|
||||||
|
"""
|
||||||
|
Http(s) link to .torrent file
|
||||||
|
"""
|
||||||
|
|
||||||
|
_ttype = TorrentTypes.URL
|
||||||
|
|
||||||
|
async def __validate__(self) -> bool:
|
||||||
|
try:
|
||||||
|
parse = urlparse(self.content)
|
||||||
|
return all([parse.scheme, parse.netloc])
|
||||||
|
except (TypeError, AttributeError):
|
||||||
|
return False
|
||||||
|
|
@ -0,0 +1,59 @@
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
############################
|
||||||
|
# Types for torrent module #
|
||||||
|
############################
|
||||||
|
|
||||||
|
# Imports
|
||||||
|
from collections.abc import Iterable
|
||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
|
||||||
|
class TorrentTypes(Enum):
|
||||||
|
"""
|
||||||
|
Types of torrents
|
||||||
|
"""
|
||||||
|
|
||||||
|
File = ".torrent file"
|
||||||
|
Magnet = "torrent magnet link"
|
||||||
|
URL = "http(s) link to .torrent file"
|
||||||
|
|
||||||
|
|
||||||
|
class ServerTypes(Enum):
|
||||||
|
"""
|
||||||
|
Types of Torrent servers API's
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
class TorrentFromServer(object):
|
||||||
|
|
||||||
|
name: str
|
||||||
|
state: str
|
||||||
|
percent: float
|
||||||
|
|
||||||
|
def __init__(self, name: str, state: str, percent: float) -> None:
|
||||||
|
self.name = name
|
||||||
|
self.state = state
|
||||||
|
self.percent = percent
|
||||||
|
|
||||||
|
def __str__(self) -> str:
|
||||||
|
return f"*Torrent:* {self.name}\n*State:* {self.state}\n*Progress:* {self.percent}"
|
||||||
|
|
||||||
|
|
||||||
|
class TorrentListBuilder(object):
|
||||||
|
"""
|
||||||
|
Torrent list type
|
||||||
|
"""
|
||||||
|
|
||||||
|
collection: list
|
||||||
|
|
||||||
|
def __init__(self) -> None:
|
||||||
|
self.collection = []
|
||||||
|
|
||||||
|
def append(self, torrent_data: Iterable) -> "TorrentListBuilder":
|
||||||
|
item = TorrentFromServer(*torrent_data)
|
||||||
|
self.collection.append(item)
|
||||||
|
return self
|
||||||
|
|
||||||
|
def __str__(self) -> str:
|
||||||
|
return "\n---\n".join(self.collection)
|
||||||
Reference in New Issue