add bot files to git

.env.example

@@ -0,0 +1,5 @@
+BOT_TOKEN = 123456789:AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
+ADMIN_ID = 987654321
+# PROXY_URL = "http://localhost:8080"
+TZ = Asia/Tehran
+CRON_JOB = "0 * * * *"

.gitignore

@@ -158,3 +158,5 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+server_list.json
+backups/

main.py

@@ -0,0 +1,149 @@
+import asyncio
+import json
+import logging
+import sys
+import zipfile
+from datetime import datetime
+from os import getenv, mkdir, path, remove
+from typing import List
+
+import paramiko
+import pytz
+from aiogram import Bot, Dispatcher, types
+from aiogram.client.session.aiohttp import AiohttpSession
+from aiogram.enums import ParseMode
+from aiogram.filters import Command, CommandStart
+from aiogram.utils.markdown import bold
+from apscheduler.schedulers.asyncio import AsyncIOScheduler
+from dotenv import load_dotenv
+from scp import SCPClient
+
+load_dotenv()
+
+
+TOKEN = getenv("BOT_TOKEN")
+
+CHAT_ID = int(getenv("CHAT_ID"))
+
+PROXY_URL = getenv("PROXY_URL")
+
+CRON_JOB = getenv("CRON_JOB")
+
+TZ = getenv("TZ", "Asia/Tehran")
+
+BACKUP_PATH = "./backups/"
+
+BACKUP_FILE_NAME = "marzban-backup.zip"
+
+
+with open("./server_list.json", "r") as jf:
+    SERVER_LIST = json.loads(jf.read())
+
+
+session = AiohttpSession()
+if PROXY_URL:
+    session.proxy = PROXY_URL
+
+BOT = Bot(TOKEN, parse_mode=ParseMode.MARKDOWN,
+          #   session=session
+          )
+
+dp = Dispatcher()
+
+if not path.exists(BACKUP_PATH):
+    mkdir(BACKUP_PATH)
+
+
+@dp.message(CommandStart())
+async def command_start_handler(message: types.Message) -> None:
+    await message.answer(f"Hello, {bold(message.from_user.full_name)}!")
+
+
+@dp.message(Command(commands="db"))
+async def send_full_backup_command(message: types.Message):
+    if message.from_user.id == CHAT_ID:
+        await send_full_backups()
+
+
+def get_list_dir(ssh: paramiko.SSHClient, path) -> List[str]:
+    stdin, stdout, stderr = ssh.exec_command(f"ls -a {path}")
+    return stdout.read().decode().split()[2:]
+
+
+def is_dir(ssh: paramiko.SSHClient, path) -> bool:
+    stdin, stdout, stderr = ssh.exec_command(
+        f'test -d {path}  && echo "is dir" || echo "isnt dir"')
+    is_dir = stdout.read().decode().strip() == 'is dir'
+    return is_dir
+
+
+def get_date():
+    return datetime.now(pytz.timezone(TZ)).strftime('%Y/%m/%d %H:%M:%S')
+
+
+def _create_zipFile(ssh, scp_client_obj, zip_file_obj, remote_path, files):
+    for f in files:
+        if is_dir(ssh, f"{remote_path}{f}"):
+            _files = get_list_dir(ssh, f"{remote_path}{f}")
+            zip_file_obj = _create_zipFile(
+                ssh, scp_client_obj, zip_file_obj, f"{remote_path}{f}/", _files)
+            continue
+        scp_client_obj.get(f"{remote_path}{f}", f"{BACKUP_PATH}{f}")
+        zip_file_obj.write(f"{BACKUP_PATH}{f}", f"{remote_path[1:]}{f}")
+        remove(f"{BACKUP_PATH}{f}")
+    return zip_file_obj
+
+
+def create_zipFile(hostname, port, username, password, var_files, opt_files):
+    try:
+        ssh = createSSHClient(hostname, port, username, password)
+        with (
+            SCPClient(ssh.get_transport()) as scp,
+            zipfile.ZipFile(BACKUP_FILE_NAME, "w", zipfile.ZIP_DEFLATED) as zf
+        ):
+            remote_var_files = get_list_dir(ssh, var_files)
+            remote_opt_files = get_list_dir(ssh, opt_files)
+            zf = _create_zipFile(ssh, scp, zf, var_files, remote_var_files)
+            zf = _create_zipFile(ssh, scp, zf, opt_files, remote_opt_files)
+    except Exception as e:
+        logging.info(e)
+        return
+    return BACKUP_FILE_NAME
+
+
+def createSSHClient(server, port, user, password):
+    client = paramiko.SSHClient()
+    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+    client.connect(server, port, user, password)
+    return client
+
+
+async def send_full_backups():
+    for i in SERVER_LIST["servers"]:
+        hostname = i["host"]
+        port = i["port"]
+        username = i['user']
+        password = i['pass']
+        var_files = i['var_files']
+        opt_files = i['opt_files']
+        bac = create_zipFile(hostname, port, username,
+                             password, var_files, opt_files)
+        if not bac:
+            continue
+        date = get_date()
+        await BOT.send_document(chat_id=CHAT_ID, document=types.FSInputFile(path=bac, filename=bac), caption=f'🕐 Date : {date}\n\n🔰 IP : `{hostname}`')
+        remove(bac)
+
+
+async def main() -> None:
+    asc = AsyncIOScheduler(timezone=TZ)
+    minute, hour, day, month, day_of_week = CRON_JOB.split()
+    asc.add_job(func=send_full_backups, trigger="cron",
+                minute=minute, hour=hour, day=day, month=month, day_of_week=day_of_week)
+    asc.start()
+    await dp.start_polling(BOT)
+
+
+if __name__ == "__main__":
+    logging.basicConfig(level=logging.INFO, stream=sys.stdout)
+    asyncio.run(main())

requirements.txt

@@ -0,0 +1,32 @@
+aiofiles==23.1.0
+aiogram==3.1.1
+aiohttp==3.8.6
+aiohttp-socks==0.8.4
+aiosignal==1.3.1
+annotated-types==0.6.0
+APScheduler==3.10.4
+async-timeout==4.0.3
+attrs==23.1.0
+bcrypt==4.0.1
+certifi==2023.7.22
+cffi==1.16.0
+charset-normalizer==3.3.1
+cryptography==41.0.4
+frozenlist==1.4.0
+idna==3.4
+magic-filter==1.0.12
+multidict==6.0.4
+paramiko==3.3.1
+pycparser==2.21
+pydantic==2.3.0
+pydantic_core==2.6.3
+PyNaCl==1.5.0
+python-dotenv==1.0.0
+python-socks==2.4.3
+pytz==2023.3.post1
+scp==0.14.5
+six==1.16.0
+typing_extensions==4.7.1
+tzdata==2023.3
+tzlocal==5.2
+yarl==1.9.2

server_list.json.example

@@ -0,0 +1,12 @@
+{
+    "servers":[
+        {
+            "host": "host",
+            "port": 22,
+            "user": "user",
+            "pass": "pass",
+            "var_files": "/var/lib/marzban/",
+            "opt_files": "/opt/marzban/"
+        }
+    ]
+}