NextERP-scripts/services/odoo/module.py
2025-04-03 17:05:42 +07:00

72 lines
2.9 KiB
Python

from services.git.handler import GitHandler
from services.odoo.connection import OdooConnection
import subprocess
class OdooModuleManager:
def __init__(self, config_path: str = "config/settings.yaml"):
self.config = OdooConnection(config_path)
def _manage_module(self, action: str, instance_name: str = None) -> None:
"""Generic method to install, uninstall, or upgrade modules."""
if action not in ["install", "uninstall", "upgrade"]:
raise ValueError(f"Invalid action: {action}")
self.config.connect(instance_name)
for instance in self.config.get_instances():
if instance_name and instance["name"] != instance_name:
continue
print(f"Processing instance: {instance['name']}")
module_names = instance.get("module_names", [])
if not module_names:
print(
f"No modules specified for {instance['name']}, skipping {action}."
)
continue
for module_name in module_names:
print(
f"{action.capitalize()}ing module: {module_name} in {instance['name']}"
)
try:
module_ids = self.config.execute(
instance["name"],
"ir.module.module",
"search",
[("name", "=", module_name), ("state", "=", "installed")],
)
if module_ids:
button_action = f"button_immediate_{action}"
self.config.execute(
instance["name"],
"ir.module.module",
button_action,
[module_ids],
)
print(
f"Module {module_name} {action}ed successfully in {instance['name']}"
)
else:
print(
f"Module {module_name} not found or not installed in {instance['name']}"
)
except Exception as e:
print(
f"Failed to {action} {module_name} in {instance['name']}: {e}"
)
def install(self, instance_name: str = None) -> None:
"""Install multiple modules for the specified instance(s)."""
self._manage_module("install", instance_name)
def uninstall(self, instance_name: str = None) -> None:
"""Uninstall multiple modules for the specified instance(s)."""
self._manage_module("uninstall", instance_name)
def upgrade(self, instance_name: str = None) -> None:
"""Upgrade multiple modules for the specified instance(s)."""
self._manage_module("upgrade", instance_name)