Skip to content

Commit

Permalink
sync and debloat
Browse files Browse the repository at this point in the history
  • Loading branch information
5hojib committed Dec 5, 2024
1 parent a7a7176 commit e6579a8
Show file tree
Hide file tree
Showing 52 changed files with 1,628 additions and 3,400 deletions.
92 changes: 4 additions & 88 deletions bot/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ def format(self, record):
error("The README.md file there to be read! Exiting now!")
bot_loop.stop()
exit(1)
except:
except Exception:
pass

task_dict_lock = Lock()
Expand All @@ -102,7 +102,6 @@ def format(self, record):
same_directory_lock = Lock()
status_dict = {}
task_dict = {}
rss_dict = {}

BOT_TOKEN = environ.get("BOT_TOKEN", "")
if len(BOT_TOKEN) == 0:
Expand Down Expand Up @@ -194,7 +193,7 @@ def format(self, record):
max_concurrent_transmissions=10,
).start()
IS_PREMIUM_USER = user.me.is_premium
except:
except Exception:
error("Failed to start client from USER_SESSION_STRING")
IS_PREMIUM_USER = False
user = ""
Expand All @@ -218,11 +217,7 @@ def format(self, record):
if DEFAULT_UPLOAD != "gd":
DEFAULT_UPLOAD = "rc"

DOWNLOAD_DIR = environ.get("DOWNLOAD_DIR", "")
if len(DOWNLOAD_DIR) == 0:
DOWNLOAD_DIR = "/usr/src/app/downloads/"
elif not DOWNLOAD_DIR.endswith("/"):
DOWNLOAD_DIR = f"{DOWNLOAD_DIR}/"
DOWNLOAD_DIR = "/usr/src/app/downloads/"

AUTHORIZED_CHATS = environ.get("AUTHORIZED_CHATS", "")
if len(AUTHORIZED_CHATS) != 0:
Expand Down Expand Up @@ -261,24 +256,10 @@ def format(self, record):
if len(INDEX_URL) == 0:
INDEX_URL = ""

SEARCH_API_LINK = environ.get("SEARCH_API_LINK", "").rstrip("/")
if len(SEARCH_API_LINK) == 0:
SEARCH_API_LINK = ""

LEECH_FILENAME_PREFIX = environ.get("LEECH_FILENAME_PREFIX", "")
if len(LEECH_FILENAME_PREFIX) == 0:
LEECH_FILENAME_PREFIX = ""

SEARCH_PLUGINS = environ.get("SEARCH_PLUGINS", "")
if len(SEARCH_PLUGINS) == 0:
SEARCH_PLUGINS = ""
else:
try:
SEARCH_PLUGINS = eval(SEARCH_PLUGINS)
except:
error(f"Wrong USENET_SERVERS format: {SEARCH_PLUGINS}")
SEARCH_PLUGINS = ""

MAX_SPLIT_SIZE = 4194304000 if IS_PREMIUM_USER else 2097152000

LEECH_SPLIT_SIZE = environ.get("LEECH_SPLIT_SIZE", "")
Expand All @@ -291,19 +272,10 @@ def format(self, record):
else:
LEECH_SPLIT_SIZE = int(LEECH_SPLIT_SIZE)

STATUS_UPDATE_INTERVAL = environ.get("STATUS_UPDATE_INTERVAL", "")
if len(STATUS_UPDATE_INTERVAL) == 0:
STATUS_UPDATE_INTERVAL = 15
else:
STATUS_UPDATE_INTERVAL = int(STATUS_UPDATE_INTERVAL)

YT_DLP_OPTIONS = environ.get("YT_DLP_OPTIONS", "")
if len(YT_DLP_OPTIONS) == 0:
YT_DLP_OPTIONS = ""

SEARCH_LIMIT = environ.get("SEARCH_LIMIT", "")
SEARCH_LIMIT = 0 if len(SEARCH_LIMIT) == 0 else int(SEARCH_LIMIT)

LEECH_DUMP_CHAT = environ.get("LEECH_DUMP_CHAT", "")
LEECH_DUMP_CHAT = "" if len(LEECH_DUMP_CHAT) == 0 else LEECH_DUMP_CHAT

Expand All @@ -312,12 +284,6 @@ def format(self, record):

CMD_SUFFIX = environ.get("CMD_SUFFIX", "")

RSS_CHAT = environ.get("RSS_CHAT", "")
RSS_CHAT = "" if len(RSS_CHAT) == 0 else RSS_CHAT

RSS_DELAY = environ.get("RSS_DELAY", "")
RSS_DELAY = 600 if len(RSS_DELAY) == 0 else int(RSS_DELAY)

TORRENT_TIMEOUT = environ.get("TORRENT_TIMEOUT", "")
TORRENT_TIMEOUT = "" if len(TORRENT_TIMEOUT) == 0 else int(TORRENT_TIMEOUT)

