mathieu/ai-suite-rocm-local
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

improve start/update/uninstall

Browse Source
This commit is contained in:
Mathieu Broillet 2024-08-29 12:50:18 +02:00
parent 33c7a0b674
commit 303f73433f
Signed by: mathieu
GPG Key ID: A08E484FE95074C1
3 changed files with 36 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
core/stack.py
View File

@ -41,12 +41,15 @@ class Stack:
def install(self):
self.check_for_broken_install()
self.create_venv()
self._install()
if self.is_installed():
self.update()
else:
self.check_for_broken_install()
self.create_venv()
self._install()
self.create_file('.installed', 'true')
logger.info(f"Installed {self.name}")
self.create_file('.installed', 'true')
logger.info(f"Installed {self.name}")
def _install(self):
pass
@ -66,17 +69,33 @@ class Stack:
def update(self, folder: str = 'webui'):
if self.is_installed():
status = self.status()
if status:
self.stop()
logger.info(f"Updating {self.name}")
self.git_pull(folder)
self._update()
if status:
self.start()
else:
logger.warning(f"Could not update {self.name} as {self.name} is not installed")
choices.any_key.ask()
def _update(self):
pass
def uninstall(self):
logger.info(f"Uninstalling {self.name}")
if self.status():
self.stop()
self.bash(f"rm -rf {self.path}")
def start(self):
if self.status():
logger.warning(f"{self.name} is already running")
if self.is_installed():
self._start()
else:
@ -90,6 +109,8 @@ class Stack:
if self.status():
logger.debug(f"Killing {self.name} with PID: {self.pid}")
psutil.Process(self.pid).kill()
else:
logger.warning(f"{self.name} is not running")
self.set_pid(None)
@ -158,7 +179,6 @@ class Stack:
return
else:
# TODO: attach to subprocess / redirect logs?
logger.info("Continuing without restarting...")
return
else:
logger.debug(f"Running command as daemon: {cmd}")

2
main.py
View File

@ -27,7 +27,7 @@ def load_services():
if __name__ == '__main__':
setup_logger(logging.DEBUG)
setup_logger(logging.INFO)
logger.info("Starting AI Suite for ROCM")
setup_config()

15
ui/interface.py
View File

@ -19,22 +19,25 @@ def handle_services(action, service):
service = loaded_services[service]
if action == "start":
logger.info(f"Starting service: {service.name}")
service.start()
elif action == "stop":
logger.info(f"Stopping service: {service.name}")
service.stop()
elif action == "install":
confirmation = choices.are_you_sure.ask()
if confirmation:
logger.info(f"Installing/updating service: {service.name}")
service.install()
elif action == "uninstall":
confirmation = choices.are_you_sure.ask()
if confirmation:
type_confirmation = questionary.text(f"Please type {service.id} to confirm uninstallation:")
if type_confirmation.ask() == service.id:
logger.info(f"Uninstalling service: {service.name}")
type_confirmation = questionary.text(f"Please type {service.id} to confirm uninstallation (or type cancel):")
value = type_confirmation.ask()
if value == "cancel":
questionary.print("Canceled", style="fg:ansired")
elif value != service.id:
questionary.print("Invalid input, please try again", style="fg:ansired")
elif value == service.id:
service.uninstall()
choices.any_key.ask()