cp2toto.py

import argparse
import asyncio
import re
import sys
import traceback

from utils.menu import select_destination, select_origin
from utils.misc import conversion_flow, remove_local_files
from utils.output import bye, welcome
from utils.scp_connect import scp
from utils.tg import send_message_to_telegram_channel


def main() -> None:
    """
    This is the main function that orchestrates the file transfer process.

    The function follows these steps:
    1. Displays a welcome message to the user.
    2. Prompts the user to select the files they want to transfer (origin files).
    3. Asks the user to specify the destination folder where the files will be transferred
    4. Asks the user if they want to convert the files to MP4 with H.265 codec.
    5. Initiates file transfer process, copying selected files to the destination folder.
    6. If successful, it asks the user if they want to remove the original files.
    7. Also, if the destination folder is the 'movies/' folder, it asks to send a message
       to the Telegram channel.
    8. Finally, the function displays a farewell message and terminates the program.

    Args:
        -tg: Optional argument to send a message to the Telegram channel and skip the rest
             of the flow.

    The function handles two types of exceptions:
    - KeyboardInterrupt:
        If the user interrupts the program (e.g., by pressing Ctrl+C), the function
        immediately exits.
    - General exceptions:
        If any other type of exception occurs, the function prints the exception details
        and exits with a status code of 1.
    """

    parser = argparse.ArgumentParser(
        description="File management and optionally send a Telegram message."
    )
    parser.add_argument(
        "-tg",
        "--send_tg_message_only",
        action="store_true",
        help="Only send a message to the Telegram channel and skip the rest of the flow.",
    )
    args = parser.parse_args()

    try:
        welcome()

        if args.send_tg_message_only:
            asyncio.run(send_message_to_telegram_channel())
            sys.exit()

        origin_files = select_origin()
        destination_folder = select_destination()

        origin_files = conversion_flow(origin_files)
        scp_completed = scp(origin_files, destination_folder)

        if scp_completed:
            pattern = r"/(movies|series)/"

            if re.search(pattern, destination_folder) is not None:
                asyncio.run(send_message_to_telegram_channel())

            remove_local_files(origin_files)

        bye()

    except KeyboardInterrupt:
        sys.exit()

    except Exception as general_error:
        print(f"A general error occurred: {general_error}")
        traceback.print_exc()
        sys.exit(1)


if __name__ == "__main__":
    main()