|
14 | 14 | # limitations under the License. |
15 | 15 | # |
16 | 16 | from collections import OrderedDict |
17 | | -from distutils import util |
18 | 17 | import os |
19 | 18 | import re |
20 | 19 | from typing import Dict, Optional, Sequence, Tuple, Type, Union |
21 | 20 | import pkg_resources |
22 | 21 |
|
23 | | -from google.api_core import client_options as client_options_lib # type: ignore |
24 | | -from google.api_core import exceptions as core_exceptions # type: ignore |
25 | | -from google.api_core import gapic_v1 # type: ignore |
26 | | -from google.api_core import retry as retries # type: ignore |
| 22 | +from google.api_core import client_options as client_options_lib |
| 23 | +from google.api_core import exceptions as core_exceptions |
| 24 | +from google.api_core import gapic_v1 |
| 25 | +from google.api_core import retry as retries |
27 | 26 | from google.auth import credentials as ga_credentials # type: ignore |
28 | 27 | from google.auth.transport import mtls # type: ignore |
29 | 28 | from google.auth.transport.grpc import SslCredentials # type: ignore |
30 | 29 | from google.auth.exceptions import MutualTLSChannelError # type: ignore |
31 | 30 | from google.oauth2 import service_account # type: ignore |
32 | 31 |
|
| 32 | +try: |
| 33 | + OptionalRetry = Union[retries.Retry, gapic_v1.method._MethodDefault] |
| 34 | +except AttributeError: # pragma: NO COVER |
| 35 | + OptionalRetry = Union[retries.Retry, object] # type: ignore |
| 36 | + |
33 | 37 | from google.cloud.iot_v1.services.device_manager import pagers |
34 | 38 | from google.cloud.iot_v1.types import device_manager |
35 | 39 | from google.cloud.iot_v1.types import resources |
@@ -303,8 +307,15 @@ def __init__( |
303 | 307 | client_options = client_options_lib.ClientOptions() |
304 | 308 |
|
305 | 309 | # Create SSL credentials for mutual TLS if needed. |
306 | | - use_client_cert = bool( |
307 | | - util.strtobool(os.getenv("GOOGLE_API_USE_CLIENT_CERTIFICATE", "false")) |
| 310 | + if os.getenv("GOOGLE_API_USE_CLIENT_CERTIFICATE", "false") not in ( |
| 311 | + "true", |
| 312 | + "false", |
| 313 | + ): |
| 314 | + raise ValueError( |
| 315 | + "Environment variable `GOOGLE_API_USE_CLIENT_CERTIFICATE` must be either `true` or `false`" |
| 316 | + ) |
| 317 | + use_client_cert = ( |
| 318 | + os.getenv("GOOGLE_API_USE_CLIENT_CERTIFICATE", "false") == "true" |
308 | 319 | ) |
309 | 320 |
|
310 | 321 | client_cert_source_func = None |
@@ -375,7 +386,7 @@ def create_device_registry( |
375 | 386 | *, |
376 | 387 | parent: str = None, |
377 | 388 | device_registry: resources.DeviceRegistry = None, |
378 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 389 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
379 | 390 | timeout: float = None, |
380 | 391 | metadata: Sequence[Tuple[str, str]] = (), |
381 | 392 | ) -> resources.DeviceRegistry: |
@@ -455,7 +466,7 @@ def get_device_registry( |
455 | 466 | request: Union[device_manager.GetDeviceRegistryRequest, dict] = None, |
456 | 467 | *, |
457 | 468 | name: str = None, |
458 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 469 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
459 | 470 | timeout: float = None, |
460 | 471 | metadata: Sequence[Tuple[str, str]] = (), |
461 | 472 | ) -> resources.DeviceRegistry: |
@@ -524,7 +535,7 @@ def update_device_registry( |
524 | 535 | *, |
525 | 536 | device_registry: resources.DeviceRegistry = None, |
526 | 537 | update_mask: field_mask_pb2.FieldMask = None, |
527 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 538 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
528 | 539 | timeout: float = None, |
529 | 540 | metadata: Sequence[Tuple[str, str]] = (), |
530 | 541 | ) -> resources.DeviceRegistry: |
@@ -609,7 +620,7 @@ def delete_device_registry( |
609 | 620 | request: Union[device_manager.DeleteDeviceRegistryRequest, dict] = None, |
610 | 621 | *, |
611 | 622 | name: str = None, |
612 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 623 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
613 | 624 | timeout: float = None, |
614 | 625 | metadata: Sequence[Tuple[str, str]] = (), |
615 | 626 | ) -> None: |
@@ -672,7 +683,7 @@ def list_device_registries( |
672 | 683 | request: Union[device_manager.ListDeviceRegistriesRequest, dict] = None, |
673 | 684 | *, |
674 | 685 | parent: str = None, |
675 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 686 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
676 | 687 | timeout: float = None, |
677 | 688 | metadata: Sequence[Tuple[str, str]] = (), |
678 | 689 | ) -> pagers.ListDeviceRegistriesPager: |
@@ -752,7 +763,7 @@ def create_device( |
752 | 763 | *, |
753 | 764 | parent: str = None, |
754 | 765 | device: resources.Device = None, |
755 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 766 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
756 | 767 | timeout: float = None, |
757 | 768 | metadata: Sequence[Tuple[str, str]] = (), |
758 | 769 | ) -> resources.Device: |
@@ -832,7 +843,7 @@ def get_device( |
832 | 843 | request: Union[device_manager.GetDeviceRequest, dict] = None, |
833 | 844 | *, |
834 | 845 | name: str = None, |
835 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 846 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
836 | 847 | timeout: float = None, |
837 | 848 | metadata: Sequence[Tuple[str, str]] = (), |
838 | 849 | ) -> resources.Device: |
@@ -903,7 +914,7 @@ def update_device( |
903 | 914 | *, |
904 | 915 | device: resources.Device = None, |
905 | 916 | update_mask: field_mask_pb2.FieldMask = None, |
906 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 917 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
907 | 918 | timeout: float = None, |
908 | 919 | metadata: Sequence[Tuple[str, str]] = (), |
909 | 920 | ) -> resources.Device: |
@@ -988,7 +999,7 @@ def delete_device( |
988 | 999 | request: Union[device_manager.DeleteDeviceRequest, dict] = None, |
989 | 1000 | *, |
990 | 1001 | name: str = None, |
991 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 1002 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
992 | 1003 | timeout: float = None, |
993 | 1004 | metadata: Sequence[Tuple[str, str]] = (), |
994 | 1005 | ) -> None: |
@@ -1053,7 +1064,7 @@ def list_devices( |
1053 | 1064 | request: Union[device_manager.ListDevicesRequest, dict] = None, |
1054 | 1065 | *, |
1055 | 1066 | parent: str = None, |
1056 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 1067 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1057 | 1068 | timeout: float = None, |
1058 | 1069 | metadata: Sequence[Tuple[str, str]] = (), |
1059 | 1070 | ) -> pagers.ListDevicesPager: |
@@ -1133,7 +1144,7 @@ def modify_cloud_to_device_config( |
1133 | 1144 | *, |
1134 | 1145 | name: str = None, |
1135 | 1146 | binary_data: bytes = None, |
1136 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 1147 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1137 | 1148 | timeout: float = None, |
1138 | 1149 | metadata: Sequence[Tuple[str, str]] = (), |
1139 | 1150 | ) -> resources.DeviceConfig: |
@@ -1219,7 +1230,7 @@ def list_device_config_versions( |
1219 | 1230 | request: Union[device_manager.ListDeviceConfigVersionsRequest, dict] = None, |
1220 | 1231 | *, |
1221 | 1232 | name: str = None, |
1222 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 1233 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1223 | 1234 | timeout: float = None, |
1224 | 1235 | metadata: Sequence[Tuple[str, str]] = (), |
1225 | 1236 | ) -> device_manager.ListDeviceConfigVersionsResponse: |
@@ -1293,7 +1304,7 @@ def list_device_states( |
1293 | 1304 | request: Union[device_manager.ListDeviceStatesRequest, dict] = None, |
1294 | 1305 | *, |
1295 | 1306 | name: str = None, |
1296 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 1307 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1297 | 1308 | timeout: float = None, |
1298 | 1309 | metadata: Sequence[Tuple[str, str]] = (), |
1299 | 1310 | ) -> device_manager.ListDeviceStatesResponse: |
@@ -1364,7 +1375,7 @@ def set_iam_policy( |
1364 | 1375 | request: Union[iam_policy_pb2.SetIamPolicyRequest, dict] = None, |
1365 | 1376 | *, |
1366 | 1377 | resource: str = None, |
1367 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 1378 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1368 | 1379 | timeout: float = None, |
1369 | 1380 | metadata: Sequence[Tuple[str, str]] = (), |
1370 | 1381 | ) -> policy_pb2.Policy: |
@@ -1490,7 +1501,7 @@ def get_iam_policy( |
1490 | 1501 | request: Union[iam_policy_pb2.GetIamPolicyRequest, dict] = None, |
1491 | 1502 | *, |
1492 | 1503 | resource: str = None, |
1493 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 1504 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1494 | 1505 | timeout: float = None, |
1495 | 1506 | metadata: Sequence[Tuple[str, str]] = (), |
1496 | 1507 | ) -> policy_pb2.Policy: |
@@ -1618,7 +1629,7 @@ def test_iam_permissions( |
1618 | 1629 | *, |
1619 | 1630 | resource: str = None, |
1620 | 1631 | permissions: Sequence[str] = None, |
1621 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 1632 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1622 | 1633 | timeout: float = None, |
1623 | 1634 | metadata: Sequence[Tuple[str, str]] = (), |
1624 | 1635 | ) -> iam_policy_pb2.TestIamPermissionsResponse: |
@@ -1703,7 +1714,7 @@ def send_command_to_device( |
1703 | 1714 | name: str = None, |
1704 | 1715 | binary_data: bytes = None, |
1705 | 1716 | subfolder: str = None, |
1706 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 1717 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1707 | 1718 | timeout: float = None, |
1708 | 1719 | metadata: Sequence[Tuple[str, str]] = (), |
1709 | 1720 | ) -> device_manager.SendCommandToDeviceResponse: |
@@ -1817,7 +1828,7 @@ def bind_device_to_gateway( |
1817 | 1828 | parent: str = None, |
1818 | 1829 | gateway_id: str = None, |
1819 | 1830 | device_id: str = None, |
1820 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 1831 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1821 | 1832 | timeout: float = None, |
1822 | 1833 | metadata: Sequence[Tuple[str, str]] = (), |
1823 | 1834 | ) -> device_manager.BindDeviceToGatewayResponse: |
@@ -1906,7 +1917,7 @@ def unbind_device_from_gateway( |
1906 | 1917 | parent: str = None, |
1907 | 1918 | gateway_id: str = None, |
1908 | 1919 | device_id: str = None, |
1909 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 1920 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1910 | 1921 | timeout: float = None, |
1911 | 1922 | metadata: Sequence[Tuple[str, str]] = (), |
1912 | 1923 | ) -> device_manager.UnbindDeviceFromGatewayResponse: |
|
0 commit comments