github-actions-init.py

import os
import platform
import shutil
import stat
import subprocess
import sys
from typing import List


def run_command(command: List[str]):
    try:
        return subprocess.run(
            command,
            check=True,
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
            text=True,
        ).stdout.strip()
    except subprocess.CalledProcessError:
        return None


def is_git_repo_and_on_branch(branch_name: str = "python"):
    return (
        run_command(["git", "rev-parse", "--is-inside-work-tree"])
        and run_command(["git", "branch", "--show-current"]) == branch_name
    )


def check_package(package_name: str):
    try:
        __import__(package_name)
        return True
    except ImportError:
        return False


def delete_path(path: str):
    if not os.path.exists(path):
        print(f"Path '{path}' does not exist.")
        return

    try:
        if os.path.isdir(path):
            shutil.rmtree(path, ignore_errors=True)
        else:
            os.remove(path)
        print(f"'{path}' deleted successfully.")
    except OSError as e:
        print(f"Error deleting '{path}': {e}")


def ensure_directories(directories: List[str]):
    base_dir = os.path.dirname(os.path.abspath(__file__))
    for d in directories:
        full_path = os.path.join(base_dir, d)
        os.makedirs(full_path, exist_ok=True)
        os.chmod(full_path, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)


def run_pip_install(*requirements_files: str):
    for req in requirements_files:
        subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", req])


# ---- Packages ----
packages = ["pycountry", "django", "requests", "pytz", "timezonefinder"]

if platform.system() == "Windows" and is_git_repo_and_on_branch():
    packages += ["PySide6", "wmi", "qtawesome", "nuitka"]

missing_packages = [pkg for pkg in packages if not check_package(pkg)]

if missing_packages:
    print(f"Missing packages: {', '.join(missing_packages)}")

    subprocess.check_call([sys.executable, "requirements_install.py", "--generate"])

    run_pip_install(
        "requirements.txt",
        "requirements-base.txt",
        "requirements-minimal.txt",
    )
else:
    print("All required packages are installed.")


# ---- Directories ----
dirs = [
    ".cache",
    "config",
    "assets/proxies",
    "assets/chrome-profiles",
    "assets/chrome",
    "tmp/cookies",
    "tmp/data",
    "dist",
]


ensure_directories(dirs)