47 lines
1 KiB
Python
47 lines
1 KiB
Python
from typing import List, Generator
|
|
|
|
import attr
|
|
|
|
from .service import Service
|
|
from .yaml import YAML
|
|
|
|
|
|
@attr.s
|
|
class Services:
|
|
content: List[Service] = attr.ib()
|
|
|
|
def __str__(self) -> str:
|
|
return YAML().dump({
|
|
"services": {
|
|
service.name: service.content
|
|
for service in self.content
|
|
}
|
|
}).strip()
|
|
|
|
def __bool__(self) -> bool:
|
|
return bool(self.content)
|
|
|
|
# def copy_configs(self):
|
|
# configs = (
|
|
# config
|
|
# for service in self.content
|
|
# for config in service.configs
|
|
# )
|
|
#
|
|
# print(list(configs))
|
|
#
|
|
# # Rootkit("rsync").
|
|
|
|
@property
|
|
def names(self) -> Generator[str, None, None]:
|
|
return (
|
|
service.name
|
|
for service in self.content
|
|
)
|
|
|
|
def filter_existing(self, service_names: List[str]) -> "Services":
|
|
return Services([
|
|
service
|
|
for service in self.content
|
|
if service.name in service_names
|
|
])
|