main.py

import sys
import subprocess

from helpers.deps import install_requirements, check_python_version, check_ffmpeg_installed


def option1():
    subprocess.run([sys.executable, "scripts/myfans_scrap.py"])


def option2():
    subprocess.run([sys.executable, "scripts/myfans_dl.py"])


def option3():
    subprocess.run([sys.executable, "scripts/myfans_image_dl.py"])


def main():
    # Check if the required packages are installed and if ffmpeg is installed.
    if not install_requirements() and check_ffmpeg_installed():
        sys.exit(1)

    options = {
        "1": option1,
        "2": option2,
        "3": option3
    }

    print("Choose an option:")
    print("1. Scrap post id")
    print("2. Download videos")
    print("3. Download images")

    choice = input("Enter your choice (1, 2 or 3): ")
    action = options.get(choice)

    if action:
        action()
    else:
        print("Invalid choice. Please enter 1, 2 or 3")


if __name__ == "__main__":
    if check_python_version():
        main()