Expand All @@ -330,9 +296,6 @@ def format(self, record):
QUEUE_UPLOAD = environ.get("QUEUE_UPLOAD", "")
QUEUE_UPLOAD = "" if len(QUEUE_UPLOAD) == 0 else int(QUEUE_UPLOAD)

INCOMPLETE_TASK_NOTIFIER = environ.get("INCOMPLETE_TASK_NOTIFIER", "")
INCOMPLETE_TASK_NOTIFIER = INCOMPLETE_TASK_NOTIFIER.lower() == "true"

STOP_DUPLICATE = environ.get("STOP_DUPLICATE", "")
STOP_DUPLICATE = STOP_DUPLICATE.lower() == "true"

Expand All @@ -342,24 +305,15 @@ def format(self, record):
USE_SERVICE_ACCOUNTS = environ.get("USE_SERVICE_ACCOUNTS", "")
USE_SERVICE_ACCOUNTS = USE_SERVICE_ACCOUNTS.lower() == "true"

WEB_PINCODE = environ.get("WEB_PINCODE", "")
WEB_PINCODE = WEB_PINCODE.lower() == "true"

AS_DOCUMENT = environ.get("AS_DOCUMENT", "")
AS_DOCUMENT = AS_DOCUMENT.lower() == "true"

EQUAL_SPLITS = environ.get("EQUAL_SPLITS", "")
EQUAL_SPLITS = EQUAL_SPLITS.lower() == "true"

MEDIA_GROUP = environ.get("MEDIA_GROUP", "")
MEDIA_GROUP = MEDIA_GROUP.lower() == "true"

USER_TRANSMISSION = environ.get("USER_TRANSMISSION", "")
USER_TRANSMISSION = USER_TRANSMISSION.lower() == "true" and IS_PREMIUM_USER

BASE_URL_PORT = environ.get("BASE_URL_PORT", "")
BASE_URL_PORT = 80 if len(BASE_URL_PORT) == 0 else int(BASE_URL_PORT)

BASE_URL = environ.get("BASE_URL", "").rstrip("/")
if len(BASE_URL) == 0:
warning("BASE_URL not provided!")
Expand All @@ -373,21 +327,6 @@ def format(self, record):
if len(UPSTREAM_BRANCH) == 0:
UPSTREAM_BRANCH = "master"

RCLONE_SERVE_URL = environ.get("RCLONE_SERVE_URL", "").rstrip("/")
if len(RCLONE_SERVE_URL) == 0:
RCLONE_SERVE_URL = ""

RCLONE_SERVE_PORT = environ.get("RCLONE_SERVE_PORT", "")
RCLONE_SERVE_PORT = 8080 if len(RCLONE_SERVE_PORT) == 0 else int(RCLONE_SERVE_PORT)

RCLONE_SERVE_USER = environ.get("RCLONE_SERVE_USER", "")
if len(RCLONE_SERVE_USER) == 0:
RCLONE_SERVE_USER = ""

RCLONE_SERVE_PASS = environ.get("RCLONE_SERVE_PASS", "")
if len(RCLONE_SERVE_PASS) == 0:
RCLONE_SERVE_PASS = ""

NAME_SUBSTITUTE = environ.get("NAME_SUBSTITUTE", "")
NAME_SUBSTITUTE = "" if len(NAME_SUBSTITUTE) == 0 else NAME_SUBSTITUTE

Expand All @@ -400,32 +339,28 @@ def format(self, record):
FFMPEG_CMDS = environ.get("FFMPEG_CMDS", "")
try:
FFMPEG_CMDS = [] if len(FFMPEG_CMDS) == 0 else eval(FFMPEG_CMDS)
except:
except Exception:
error(f"Wrong FFMPEG_CMDS format: {FFMPEG_CMDS}")
FFMPEG_CMDS = []

