application.py

try:
    from argparse import ArgumentParser, Namespace
except (ImportError, ModuleNotFoundError) as e:
    print("Modules needed not found: " + str(e))

from out import cPrint, Colors
from upgrader import PiPUpgrader


def main(arguments: Namespace):
    try:
        import pickle
        import logging
        import secrets
        import string

        from os import path
        from telegram.ext import Updater, CommandHandler
    except (ImportError, ModuleNotFoundError) as import_error:
        print("Modules needed not found: " + str(import_error))
        exit(-1)
    else:
        token = arguments.token
        youtube_api_key = arguments.youtube
        creator_id = arguments.creator
        database_user = arguments.db_user
        database_password = arguments.db_password
        must_show_version = arguments.version
        if must_show_version:
            print("Version")
            exit(0)
        if not path.exists("app_data.dict"):
            if not token:
                raise ValueError("You must add token at least the first time you execute this app")
            elif not youtube_api_key:
                raise ValueError("You must include the YouTube API Key at least the first time you execute this app")
            elif not creator_id:
                raise ValueError("You must include the creator ID (Telegram) at least the first time you execute "
                                 "this app")
            else:
                if not database_user:
                    database_user = "youtube_md_bot"
                if not database_password:
                    alphabet = string.ascii_letters + string.digits
                    database_password = ''.join(secrets.choice(alphabet) for i in range(32))
                with open("app_data.dict", "wb") as app_data_file:
                    app_data = {"TOKEN": token,
                                "YT_API": youtube_api_key,
                                "CREATOR_ID": creator_id,
                                "DB_USER": database_user,
                                "DB_PASSWORD:": database_password}
                    pickle.dump(app_data, app_data_file, pickle.HIGHEST_PROTOCOL)
                main(arguments)
        else:
            cPrint("Initializing bot...", Colors.GREEN)
            cPrint("Looking for packages updates...", Colors.GREEN)
            upgrader = PiPUpgrader("requirements.txt")
            upgrader.upgradePackages()
            cPrint("Obtaining values...", Colors.GREEN)
            with open("app_data.dict", "rb") as app_data_file:
                # unpickler = pickle.Unpickler(app_data_file)
                app_data = pickle.load(app_data_file)
            updater = Updater(token=app_data["TOKEN"], workers=50)
            dispatcher = updater.dispatcher
            logging.basicConfig(format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
                                level=logging.DEBUG)
            updater.start_polling(poll_interval=5, timeout=60)


if __name__ == '__main__':
    args = ArgumentParser()
    args.add_argument("-t",
                      "--token",
                      help="Telegram token obtained via BotFather",
                      type=str)
    args.add_argument("-y",
                      "--youtube",
                      help="YouTube API Key",
                      type=str)
    args.add_argument("-c",
                      "--creator",
                      help="Telegram ID of the creator",
                      type=int)
    args.add_argument("-dbu",
                      "--db_user",
                      help="Database user (can be empty)",
                      type=str)
    args.add_argument("-dbp",
                      "--db_password",
                      help="Database password (can be empty)",
                      type=str)
    args.add_argument("-v",
                      "--version",
                      help="Application version",
                      action="store_true")
    main(args.parse_args())