feat: add domain attribute type hints to bound models (#300)

hetznercloud · Sep 25, 2023 · 6d46d06 · 6d46d06
1 parent 55d2b20
commit 6d46d06
Show file tree

Hide file tree

Showing 18 changed files with 18 additions and 18 deletions.
diff --git a/hcloud/actions/client.py b/hcloud/actions/client.py
@@ -11,7 +11,7 @@
     from .._client import Client
 
 
-class BoundAction(BoundModelBase):
+class BoundAction(BoundModelBase, Action):
     _client: ActionsClient
 
     model = Action

diff --git a/hcloud/certificates/client.py b/hcloud/certificates/client.py
@@ -15,7 +15,7 @@
     from .._client import Client
 
 
-class BoundCertificate(BoundModelBase):
+class BoundCertificate(BoundModelBase, Certificate):
     _client: CertificatesClient
 
     model = Certificate

diff --git a/hcloud/datacenters/client.py b/hcloud/datacenters/client.py
@@ -11,7 +11,7 @@
     from .._client import Client
 
 
-class BoundDatacenter(BoundModelBase):
+class BoundDatacenter(BoundModelBase, Datacenter):
     _client: DatacentersClient
 
     model = Datacenter

diff --git a/hcloud/firewalls/client.py b/hcloud/firewalls/client.py
@@ -17,7 +17,7 @@
     from .._client import Client
 
 
-class BoundFirewall(BoundModelBase):
+class BoundFirewall(BoundModelBase, Firewall):
     _client: FirewallsClient
 
     model = Firewall

diff --git a/hcloud/floating_ips/client.py b/hcloud/floating_ips/client.py
@@ -13,7 +13,7 @@
     from ..servers import BoundServer, Server
 
 
-class BoundFloatingIP(BoundModelBase):
+class BoundFloatingIP(BoundModelBase, FloatingIP):
     _client: FloatingIPsClient
 
     model = FloatingIP

diff --git a/hcloud/images/client.py b/hcloud/images/client.py
@@ -10,7 +10,7 @@
     from .._client import Client
 
 
-class BoundImage(BoundModelBase):
+class BoundImage(BoundModelBase, Image):
     _client: ImagesClient
 
     model = Image

diff --git a/hcloud/isos/client.py b/hcloud/isos/client.py
@@ -10,7 +10,7 @@
     from .._client import Client
 
 
-class BoundIso(BoundModelBase):
+class BoundIso(BoundModelBase, Iso):
     _client: IsosClient
 
     model = Iso

diff --git a/hcloud/load_balancer_types/client.py b/hcloud/load_balancer_types/client.py
@@ -9,7 +9,7 @@
     from .._client import Client
 
 
-class BoundLoadBalancerType(BoundModelBase):
+class BoundLoadBalancerType(BoundModelBase, LoadBalancerType):
     _client: LoadBalancerTypesClient
 
     model = LoadBalancerType

diff --git a/hcloud/load_balancers/client.py b/hcloud/load_balancers/client.py
@@ -34,7 +34,7 @@
     from ..networks import Network
 
 
-class BoundLoadBalancer(BoundModelBase):
+class BoundLoadBalancer(BoundModelBase, LoadBalancer):
     _client: LoadBalancersClient
 
     model = LoadBalancer

diff --git a/hcloud/locations/client.py b/hcloud/locations/client.py
@@ -9,7 +9,7 @@
     from .._client import Client
 
 
-class BoundLocation(BoundModelBase):
+class BoundLocation(BoundModelBase, Location):
     _client: LocationsClient
 
     model = Location

diff --git a/hcloud/networks/client.py b/hcloud/networks/client.py
@@ -10,7 +10,7 @@
     from .._client import Client
 
 
-class BoundNetwork(BoundModelBase):
+class BoundNetwork(BoundModelBase, Network):
     _client: NetworksClient
 
     model = Network

diff --git a/hcloud/placement_groups/client.py b/hcloud/placement_groups/client.py
@@ -10,7 +10,7 @@
     from .._client import Client
 
 
-class BoundPlacementGroup(BoundModelBase):
+class BoundPlacementGroup(BoundModelBase, PlacementGroup):
     _client: PlacementGroupsClient
 
     model = PlacementGroup

diff --git a/hcloud/primary_ips/client.py b/hcloud/primary_ips/client.py
@@ -11,7 +11,7 @@
     from ..datacenters import BoundDatacenter, Datacenter
 
 
-class BoundPrimaryIP(BoundModelBase):
+class BoundPrimaryIP(BoundModelBase, PrimaryIP):
     _client: PrimaryIPsClient
 
     model = PrimaryIP

diff --git a/hcloud/server_types/client.py b/hcloud/server_types/client.py
@@ -9,7 +9,7 @@
     from .._client import Client
 
 
-class BoundServerType(BoundModelBase):
+class BoundServerType(BoundModelBase, ServerType):
     _client: ServerTypesClient
 
     model = ServerType

diff --git a/hcloud/servers/client.py b/hcloud/servers/client.py
@@ -42,7 +42,7 @@
     from .domain import ServerCreatePublicNetwork
 
 
-class BoundServer(BoundModelBase):
+class BoundServer(BoundModelBase, Server):
     _client: ServersClient
 
     model = Server

diff --git a/hcloud/ssh_keys/client.py b/hcloud/ssh_keys/client.py
@@ -9,7 +9,7 @@
     from .._client import Client
 
 
-class BoundSSHKey(BoundModelBase):
+class BoundSSHKey(BoundModelBase, SSHKey):
     _client: SSHKeysClient
 
     model = SSHKey

diff --git a/hcloud/volumes/client.py b/hcloud/volumes/client.py
@@ -13,7 +13,7 @@
     from ..servers import BoundServer, Server
 
 
-class BoundVolume(BoundModelBase):
+class BoundVolume(BoundModelBase, Volume):
     _client: VolumesClient
 
     model = Volume

diff --git a/tests/unit/core/test_client.py b/tests/unit/core/test_client.py
@@ -20,7 +20,7 @@ def __init__(self, id, name="", description=""):
                 self.name = name
                 self.description = description
 
-        class BoundModel(BoundModelBase):
+        class BoundModel(BoundModelBase, Model):
             model = Model
 
         return BoundModel