Narcissus/meshbook
2
0
Fork 0
mirror of https://github.com/DaanSelen/meshbook.git synced 2026-02-20 16:32:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added multiple group and device targeting with 'groups' and 'devices'. Also more error handling.

Browse Source
This commit is contained in:
Daan Selen
2025-02-13 14:22:03 +01:00
parent 77b1cfc73c
commit 4f75969ed8
1 changed files with 56 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
meshbook.py
View File

@@ -141,21 +141,53 @@ async def gather_targets(playbook: dict, group_list: dict[str, list[dict]], os_c
target_list = [] target_list = []
target_os = playbook.get("target_os") target_os = playbook.get("target_os")
if "device" in playbook: async def process_device_or_group(pseudo_target, group_list, os_categories, target_os) -> list[str]:
pseudo_target = playbook["device"] """Helper function to process devices or groups."""
matched_devices = []
for group in group_list: for group in group_list:
for device in group_list[group]: for device in group_list[group]:
if device["device_name"] == pseudo_target: if device["device_name"] == pseudo_target:
matched_devices = await filter_targets([device], os_categories, target_os) matched_devices.append(device)
if matched_devices:
return await filter_targets(matched_devices, os_categories, target_os)
return []
match playbook:
case {"device": pseudo_target}: # Single device target
if isinstance(pseudo_target, str):
matched_devices = await process_device_or_group(pseudo_target, group_list, os_categories, target_os)
target_list.extend(matched_devices)
else:
console("\033[91mPlease use devices (Notice the 'S') for multiple devices.\x1B[0m", True)
case {"devices": pseudo_target}: # List of devices
if isinstance(pseudo_target, list):
for sub_pseudo_device in pseudo_target:
matched_devices = await process_device_or_group(sub_pseudo_device, group_list, os_categories, target_os)
target_list.extend(matched_devices) target_list.extend(matched_devices)
elif "group" in playbook: else:
pseudo_target = playbook["group"] console("\033[91mThe 'devices' method is being used, but only one string is given. Did you mean 'device'?\x1B[0m", True)
if pseudo_target in group_list: case {"group": pseudo_target}: # Single group target
matched_devices = await filter_targets(group_list[pseudo_target], os_categories, target_os) if isinstance(pseudo_target, str) and pseudo_target in group_list:
target_list.extend(matched_devices) matched_devices = await filter_targets(group_list[pseudo_target], os_categories, target_os)
target_list.extend(matched_devices)
else:
console("\033[91mPlease use groups (Notice the 'S') for multiple groups.\x1B[0m", True)
case {"groups": pseudo_target}: # List of groups
if isinstance(pseudo_target, list):
for sub_pseudo_target in pseudo_target:
if sub_pseudo_target in group_list:
matched_devices = await filter_targets(group_list[sub_pseudo_target], os_categories, target_os)
target_list.extend(matched_devices)
else:
console("\033[91mThe 'groups' method is being used, but only one string is given. Did you mean 'group'?\x1B[0m", True)
return target_list return target_list
@@ -220,10 +252,10 @@ async def main():
console("Operating System Categorisation file: " + args.oscategories) console("Operating System Categorisation file: " + args.oscategories)
console("Congiguration file: " + args.conf) console("Congiguration file: " + args.conf)
if "device" in playbook: if "device" in playbook:
console("Target device: " + playbook["device"]) console("Target device: " + str(playbook["device"]))
elif "group" in playbook: elif "group" in playbook:
console("Target group: " + playbook["group"]) console("Target group: " + str(playbook["group"]))
console("Grace: " + str((not args.nograce))) # Negation of bool for correct explanation console("Grace: " + str((not args.nograce))) # Negation of bool for correct explanation
console("Silent: False") # Can be pre-defined because if silent flag was passed then none of this would be printed. console("Silent: False") # Can be pre-defined because if silent flag was passed then none of this would be printed.
@@ -249,7 +281,20 @@ async def main():
else: else:
console(("-" * 40)) console(("-" * 40))
target_name = playbook["group"] if "group" in playbook else playbook["device"] # Quickly get the name.
match playbook:
case {"group": candidate_target_name}:
target_name = candidate_target_name
case {"groups": candidate_target_name}:
target_name = str(candidate_target_name)
case {"device": candidate_target_name}:
target_name = candidate_target_name
case {"devices": candidate_target_name}:
target_name = str(candidate_target_name)
console(("\033[91mExecuting playbook on the target(s): " + target_name + ".\x1B[0m")) console(("\033[91mExecuting playbook on the target(s): " + target_name + ".\x1B[0m"))
if not args.nograce: if not args.nograce: