googleapis · parthea · Jan 20, 2023 · Jan 19, 2023 · Jan 19, 2023 · Jan 20, 2023
diff --git a/google/cloud/compute/__init__.py b/google/cloud/compute/__init__.py
@@ -286,6 +286,8 @@
     AggregatedListVpnGatewaysRequest,
     AggregatedListVpnTunnelsRequest,
     AliasIpRange,
+    AllocationResourceStatus,
+    AllocationResourceStatusSpecificSKUAllocation,
     AllocationSpecificSKUAllocationAllocatedInstancePropertiesReservedDisk,
     AllocationSpecificSKUAllocationReservedInstanceProperties,
     AllocationSpecificSKUReservation,
@@ -1689,6 +1691,8 @@
     "AggregatedListVpnGatewaysRequest",
     "AggregatedListVpnTunnelsRequest",
     "AliasIpRange",
+    "AllocationResourceStatus",
+    "AllocationResourceStatusSpecificSKUAllocation",
     "AllocationSpecificSKUAllocationAllocatedInstancePropertiesReservedDisk",
     "AllocationSpecificSKUAllocationReservedInstanceProperties",
     "AllocationSpecificSKUReservation",

diff --git a/google/cloud/compute_v1/__init__.py b/google/cloud/compute_v1/__init__.py
@@ -184,6 +184,8 @@
     AggregatedListVpnGatewaysRequest,
     AggregatedListVpnTunnelsRequest,
     AliasIpRange,
+    AllocationResourceStatus,
+    AllocationResourceStatusSpecificSKUAllocation,
     AllocationSpecificSKUAllocationAllocatedInstancePropertiesReservedDisk,
     AllocationSpecificSKUAllocationReservedInstanceProperties,
     AllocationSpecificSKUReservation,
@@ -1501,6 +1503,8 @@
     "AggregatedListVpnGatewaysRequest",
     "AggregatedListVpnTunnelsRequest",
     "AliasIpRange",
+    "AllocationResourceStatus",
+    "AllocationResourceStatusSpecificSKUAllocation",
     "AllocationSpecificSKUAllocationAllocatedInstancePropertiesReservedDisk",
     "AllocationSpecificSKUAllocationReservedInstanceProperties",
     "AllocationSpecificSKUReservation",

