diff --git a/src/pix360core/classes/__init__.py b/src/pix360core/classes/__init__.py index 60160eb..90dab0e 100644 --- a/src/pix360core/classes/__init__.py +++ b/src/pix360core/classes/__init__.py @@ -1,4 +1,4 @@ from .modules import BaseModule, DownloaderModule -from .exceptions import DownloadError, StitchingError, ConversionError +from .exceptions import DownloadError, StitchingError, ConversionError, InstallError from .http import HTTPRequest from .stitching import BaseStitcher, PILStitcher, BlenderStitcher, DEFAULT_CUBEMAP_TO_EQUIRECTANGULAR_STITCHER, DEFAULT_STITCHER \ No newline at end of file diff --git a/src/pix360core/classes/exceptions.py b/src/pix360core/classes/exceptions.py index 1238bef..933e1bc 100644 --- a/src/pix360core/classes/exceptions.py +++ b/src/pix360core/classes/exceptions.py @@ -11,4 +11,9 @@ class DownloadError(ConversionError): class StitchingError(ConversionError): """Generic error that occurred while attempting to stitch content """ + pass + +class InstallError(Exception): + """Generic error that occurred while attempting to install a module + """ pass \ No newline at end of file diff --git a/src/pix360core/installer.py b/src/pix360core/installer.py new file mode 100644 index 0000000..45f780d --- /dev/null +++ b/src/pix360core/installer.py @@ -0,0 +1,31 @@ +from .classes import HTTPRequest, InstallError + +import json +import subprocess + +class Installer: + MODULES_JSON = "https://git.kumi/-/snippets/131/raw/main/modules.json" + + def __init__(self): + self.available_modules = self.fetch_modules() + + def fetch_modules(self): + request = HTTPRequest(self.MODULES_JSON).open() + return json.loads(request.read().decode()) + + def install_all(self): + for module in self.available_modules: + try: + self.install_module(module) + except InstallError as e: + print(f"Error while installing module {module['name']}: {e}") + + def install_module(self, module): + try: + subprocess.run(["pip", "install", "-U", module["source"]], check=True) + except Exception as e: + raise InstallError(f"Error while installing module {module['name']}: {e}") from e + +if __name__ == "__main__": + installer = Installer() + installer.install_all() \ No newline at end of file diff --git a/src/pix360core/management/commands/installmodules.py b/src/pix360core/management/commands/installmodules.py new file mode 100644 index 0000000..9d8103f --- /dev/null +++ b/src/pix360core/management/commands/installmodules.py @@ -0,0 +1,14 @@ +from django.core.management.base import BaseCommand, CommandError + +from ...installer import Installer + +import logging + +class Command(BaseCommand): + help = 'Run the worker' + + def handle(self, *args, **options): + """Handle the command + """ + installer = Installer() + installer.install_all()