Projet

fastprod_backend/fastprod/utils/inventory.py

@@ -0,0 +1,28 @@
+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