diff --git a/google/cloud/compute_v1/services/accelerator_types/client.py b/google/cloud/compute_v1/services/accelerator_types/client.py
@@ -718,7 +718,7 @@ def list(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "AcceleratorTypesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/addresses/client.py b/google/cloud/compute_v1/services/addresses/client.py
@@ -1403,7 +1403,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "AddressesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/autoscalers/client.py b/google/cloud/compute_v1/services/autoscalers/client.py
@@ -1594,7 +1594,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "AutoscalersClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/backend_buckets/client.py b/google/cloud/compute_v1/services/backend_buckets/client.py
@@ -2154,7 +2154,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "BackendBucketsClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/backend_services/client.py b/google/cloud/compute_v1/services/backend_services/client.py
@@ -2885,7 +2885,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "BackendServicesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/disk_types/client.py b/google/cloud/compute_v1/services/disk_types/client.py
@@ -712,7 +712,7 @@ def list(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "DiskTypesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/disks/client.py b/google/cloud/compute_v1/services/disks/client.py
@@ -2839,7 +2839,7 @@ def test_iam_permissions(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "DisksClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/external_vpn_gateways/client.py b/google/cloud/compute_v1/services/external_vpn_gateways/client.py
@@ -1356,7 +1356,7 @@ def test_iam_permissions(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "ExternalVpnGatewaysClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/firewall_policies/client.py b/google/cloud/compute_v1/services/firewall_policies/client.py
@@ -3040,7 +3040,7 @@ def test_iam_permissions(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "FirewallPoliciesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/firewalls/client.py b/google/cloud/compute_v1/services/firewalls/client.py
@@ -1433,7 +1433,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "FirewallsClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/forwarding_rules/client.py b/google/cloud/compute_v1/services/forwarding_rules/client.py
@@ -1933,7 +1933,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "ForwardingRulesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/global_addresses/client.py b/google/cloud/compute_v1/services/global_addresses/client.py
@@ -1235,7 +1235,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "GlobalAddressesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/global_forwarding_rules/client.py b/google/cloud/compute_v1/services/global_forwarding_rules/client.py
@@ -1717,7 +1717,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "GlobalForwardingRulesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/global_network_endpoint_groups/client.py b/google/cloud/compute_v1/services/global_network_endpoint_groups/client.py
@@ -1669,7 +1669,7 @@ def list_network_endpoints(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "GlobalNetworkEndpointGroupsClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/global_operations/client.py b/google/cloud/compute_v1/services/global_operations/client.py
@@ -904,7 +904,7 @@ def wait(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "GlobalOperationsClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/global_organization_operations/client.py b/google/cloud/compute_v1/services/global_organization_operations/client.py
@@ -664,7 +664,7 @@ def list(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "GlobalOrganizationOperationsClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/global_public_delegated_prefixes/client.py b/google/cloud/compute_v1/services/global_public_delegated_prefixes/client.py
@@ -1269,7 +1269,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "GlobalPublicDelegatedPrefixesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/health_checks/client.py b/google/cloud/compute_v1/services/health_checks/client.py
@@ -1552,7 +1552,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "HealthChecksClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/image_family_views/client.py b/google/cloud/compute_v1/services/image_family_views/client.py
@@ -520,7 +520,7 @@ def get(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "ImageFamilyViewsClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/images/client.py b/google/cloud/compute_v1/services/images/client.py
@@ -2131,7 +2131,7 @@ def test_iam_permissions(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "ImagesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/instance_group_managers/client.py b/google/cloud/compute_v1/services/instance_group_managers/client.py
@@ -5067,7 +5067,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "InstanceGroupManagersClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/instance_groups/client.py b/google/cloud/compute_v1/services/instance_groups/client.py
@@ -2143,7 +2143,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "InstanceGroupsClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/instance_templates/client.py b/google/cloud/compute_v1/services/instance_templates/client.py
@@ -1394,7 +1394,7 @@ def test_iam_permissions(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "InstanceTemplatesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/instances/client.py b/google/cloud/compute_v1/services/instances/client.py
@@ -9918,7 +9918,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "InstancesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/interconnect_attachments/client.py b/google/cloud/compute_v1/services/interconnect_attachments/client.py
@@ -1703,7 +1703,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "InterconnectAttachmentsClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/interconnect_locations/client.py b/google/cloud/compute_v1/services/interconnect_locations/client.py
@@ -605,7 +605,7 @@ def list(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "InterconnectLocationsClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/interconnects/client.py b/google/cloud/compute_v1/services/interconnects/client.py
@@ -1543,7 +1543,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "InterconnectsClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/license_codes/client.py b/google/cloud/compute_v1/services/license_codes/client.py
@@ -619,7 +619,7 @@ def test_iam_permissions(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "LicenseCodesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/licenses/client.py b/google/cloud/compute_v1/services/licenses/client.py
@@ -1385,7 +1385,7 @@ def test_iam_permissions(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "LicensesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/machine_images/client.py b/google/cloud/compute_v1/services/machine_images/client.py
@@ -1386,7 +1386,7 @@ def test_iam_permissions(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "MachineImagesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/machine_types/client.py b/google/cloud/compute_v1/services/machine_types/client.py
@@ -710,7 +710,7 @@ def list(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "MachineTypesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/network_attachments/client.py b/google/cloud/compute_v1/services/network_attachments/client.py
@@ -1573,7 +1573,7 @@ def test_iam_permissions(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "NetworkAttachmentsClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/network_edge_security_services/client.py b/google/cloud/compute_v1/services/network_edge_security_services/client.py
@@ -1376,7 +1376,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "NetworkEdgeSecurityServicesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/network_endpoint_groups/client.py b/google/cloud/compute_v1/services/network_endpoint_groups/client.py
@@ -2022,7 +2022,7 @@ def test_iam_permissions(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "NetworkEndpointGroupsClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/network_firewall_policies/client.py b/google/cloud/compute_v1/services/network_firewall_policies/client.py
@@ -3120,7 +3120,7 @@ def test_iam_permissions(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "NetworkFirewallPoliciesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/networks/client.py b/google/cloud/compute_v1/services/networks/client.py
@@ -2295,7 +2295,7 @@ def error_code(self):
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "NetworksClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/node_groups/client.py b/google/cloud/compute_v1/services/node_groups/client.py
@@ -2717,7 +2717,7 @@ def test_iam_permissions(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "NodeGroupsClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/node_templates/client.py b/google/cloud/compute_v1/services/node_templates/client.py
@@ -1573,7 +1573,7 @@ def test_iam_permissions(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "NodeTemplatesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/node_types/client.py b/google/cloud/compute_v1/services/node_types/client.py
@@ -711,7 +711,7 @@ def list(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "NodeTypesClient":
         return self
 
     def __exit__(self, type, value, traceback):

diff --git a/google/cloud/compute_v1/services/packet_mirrorings/client.py b/google/cloud/compute_v1/services/packet_mirrorings/client.py
@@ -1528,7 +1528,7 @@ def test_iam_permissions(
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "PacketMirroringsClient":
         return self
 
     def __exit__(self, type, value, traceback):