setup.py

#!/usr/bin/env python3

import os.path
import subprocess as sp
import sys
from src import detect_os

installer_dir = os.path.dirname(os.path.abspath(__file__))
is_efi = os.path.exists("/sys/firmware/efi")
use_other_iso = "" # e.g. "arch" if using Arch iso to install different OS like Fedora # TODO remove

try: # if using iso to install another OS, two extra args should be passed
    if is_efi:
        args = list(sys.argv[0:4]) # just first 3 arguments (exclude distro arguments)
        distro = sys.argv[4]
        distro_name = sys.argv[5]
    else:
        args = list(sys.argv[0:3]) # just first 2 arguments (exclude distro arguments)
        distro = sys.argv[3]
        distro_name = sys.argv[4]
except IndexError:
    distro = detect_os.get_distro_id()
    distro_name = detect_os.get_distro_name()

if distro:
    if use_other_iso != "":
        distro_for_prep = use_other_iso
    else:
        distro_for_prep = distro
#    try: # CAUTION: comment lines 30-37 & unindent line 38 if prepared manually
#        if is_efi:
#            sp.check_output([f'./src/prep/{distro_for_prep}_live.sh', f'{args[1]}', f'{args[2]}', f'{args[3]}'])
#        else:
#            sp.check_output([f'./src/prep/{distro_for_prep}_live.sh', f'{args[1]}', f'{args[2]}'])
#    except sp.CalledProcessError as e:
#        print(f"F: There was an error in prep steps! {e.output.decode('utf-8')}")
#    else:
    __import__(f"src.distros.{distro}.installer")
else:
    print("F: Distribution could not be detected!")