faerbit/dcqc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add PartOf

Browse Source
This commit is contained in:
Fabian 2024-11-04 00:58:27 +01:00
parent a429621e42
commit bfabf37420
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
main.py
View File

@ -39,6 +39,9 @@ def enforce_list(key: str, dict_: dict):
abort(f"{key} is not a list. Offending dict: {dict_}") abort(f"{key} is not a list. Offending dict: {dict_}")
def add_part_of(args, unit_file: IniFile):
unit_file["Unit"]["PartOf"] = args.target
def write_file(ini_file: IniFile, path: Path): def write_file(ini_file: IniFile, path: Path):
with open(path, "w", encoding="utf-8") as f: with open(path, "w", encoding="utf-8") as f:
ini_file.write(f) ini_file.write(f)
@ -79,6 +82,7 @@ def write_network_units(args, yaml_dict):
unit_file = IniFile() unit_file = IniFile()
unit_file["Unit"] = {} unit_file["Unit"] = {}
unit_file["Unit"]["Description"] = f"{network_name.capitalize()} network" unit_file["Unit"]["Description"] = f"{network_name.capitalize()} network"
add_part_of(args, unit_file)
unit_file["Network"] = {} unit_file["Network"] = {}
if "name" in network: if "name" in network:
@ -105,6 +109,7 @@ def write_service_units(args, yaml_dict):
unit_file = IniFile() unit_file = IniFile()
unit_file["Unit"] = {} unit_file["Unit"] = {}
unit_file["Unit"]["Description"] = f"{service_name.capitalize()} container" unit_file["Unit"]["Description"] = f"{service_name.capitalize()} container"
add_part_of(args, unit_file)
# currently decreases reliability instead of increasing it # currently decreases reliability instead of increasing it
# if "depends_on" in service: # if "depends_on" in service:
# enforce_list("depends_on", service) # enforce_list("depends_on", service)
@ -180,6 +185,8 @@ def parse_args():
parser.add_argument("compose_file") parser.add_argument("compose_file")
parser.add_argument("-o", "--output-dir", help="output directory for unit files", parser.add_argument("-o", "--output-dir", help="output directory for unit files",
default="/etc/containers/systemd", type=Path) default="/etc/containers/systemd", type=Path)
parser.add_argument("-t", "--target",
help="Add PartOf dependency to all unit files", default="containers.target")
return parser.parse_args() return parser.parse_args()