mirror of
https://github.com/yavook/kiwi-scp.git
synced 2024-11-22 04:43:00 +00:00
"kiwi cmd"
This commit is contained in:
parent
2241aa500f
commit
7ee7d5f300
2 changed files with 80 additions and 6 deletions
32
kiwi_scp/commands/cmd_cmd.py
Normal file
32
kiwi_scp/commands/cmd_cmd.py
Normal file
|
@ -0,0 +1,32 @@
|
||||||
|
from typing import Tuple
|
||||||
|
|
||||||
|
import click
|
||||||
|
|
||||||
|
from .cli import KiwiCommand, KiwiCommandType
|
||||||
|
from .decorators import kiwi_command
|
||||||
|
from ..executable import COMPOSE_EXE
|
||||||
|
from ..instance import Instance, Project
|
||||||
|
|
||||||
|
|
||||||
|
@click.argument(
|
||||||
|
"compose_args",
|
||||||
|
metavar="[ARG]...",
|
||||||
|
nargs=-1,
|
||||||
|
)
|
||||||
|
@click.argument(
|
||||||
|
"compose_cmd",
|
||||||
|
metavar="CMD",
|
||||||
|
)
|
||||||
|
@kiwi_command(
|
||||||
|
"cmd",
|
||||||
|
KiwiCommandType.PROJECT,
|
||||||
|
short_help="Run docker-compose command",
|
||||||
|
)
|
||||||
|
class CMD(KiwiCommand):
|
||||||
|
"""Run raw docker-compose command in a project"""
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def run_for_existing_project(cls, instance: Instance, project: Project, compose_cmd: str = None,
|
||||||
|
compose_args: Tuple[str] = None) -> None:
|
||||||
|
if project.project_config.enabled:
|
||||||
|
COMPOSE_EXE.run([compose_cmd, *compose_args], **project.process_kwargs)
|
|
@ -1,13 +1,13 @@
|
||||||
import functools
|
import functools
|
||||||
import re
|
import re
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Generator, List, Tuple, Optional
|
from typing import Generator, List, Optional, Dict, Any
|
||||||
|
|
||||||
import attr
|
import attr
|
||||||
from ruamel.yaml.comments import CommentedMap
|
from ruamel.yaml.comments import CommentedMap
|
||||||
|
|
||||||
from ._constants import COMPOSE_FILE_NAME
|
from ._constants import COMPOSE_FILE_NAME, CONF_DIRECTORY_NAME
|
||||||
from .config import KiwiConfig
|
from .config import KiwiConfig, ProjectConfig
|
||||||
from .misc import YAML
|
from .misc import YAML
|
||||||
|
|
||||||
_RE_CONFDIR = re.compile(r"^\s*\$(?:CONFDIR|{CONFDIR})/+(.*)$", flags=re.UNICODE)
|
_RE_CONFDIR = re.compile(r"^\s*\$(?:CONFDIR|{CONFDIR})/+(.*)$", flags=re.UNICODE)
|
||||||
|
@ -57,6 +57,7 @@ class Services:
|
||||||
@attr.s
|
@attr.s
|
||||||
class Project:
|
class Project:
|
||||||
directory: Path = attr.ib()
|
directory: Path = attr.ib()
|
||||||
|
config: KiwiConfig = attr.ib()
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
@functools.lru_cache(maxsize=10)
|
@functools.lru_cache(maxsize=10)
|
||||||
|
@ -68,6 +69,34 @@ class Project:
|
||||||
def name(self) -> str:
|
def name(self) -> str:
|
||||||
return self.directory.name
|
return self.directory.name
|
||||||
|
|
||||||
|
@property
|
||||||
|
def project_config(self) -> ProjectConfig:
|
||||||
|
return self.config.get_project_config(self.name)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def process_kwargs(self) -> Dict[str, Any]:
|
||||||
|
directory: Path = self.directory
|
||||||
|
project_name: str = self.name
|
||||||
|
kiwi_hub_name: str = self.config.network.name
|
||||||
|
target_root_dir: Path = self.config.storage.directory
|
||||||
|
conf_dir: Path = target_root_dir.joinpath(CONF_DIRECTORY_NAME)
|
||||||
|
target_dir: Path = target_root_dir.joinpath(project_name)
|
||||||
|
|
||||||
|
result: Dict[str, Any] = {
|
||||||
|
"cwd": str(directory),
|
||||||
|
"env": {
|
||||||
|
"COMPOSE_PROJECT_NAME": project_name,
|
||||||
|
"KIWI_HUB_NAME": kiwi_hub_name,
|
||||||
|
"TARGETROOT": str(target_root_dir),
|
||||||
|
"CONFDIR": str(conf_dir),
|
||||||
|
"TARGETDIR": str(target_dir),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
result["env"].update(self.config.environment)
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def services(self) -> Services:
|
def services(self) -> Services:
|
||||||
yml = Project._parse_compose_file(self.directory)
|
yml = Project._parse_compose_file(self.directory)
|
||||||
|
@ -88,7 +117,20 @@ class Instance:
|
||||||
|
|
||||||
return KiwiConfig.from_directory(self.directory)
|
return KiwiConfig.from_directory(self.directory)
|
||||||
|
|
||||||
def get_project(self, project_name: str) -> Optional[Project]:
|
@staticmethod
|
||||||
for project in self.config.projects:
|
@functools.lru_cache(maxsize=None)
|
||||||
|
def __get_project(instance_directory: Path, project_name: str) -> Optional[Project]:
|
||||||
|
instance = Instance(instance_directory)
|
||||||
|
config = instance.config
|
||||||
|
|
||||||
|
for project in config.projects:
|
||||||
if project.name == project_name:
|
if project.name == project_name:
|
||||||
return Project(self.directory.joinpath(project.name))
|
return Project(
|
||||||
|
directory=instance_directory.joinpath(project.name),
|
||||||
|
config=config,
|
||||||
|
)
|
||||||
|
|
||||||
|
def get_project(self, project_name: str) -> Optional[Project]:
|
||||||
|
project = Instance.__get_project(self.directory, project_name)
|
||||||
|
project.instance = self
|
||||||
|
return project
|
||||||
|
|
Loading…
Reference in a new issue