import logging import os from ..._constants import CONF_DIRECTORY_NAME from ...config import LoadedConfig class Project: __name = None def __init__(self, name): self.__name = name @classmethod def from_file_name(cls, file_name): if os.path.isdir(file_name): config = LoadedConfig.get() if file_name.endswith(config['markers:disabled']): file_name = file_name[:-len(config['markers:disabled'])] if file_name.endswith(config['markers:project']): file_name = file_name[:-len(config['markers:project'])] return cls(file_name) return None def get_name(self): return self.__name def dir_name(self): if self.is_enabled(): return self.enabled_dir_name() elif self.is_disabled(): return self.disabled_dir_name() else: return None def enabled_dir_name(self): return f"{self.__name}{LoadedConfig.get()['markers:project']}" def disabled_dir_name(self): return f"{self.enabled_dir_name()}{LoadedConfig.get()['markers:disabled']}" def conf_dir_name(self): return os.path.join(self.dir_name(), CONF_DIRECTORY_NAME) def compose_file_name(self): return os.path.join(self.dir_name(), 'docker-compose.yml') def target_dir_name(self): return os.path.join(LoadedConfig.get()['runtime:storage'], self.enabled_dir_name()) def exists(self): return os.path.isdir(self.enabled_dir_name()) or os.path.isdir(self.disabled_dir_name()) def is_enabled(self): return os.path.isdir(self.enabled_dir_name()) def is_disabled(self): return os.path.isdir(self.disabled_dir_name()) def has_configs(self): return os.path.isdir(self.conf_dir_name()) def enable(self): if self.is_disabled(): logging.info(f"Enabling project '{self.get_name()}'") os.rename(self.dir_name(), self.enabled_dir_name()) elif self.is_enabled(): logging.warning(f"Project '{self.get_name()}' is enabled!") else: logging.warning(f"Project '{self.get_name()}' not found in instance!") return False return True def disable(self): if self.is_enabled(): logging.info(f"Disabling project '{self.get_name()}'") os.rename(self.dir_name(), self.disabled_dir_name()) elif self.is_disabled(): logging.warning(f"Project '{self.get_name()}' is disabled!") else: logging.warning(f"Project '{self.get_name()}' not found in instance!") return False return True class Projects: __projects = None def __getitem__(self, item): return self.__projects[item] def __str__(self): return str([ project.get_name() for project in self.__projects ]) @classmethod def from_names(cls, project_names): result = cls() result.__projects = [ Project(name) for name in project_names if isinstance(name, str) ] return result @classmethod def from_projects(cls, projects): result = cls() result.__projects = [ project for project in projects if isinstance(project, Project) ] return result @classmethod def from_dir(cls): return cls.from_projects([ Project.from_file_name(file_name) for file_name in os.listdir() ]) @classmethod def from_args(cls, args): if args is not None and 'projects' in args: if isinstance(args.projects, list) and args.projects: return cls.from_names(args.projects) elif isinstance(args.projects, str): return cls.from_names([args.projects]) return cls() def empty(self): return not self.__projects def filter_exists(self): result = Projects() result.__projects = [ project for project in self.__projects if project.exists() ] return result def filter_enabled(self): result = Projects() result.__projects = [ project for project in self.__projects if project.is_enabled() ] return result def filter_disabled(self): result = Projects() result.__projects = [ project for project in self.__projects if project.is_disabled() ] return result