config_dict = {
"AS_DOCUMENT": AS_DOCUMENT,
"AUTHORIZED_CHATS": AUTHORIZED_CHATS,
"BASE_URL": BASE_URL,
"BASE_URL_PORT": BASE_URL_PORT,
"BOT_TOKEN": BOT_TOKEN,
"CMD_SUFFIX": CMD_SUFFIX,
"DATABASE_URL": DATABASE_URL,
"DEFAULT_UPLOAD": DEFAULT_UPLOAD,
"DOWNLOAD_DIR": DOWNLOAD_DIR,
"EQUAL_SPLITS": EQUAL_SPLITS,
"EXTENSION_FILTER": EXTENSION_FILTER,
"FFMPEG_CMDS": FFMPEG_CMDS,
"FILELION_API": FILELION_API,
"GDRIVE_ID": GDRIVE_ID,
"INCOMPLETE_TASK_NOTIFIER": INCOMPLETE_TASK_NOTIFIER,
"INDEX_URL": INDEX_URL,
"IS_TEAM_DRIVE": IS_TEAM_DRIVE,
"LEECH_DUMP_CHAT": LEECH_DUMP_CHAT,
"LEECH_FILENAME_PREFIX": LEECH_FILENAME_PREFIX,
"LEECH_SPLIT_SIZE": LEECH_SPLIT_SIZE,
"MEDIA_GROUP": MEDIA_GROUP,
"MIXED_LEECH": MIXED_LEECH,
"NAME_SUBSTITUTE": NAME_SUBSTITUTE,
"OWNER_ID": OWNER_ID,
Expand All @@ -434,17 +369,7 @@ def format(self, record):
"QUEUE_UPLOAD": QUEUE_UPLOAD,
"RCLONE_FLAGS": RCLONE_FLAGS,
"RCLONE_PATH": RCLONE_PATH,
"RCLONE_SERVE_URL": RCLONE_SERVE_URL,
"RCLONE_SERVE_USER": RCLONE_SERVE_USER,
"RCLONE_SERVE_PASS": RCLONE_SERVE_PASS,
"RCLONE_SERVE_PORT": RCLONE_SERVE_PORT,
"RSS_CHAT": RSS_CHAT,
"RSS_DELAY": RSS_DELAY,
"SEARCH_API_LINK": SEARCH_API_LINK,
"SEARCH_LIMIT": SEARCH_LIMIT,
"SEARCH_PLUGINS": SEARCH_PLUGINS,
"STATUS_LIMIT": STATUS_LIMIT,
"STATUS_UPDATE_INTERVAL": STATUS_UPDATE_INTERVAL,
"STOP_DUPLICATE": STOP_DUPLICATE,
"STREAMWISH_API": STREAMWISH_API,
"SUDO_USERS": SUDO_USERS,
Expand All @@ -457,7 +382,6 @@ def format(self, record):
"UPSTREAM_BRANCH": UPSTREAM_BRANCH,
"USER_SESSION_STRING": USER_SESSION_STRING,
"USE_SERVICE_ACCOUNTS": USE_SERVICE_ACCOUNTS,
"WEB_PINCODE": WEB_PINCODE,
"YT_DLP_OPTIONS": YT_DLP_OPTIONS,
}

Expand Down Expand Up @@ -491,14 +415,6 @@ def format(self, record):
run(["chmod", "600", ".netrc"], check=False)
run(["cp", ".netrc", "/root/.netrc"], check=False)

trackers = (
check_output(
"curl -Ns https://raw.githubusercontent.com/XIU2/TrackersListCollection/master/all.txt https://ngosang.github.io/trackerslist/trackers_all_http.txt https://newtrackon.com/api/all https://raw.githubusercontent.com/hezhijie0327/Trackerslist/main/trackerslist_tracker.txt | awk '$0' | tr '\n\n' ','",
shell=True,
)
.decode("utf-8")
.rstrip(",")
)
with open("a2c.conf", "a+") as a:
if TORRENT_TIMEOUT is not None:
a.write(f"bt-stop-timeout={TORRENT_TIMEOUT}\n")
Expand Down
5 changes: 1 addition & 4 deletions bot/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,6 @@
delete_message,
five_minute_del,
)
from .helper.mirror_leech_utils.rclone_utils.serve import rclone_serve_booter


@new_task
Expand Down Expand Up @@ -232,7 +231,6 @@ async def log(_, message):
/{BotCommands.AExecCommand}: Exec async functions (Only Owner).
/{BotCommands.ExecCommand}: Exec sync functions (Only Owner).
/{BotCommands.ClearLocalsCommand}: Clear {BotCommands.AExecCommand} or {BotCommands.ExecCommand} locals (Only Owner).
/{BotCommands.RssCommand}: RSS Menu.
"""


Expand All @@ -246,7 +244,7 @@ async def restart_notification():
cmd = r"""remote_url=$(git config --get remote.origin.url) &&
if echo "$remote_url" | grep -qE "github\.com[:/](.*)/(.*?)(\.git)?$"; then
last_commit=$(git log -1 --pretty=format:'%h') &&
commit_link="https://github.com/5hojib/Aeon/commit/$last_commit" &&
commit_link="https://github.com/AeonOrg/Aeon-MLTB/commit/$last_commit" &&
echo $commit_link;
else
echo "Failed to extract repository name and owner name from the remote URL.";
Expand Down Expand Up @@ -323,7 +321,6 @@ async def main():
bot_settings.initiate_search_tools(),
restart_notification(),
telegraph.create_account(),
rclone_serve_booter(),
sync_to_async(start_aria2_listener, wait=False),
)
create_help_buttons()
Expand Down
Loading

0 comments on commit e6579a8

Please sign in to comment.