get IPs from API

roles/5g/base/files/gen_oai.py

@@ -2,7 +2,24 @@ import os
 import yaml
 from jinja2 import Environment, FileSystemLoader, BaseLoader
 import ipaddr
-    
+
+import requests
+
+def getIP(cluster="central"):
+  url = f"http://192.0.2.1:8000/ip/{cluster}"
+
+  headers = {
+    'Accept': 'application/json',
+    'Bearer': os.environ['token']
+  }
+
+  response = requests.request("GET", url, headers=headers, data={})
+  if response.status_code != requests.codes.OK:
+    raise Exception(response.json()["detail"])
+  
+  info = response.json()
+  return info['ip'], info['prefixlen']
+
 def to_yaml(data):
     return yaml.safe_dump(data).strip()
 
@@ -12,46 +29,53 @@ def createFile(content, filename):
 ###########
 
 def genIPs(gcn):
-  net = ipaddr.IPNetwork(gcn['multus']['network'])
-  base_ip = net.ip + 2
+  # net = ipaddr.IPNetwork(gcn['multus']['network'])
+  # base_ip = net.ip + 2
   hostInterface = gcn['multus']['hostInterface']
 
+  reserved = list()
+  for i in range(0, 18):
+    reserved.append(getIP())
+
+  prefixlen = reserved[0][1]
+
   ips = {
     "amf": {
-      "n2": {'ip': str(base_ip), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface}
+      "n2": {'ip': reserved[0][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface}
+      # "n2": {'ip': str(base_ip), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface}
     },
     "smf": {
-      "n4": {'ip': str(base_ip + 1), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface}
+      "n4": {'ip': reserved[1][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface}
     },
     "upf": {
-      "n3": {'ip': str(base_ip + 2), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface},
-      "n4": {'ip': str(base_ip + 3), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface},
-      "n6": {'ip': str(base_ip + 4), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface}
+      "n3": {'ip': reserved[2][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface},
+      "n4": {'ip': reserved[3][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface},
+      "n6": {'ip': reserved[4][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface}
     },
     "gnb": {
-      "n2": {'ip': str(base_ip + 5), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface},
-      "n3": {'ip': str(base_ip + 6), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface},
+      "n2": {'ip': reserved[5][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface},
+      "n3": {'ip': reserved[6][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface},
     },
     "du": {
-      "f1": {'ip': str(base_ip + 7), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface}
+      "f1": {'ip': reserved[7][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface}
     }, #
     "cu": {
-      "f1": {'ip': str(base_ip + 8), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface},
-      "n2": {'ip': str(base_ip + 9), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface},
-      "n3": {'ip': str(base_ip + 10), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface}
+      "f1": {'ip': reserved[8][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface},
+      "n2": {'ip': reserved[9][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface},
+      "n3": {'ip': reserved[10][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface}
     },
     "cucp": {
-      "e1": {'ip': str(base_ip + 11), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface},
-      "n2": {'ip': str(base_ip + 12), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface},
-      "f1": {'ip': str(base_ip + 8), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface}
+      "e1": {'ip': reserved[11][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface},
+      "n2": {'ip': reserved[12][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface},
+      "f1": {'ip': reserved[8][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface}
     },
     "cuup": {
-      "e1": {'ip': str(base_ip + 14), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface},
-      "n3": {'ip': str(base_ip + 15), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface},
-      "f1": {'ip': str(base_ip + 16), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface}
+      "e1": {'ip': reserved[14][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface},
+      "n3": {'ip': reserved[15][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface},
+      "f1": {'ip': reserved[16][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface}
     },
     "trafficserver": {
-      'ip': str(base_ip + 17), 'prefixlen': net.prefixlen, 'hostInterface': hostInterface
+      'ip': reserved[17][0], 'prefixlen': prefixlen, 'hostInterface': hostInterface
     },
     "nrf": {
       "loadBalancerIP": gcn['core']['nrfLoadBalancerIP']