Implemented support for VPC DBaaS Integration

Author: ezilber-akamai

linode_api4/groups/database.py

@@ -10,6 +10,7 @@
     Database,
     DatabaseEngine,
     DatabaseType,
+    DatabasePrivateNetwork,
     MySQLDatabase,
     PostgreSQLDatabase,
     drop_null_keys,
@@ -126,6 +127,7 @@ def mysql_create(
         engine,
         ltype,
         engine_config: Union[MySQLDatabaseConfigOptions, Dict[str, Any]] = None,
+        private_network: Union[DatabasePrivateNetwork, Dict[str, Any]] = None,
         **kwargs,
     ):
         """
@@ -159,6 +161,8 @@ def mysql_create(
         :type ltype: str or Type
         :param engine_config: The configuration options for this MySQL cluster
         :type engine_config: Dict[str, Any] or MySQLDatabaseConfigOptions
+        :param private_network: The private network settings to use for this cluster
+        :type private_network: Dict[str, Any] or DatabasePrivateNetwork
         """
 
         params = {
@@ -167,6 +171,7 @@ def mysql_create(
             "engine": engine,
             "type": ltype,
             "engine_config": engine_config,
+            "private_network": private_network,
         }
         params.update(kwargs)
 
@@ -262,6 +267,7 @@ def postgresql_create(
         engine_config: Union[
             PostgreSQLDatabaseConfigOptions, Dict[str, Any]
         ] = None,
+        private_network: Union[DatabasePrivateNetwork, Dict[str, Any]] = None,
         **kwargs,
     ):
         """
@@ -295,6 +301,8 @@ def postgresql_create(
         :type ltype: str or Type
         :param engine_config: The configuration options for this PostgreSQL cluster
         :type engine_config: Dict[str, Any] or PostgreSQLDatabaseConfigOptions
+        :param private_network: The private network settings to use for this cluster
+        :type private_network: Dict[str, Any] or DatabasePrivateNetwork
         """
 
         params = {
@@ -303,6 +311,7 @@ def postgresql_create(
             "engine": engine,
             "type": ltype,
             "engine_config": engine_config,
+            "private_network": private_network,
         }
         params.update(kwargs)
 

linode_api4/objects/database.py

@@ -73,6 +73,15 @@ def invalidate(self):
 
         Base.invalidate(self)
 
+@dataclass
+class DatabasePrivateNetwork(JSONObject):
+    """
+    DatabasePrivateNetwork is used to specify
+    a Database Cluster's private network settings during its creation.
+    """
+    vpc_id: int
+    subnet_id: int
+    public_access: bool
 
 @deprecated(
     reason="Backups are not supported for non-legacy database clusters."
@@ -304,6 +313,7 @@ class MySQLDatabase(Base):
         "engine_config": Property(
             mutable=True, json_object=MySQLDatabaseConfigOptions
         ),
+        "private_network": Property(mutable=True, json_object=DatabasePrivateNetwork),
     }
 
     @property
@@ -470,6 +480,7 @@ class PostgreSQLDatabase(Base):
         "engine_config": Property(
             mutable=True, json_object=PostgreSQLDatabaseConfigOptions
         ),
+        "private_network": Property(mutable=True, json_object=DatabasePrivateNetwork),
     }
 
     @property
@@ -636,6 +647,7 @@ class Database(Base):
         "updated": Property(),
         "updates": Property(),
         "version": Property(),
+        "private_network": Property(json_object=DatabasePrivateNetwork),
     }
 
     @property