28 lines
984 B
Python
28 lines
984 B
Python
import yaml
|
|
|
|
def update_hosts_yaml(file_path="fastprod/inventory/hosts.yaml", items=None):
|
|
if items:
|
|
with open(file_path, 'w') as yaml_file:
|
|
yaml.safe_dump(items, yaml_file)
|
|
return True
|
|
|
|
def add_item_to_hosts_yaml(file_path="fastprod/inventory/hosts.yaml", save=True, item=None):
|
|
with open(file_path, 'r') as yaml_file:
|
|
hosts = yaml.safe_load(yaml_file)
|
|
|
|
new_hosts = hosts.copy()
|
|
new_hosts[item.get('data').get('device_name')] = item
|
|
|
|
if save:
|
|
update_hosts_yaml(items=new_hosts)
|
|
|
|
return new_hosts[item.get('data').get('device_name')]
|
|
|
|
def delete_item_from_hosts_yaml(yaml_file_path="fastprod/inventory/hosts.yaml", save=True,item=None, ):
|
|
with open(yaml_file_path,'r') as yaml_file:
|
|
hosts = yaml.safe_load(yaml_file)
|
|
new_hosts = hosts.copy()
|
|
del new_hosts[item.get('data').get('device_name')]
|
|
if save:
|
|
update_hosts_yaml(items=new_hosts)
|
|
return True |