utils.py

import sys
import logging
import argparse


def setup_logging(level=logging.INFO):
    root = logging.getLogger()
    root.setLevel(level)
    ch = logging.StreamHandler(sys.stdout)
    ch.setLevel(logging.DEBUG)
    formatter = logging.Formatter("%(asctime)s [%(levelname)s] %(name)s: %(message)s")
    ch.setFormatter(formatter)
    root.addHandler(ch)


def add_api_args(parser: argparse.ArgumentParser):
    parser.add_argument("api_id", help="API id")  # https://my.telegram.org/apps
    parser.add_argument("api_hash", help="API hash")
    parser.add_argument("phone", help="Phone")


def add_proxy_args(parser: argparse.ArgumentParser):
    parser.add_argument("--proxy_server", help="Proxy server", default="", required=False)
    parser.add_argument("--proxy_port", help="Proxy port", default="", required=False)
    parser.add_argument(
        "--proxy_type",
        help="Proxy type (socks5, http, mtproxy)",
        default="",
        required=False,
    )
    parser.add_argument("--proxy_username", help="Proxy user name", default="", required=False)
    parser.add_argument("--proxy_password", help="Proxy password", default="", required=False)
    parser.add_argument("--proxy_secret", help="Proxy secret (mtproxy)", default="", required=False)


def parse_proxy_type(args):
    obj_type = {
        "socks5": "proxyTypeSocks5",
        "http": "proxyTypeHttp",
        "mtproxy": "proxyTypeMtproto",
    }
    if args.proxy_type not in obj_type:
        return None
    obj = {"@type": obj_type[args.proxy_type]}
    if args.proxy_type in ["http", "socks5"]:
        obj["username"] = args.proxy_username
        obj["password"] = args.proxy_password
    else:
        obj["secret"] = args.secret
    return obj