feat: add server and load balancer private_net_for helper method

Author: jooola

hcloud/load_balancers/domain.py

@@ -13,7 +13,7 @@
     from ..load_balancer_types import BoundLoadBalancerType
     from ..locations import BoundLocation
     from ..metrics import Metrics
-    from ..networks import BoundNetwork
+    from ..networks import BoundNetwork, Network
     from ..servers import BoundServer
     from .client import BoundLoadBalancer
 
@@ -107,6 +107,16 @@ def __init__(
         self.ingoing_traffic = ingoing_traffic
         self.included_traffic = included_traffic
 
+    def private_net_for(self, network: BoundNetwork | Network) -> PrivateNet | None:
+        """
+        Returns the load balancer's network attachment information in the given Network,
+        and None if no attachment was found.
+        """
+        for o in self.private_net or []:
+            if o.network.id == network.id:
+                return o
+        return None
+
 
 class LoadBalancerService(BaseDomain):
     """LoadBalancerService Domain

hcloud/servers/domain.py

@@ -14,7 +14,7 @@
     from ..images import BoundImage
     from ..isos import BoundIso
     from ..metrics import Metrics
-    from ..networks import BoundNetwork
+    from ..networks import BoundNetwork, Network
     from ..placement_groups import BoundPlacementGroup
     from ..primary_ips import BoundPrimaryIP, PrimaryIP
     from ..server_types import BoundServerType
@@ -154,6 +154,16 @@ def __init__(
         self.primary_disk_size = primary_disk_size
         self.placement_group = placement_group
 
+    def private_net_for(self, network: BoundNetwork | Network) -> PrivateNet | None:
+        """
+        Returns the server's network attachment information in the given Network,
+        and None if no attachment was found.
+        """
+        for o in self.private_net or []:
+            if o.network.id == network.id:
+                return o
+        return None
+
 
 class CreateServerResponse(BaseDomain):
     """Create Server Response Domain

tests/unit/load_balancers/test_domain.py

@@ -21,6 +21,7 @@
     PrivateNet,
     PublicNetwork,
 )
+from hcloud.networks import Network
 
 
 @pytest.mark.parametrize(
@@ -56,3 +57,19 @@ class TestLoadBalancers:
     def test_created_is_datetime(self):
         lb = LoadBalancer(id=1, created="2016-01-30T23:50+00:00")
         assert lb.created == datetime.datetime(2016, 1, 30, 23, 50, tzinfo=timezone.utc)
+
+    def test_private_net_for(self):
+        network1 = Network(id=1)
+        network2 = Network(id=2)
+        network3 = Network(id=3)
+
+        private_net1 = PrivateNet(network=network1, ip="127.0.0.1")
+        private_net2 = PrivateNet(network=network2, ip="127.0.0.1")
+
+        server = LoadBalancer(
+            id=42,
+            private_net=[private_net1, private_net2],
+        )
+
+        assert server.private_net_for(network1) == private_net1
+        assert server.private_net_for(network3) is None

tests/unit/servers/test_domain.py

@@ -5,6 +5,7 @@
 
 import pytest
 
+from hcloud.networks import Network
 from hcloud.servers import (
     IPv4Address,
     IPv6Network,
@@ -46,3 +47,23 @@ def test_created_is_datetime(self):
         assert server.created == datetime.datetime(
             2016, 1, 30, 23, 50, tzinfo=timezone.utc
         )
+
+    def test_private_net_for(self):
+        network1 = Network(id=1)
+        network2 = Network(id=2)
+        network3 = Network(id=3)
+
+        private_net1 = PrivateNet(
+            network=network1, ip="127.0.0.1", alias_ips=[], mac_address=""
+        )
+        private_net2 = PrivateNet(
+            network=network2, ip="127.0.0.1", alias_ips=[], mac_address=""
+        )
+
+        server = Server(
+            id=42,
+            private_net=[private_net1, private_net2],
+        )
+
+        assert server.private_net_for(network1) == private_net1
+        assert server.private_net_for(network3) is None