from importlib.util import find_spec
from subprocess import check_call, run
from platform import python_version
from sys import exit

class Install:

    def __init__(self) -> None:
        self.PYTHON_MIN_VERSION = '3.10'
        self.module_to_install = ['sqlalchemy','psutil','requests']

        if not self.checkPythonVersion():
            # Tester si c'est la bonne version de python
            exit("Python Version Error")
        else:
            # Sinon tester les dependances python et les installer avec pip
            self.checkDependencies()

        return None

    def checkPythonVersion(self) -> bool:
        """Test si la version de python est autorisée ou non

        Returns:
            bool: True si la version de python est autorisé sinon False
        """
        python_required_version = self.PYTHON_MIN_VERSION.split('.')
        python_current_version = python_version().split('.')

        if int(python_current_version[0]) < int(python_required_version[0]):
            print(f"## Your python version must be greather than or equal to {self.PYTHON_MIN_VERSION} ##")
            return False
        elif int(python_current_version[1]) < int(python_required_version[1]):
            print(f"### Your python version must be greather than or equal to {self.PYTHON_MIN_VERSION} ###")
            return False

        print(f"===> Version of python : {python_version()} ==> OK")

        return True
    
    def checkDependencies(self) -> None:
        """### Verifie les dépendances si elles sont installées
        - Test si les modules sont installés
        - Met a jour pip
        - Install les modules manquants
        """
        do_install = False

        for module in self.module_to_install:
            if find_spec(module) is None:
                do_install = True

        if not do_install:
            return None

        print("===> Verifier si pip est a jour")
        run(['python', '-m', 'pip', 'install', '--upgrade', 'pip'])

        if find_spec('greenlet') is None:
            check_call(['pip','install', '--only-binary', ':all:', 'greenlet'])
            print('====> Module Greenlet installé')

        for module in self.module_to_install:
            if find_spec(module) is None:
                print("### Trying to install missing python packages ###")
                check_call(['pip','install', module])
                print(f"====> Module {module} installé")
            else:
                print(f"==> {module} already installed")