Adding 'make update' to update from git repository. remove previous installation.py file, the update of packages is done via install.py file.

core/install.py

@@ -1,23 +1,99 @@
 import argparse
 import os
 import sys
+import json
+from dataclasses import dataclass
+from subprocess import check_call, CalledProcessError, check_output
 from pathlib import Path
 from platform import python_version_tuple
+import traceback
 
 parser = argparse.ArgumentParser(description="Python Installation Code")
 parser.add_argument('--check-version', action='store_true', help='Check if the python version is ok!')
 parser.add_argument('--install', action='store_true', help='Run the installation')
+parser.add_argument('--git-update', action='store_true', help='Update from git (main repository)')
 args = parser.parse_args()
 
 PYTHON_REQUIRED_VERSION = (3, 10, 0)
 PYTHON_SYSTEM_VERSION = tuple(map(int, python_version_tuple()))
 ROOT_PATH = os.getcwd()
-PYENV = Path(f'{ROOT_PATH}/.pyenv/bin/python')
+PYENV = Path(ROOT_PATH).joinpath('.pyenv/bin/python') if os.name != 'nt' else Path(ROOT_PATH).joinpath('.pyenv/Scripts/python.exe')
+PIPENV = Path(f'{ROOT_PATH}/.pyenv/bin/pip') if os.name != 'nt' else Path(f'{ROOT_PATH}/.pyenv/Scripts/pip.exe')
 USER_HOME_DIRECTORY = Path.home()
 SYSTEMD_PATH = Path(USER_HOME_DIRECTORY).joinpath('.config', 'systemd', 'user')
 PY_EXEC = 'defender.py'
 SERVICE_FILE_NAME = 'defender.service'
 
+@dataclass
+class Package:
+	name: str = None
+	version: str = None
+
+def __load_required_package_versions() -> list[Package]:
+	"""This will create Package model with package names and required version
+        """
+	try:
+		DB_PACKAGES: list[Package] = []		
+		version_filename = Path(ROOT_PATH).joinpath('version.json')  # f'.{os.sep}version.json'
+		with open(version_filename, 'r') as version_data:
+			package_info:dict[str, str] = json.load(version_data)
+
+		for name, version in package_info.items():
+			if name == 'version':
+				continue
+			DB_PACKAGES.append(
+				Package(name=name, version=version)
+				)
+
+		return DB_PACKAGES
+
+	except FileNotFoundError as fe:
+		print(f"File not found: {fe}")
+	except Exception as err:
+		print(f"General Error: {err}")
+
+def update_packages() -> None:
+	try:
+		newVersion = False
+		db_packages = __load_required_package_versions()
+		print(ROOT_PATH)
+		if sys.prefix not in PYENV.__str__():
+			print(f"You are probably running a new installation or you are not using your virtual env {PYENV}")
+			return newVersion
+
+		print(f"> Checking for dependencies versions ==> WAIT")
+		for package in db_packages:
+			newVersion = False
+			_required_version = package.version
+			_installed_version: str = None
+			output = check_output([PIPENV, 'show', package.name])
+			for line in output.decode().splitlines():
+				if line.startswith('Version:'):
+					_installed_version = line.split(':')[1].strip()
+					break
+
+			required_version = tuple(map(int, _required_version.split('.')))
+			installed_version = tuple(map(int, _installed_version.split('.')))
+
+			if required_version > installed_version:
+				print(f'> New version of {package.name} is available {installed_version} ==> {required_version}')
+				newVersion = True
+
+			if newVersion:
+				check_call([PIPENV, 'install', '--upgrade', package.name])
+
+		print(f"> Dependencies versions ==> OK")
+		return newVersion
+
+	except CalledProcessError:
+		print(f"[!] Package {package.name} not installed [!]")
+	except Exception as err:
+		print(f"UpdatePackage Error: {err}")
+		traceback.print_exc()
+
+def run_git_update() -> None:
+	check_call(['git', 'pull', 'origin', 'main'])
+
 def check_python_requirement():
     if PYTHON_SYSTEM_VERSION < PYTHON_REQUIRED_VERSION:
 	    raise RuntimeError(f"Your Python Version is not meeting the requirement, System Version: {PYTHON_SYSTEM_VERSION} < Required Version {PYTHON_REQUIRED_VERSION}")
@@ -64,6 +140,10 @@ def main():
 	if args.install:
 		create_service_file()
 		sys.exit(0)
+	
+	if args.git_update:
+		run_git_update()
+		sys.exit(0)
 
 
 if __name__ == "__main__":