some terror with structures

src/make_res.py

@@ -0,0 +1,81 @@
+from check_correct import check_correct
+from parse import parse_csv_to_structure
+import sys
+from structures import Host, Wire, Interface
+import yaml
+
+
+"""
+meta:
+  id: "lab-02-basic-utils"
+  name: "Basic Network Utilities & Traffic Monitoring"
+  description: "Two hosts connected directly to each other."
+
+links:
+  - endpoints: ["PC1", "PC2"]
+
+nodes:
+  - name: PC1
+    role: host
+    interfaces:
+      - ip: "10.0.12.1/24"
+
+  - name: PC2
+    role: host
+    interfaces:
+      - ip: "10.0.12.2/24"
+"""
+
+
+def make_res(data, name):
+    res = {
+        "meta": {
+            "id": name,
+            "name": name,
+        },
+        "links": [],
+        "nodes": [],
+    }
+
+    host_int = {}
+
+    for key, value in data.items():
+        if value.get("type") == "host":
+            value["name"] = key
+            host = Host(value)
+            node_entry = {
+                "name": host.name,
+                "role": "host",
+                "interfaces": [
+                    {"ip": Interface(data[iface]).ip_address}
+                    for iface in host.interfaces
+                ],
+            }
+            res["nodes"].append(node_entry)
+
+            host_int[host.interfaces[0]] = host.name
+
+    for key, value in data.items():
+        if value.get("type") == "wire":
+            wire = Wire(value)
+            link_entry = {
+                "endpoints": [
+                    host_int[wire.endpoints[0]],
+                    host_int[wire.endpoints[1]],
+                ]
+            }
+            res["links"].append(link_entry)
+
+    return res
+
+
+if __name__ == "__main__":
+    data = parse_csv_to_structure(sys.argv[1])
+    print(data)
+    if check_correct(data):
+        print("Data is correct.")
+    res = make_res(data, sys.argv[1])
+    print(res)
+
+    with open("output.yaml", "w") as f:
+        yaml.dump(res, f)