cli.py

def get_user_credentials():
    """Utility function to get a user's credentials from environment variables."""
    try:
        username = os.environ["TELEGRAM_USERNAME"]
        api_id = int(os.environ["TELEGRAM_API_ID"])
        api_hash = os.environ["TELEGRAM_API_HASH"]
    except (TypeError, KeyError):
        print(
            "error: Please set the following environment variables:\n"
            "TELEGRAM_USERNAME, TELEGRAM_API_ID, TELEGRAM_API_HASH",
            file=sys.stderr,
        )
        sys.exit(1)
    return username, api_id, api_hash

def queue(args=None):
    """The queue command."""
    parser = argparse.ArgumentParser(prog="queue")
    opts = parser.parse_args(args=args)
    print(opts)
    return 0

def review(args=None):
    """The review command."""
    parser = argparse.ArgumentParser(prog="review")
    opts = parser.parse_args(args=args)  # noqa

    username, api_id, api_hash = get_user_credentials()
    client = TelegramClient("review", api_id, api_hash)
    client.start()

    messages = list(reversed(client.iter_messages(username)))
    print(len(messages))

    download_media = False
    for message in messages:
        if not message.message:
            if download_media:
                media = message.download_media()
                if media:
                    print(media)
        else:
            print(message.message)
            print()

    return 0