make get_ipv4 generic over service type

Author: willcl-ark

src/warnet/backend/kubernetes_backend.py

@@ -567,8 +567,7 @@ def create_pod_object(
             ),
         )
 
-    def get_tank_ipv4(self, index: int) -> str | None:
-        pod_name = self.get_pod_name(index, ServiceType.BITCOIN)
+    def get_ipv4(self, pod_name: str, service_type: ServiceType) -> str | None:
         pod = self.get_pod(pod_name)
         if pod:
             return pod.status.pod_ip

src/warnet/warnet.py

@@ -10,6 +10,7 @@
 import networkx
 
 from .backend.kubernetes_backend import KubernetesBackend
+from .services import ServiceType
 from .tank import Tank
 from .utils import gen_config_dir, load_schema, validate_graph_schema
 
@@ -132,7 +133,8 @@ def from_network(cls, network_name):
         if "services" in self.graph.graph:
             self.services = self.graph.graph["services"].split()
         for tank in self.tanks:
-            tank.ipv4 = self.container_interface.get_tank_ipv4(tank.index)
+            pod_name = self.container_interface.get_pod_name(tank.index, ServiceType.BITCOIN)
+            tank.ipv4 = self.container_interface.get_ipv4(pod_name, ServiceType.BITCOIN)
         return self
 
     def tanks_from_graph(self):