Skip to content

Commit 2d4415d

Browse files
Fixing constants + absolute imports
1 parent 0af2a3c commit 2d4415d

File tree

6 files changed

+39
-27
lines changed

6 files changed

+39
-27
lines changed

aws_lambda_powertools/utilities/parameters/appconfig.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,10 @@
1515
resolve_env_var_choice,
1616
resolve_max_age,
1717
)
18+
from aws_lambda_powertools.utilities.parameters.base import BaseProvider
19+
from aws_lambda_powertools.utilities.parameters.constants import DEFAULT_MAX_AGE_SECS, DEFAULT_PROVIDERS
1820
from aws_lambda_powertools.warnings import PowertoolsDeprecationWarning
1921

20-
from .base import DEFAULT_MAX_AGE_SECS, DEFAULT_PROVIDERS, BaseProvider
21-
2222
if TYPE_CHECKING:
2323
from botocore.config import Config
2424
from mypy_boto3_appconfigdata.client import AppConfigDataClient

aws_lambda_powertools/utilities/parameters/base.py

Lines changed: 6 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,6 @@
44

55
from __future__ import annotations
66

7-
import base64
8-
import json
97
import os
108
from abc import ABC, abstractmethod
119
from datetime import datetime, timedelta
@@ -19,21 +17,12 @@
1917
if TYPE_CHECKING:
2018
from aws_lambda_powertools.utilities.parameters.types import TransformOptions
2119

22-
DEFAULT_MAX_AGE_SECS = "300"
23-
24-
# These providers will be dynamically initialized on first use of the helper functions
25-
DEFAULT_PROVIDERS: dict[str, Any] = {}
26-
TRANSFORM_METHOD_JSON = "json"
27-
TRANSFORM_METHOD_BINARY = "binary"
28-
SUPPORTED_TRANSFORM_METHODS = [TRANSFORM_METHOD_JSON, TRANSFORM_METHOD_BINARY]
29-
30-
TRANSFORM_METHOD_MAPPING = {
31-
TRANSFORM_METHOD_JSON: json.loads,
32-
TRANSFORM_METHOD_BINARY: base64.b64decode,
33-
".json": json.loads,
34-
".binary": base64.b64decode,
35-
None: lambda x: x,
36-
}
20+
21+
from aws_lambda_powertools.utilities.parameters.constants import (
22+
DEFAULT_MAX_AGE_SECS,
23+
DEFAULT_PROVIDERS,
24+
TRANSFORM_METHOD_MAPPING,
25+
)
3726

3827

3928
class ExpirableValue(NamedTuple):
Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
from __future__ import annotations
2+
3+
import base64
4+
import json
5+
from typing import Any, Literal
6+
7+
SSM_PARAMETER_TYPES = Literal["String", "StringList", "SecureString"]
8+
SSM_PARAMETER_TIER = Literal["Standard", "Advanced", "Intelligent-Tiering"]
9+
10+
DEFAULT_MAX_AGE_SECS = "300"
11+
12+
# These providers will be dynamically initialized on first use of the helper functions
13+
DEFAULT_PROVIDERS: dict[str, Any] = {}
14+
TRANSFORM_METHOD_JSON = "json"
15+
TRANSFORM_METHOD_BINARY = "binary"
16+
TRANSFORM_METHOD_MAPPING = {
17+
TRANSFORM_METHOD_JSON: json.loads,
18+
TRANSFORM_METHOD_BINARY: base64.b64decode,
19+
".json": json.loads,
20+
".binary": base64.b64decode,
21+
None: lambda x: x,
22+
}

aws_lambda_powertools/utilities/parameters/dynamodb.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,10 +10,9 @@
1010
import boto3
1111
from boto3.dynamodb.conditions import Key
1212

13+
from aws_lambda_powertools.utilities.parameters.base import BaseProvider
1314
from aws_lambda_powertools.warnings import PowertoolsDeprecationWarning
1415

15-
from .base import BaseProvider
16-
1716
if TYPE_CHECKING:
1817
from botocore.config import Config
1918
from mypy_boto3_dynamodb.service_resource import DynamoDBServiceResource

aws_lambda_powertools/utilities/parameters/secrets.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,8 @@
1515
from aws_lambda_powertools.shared import constants
1616
from aws_lambda_powertools.shared.functions import resolve_max_age
1717
from aws_lambda_powertools.shared.json_encoder import Encoder
18-
from aws_lambda_powertools.utilities.parameters.base import DEFAULT_MAX_AGE_SECS, DEFAULT_PROVIDERS, BaseProvider
18+
from aws_lambda_powertools.utilities.parameters.base import BaseProvider
19+
from aws_lambda_powertools.utilities.parameters.constants import DEFAULT_MAX_AGE_SECS, DEFAULT_PROVIDERS
1920
from aws_lambda_powertools.utilities.parameters.exceptions import SetSecretError
2021
from aws_lambda_powertools.warnings import PowertoolsDeprecationWarning
2122

aws_lambda_powertools/utilities/parameters/ssm.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,15 @@
1818
slice_dictionary,
1919
)
2020
from aws_lambda_powertools.utilities.parameters.base import (
21-
DEFAULT_MAX_AGE_SECS,
22-
DEFAULT_PROVIDERS,
2321
BaseProvider,
2422
transform_value,
2523
)
24+
from aws_lambda_powertools.utilities.parameters.constants import (
25+
DEFAULT_MAX_AGE_SECS,
26+
DEFAULT_PROVIDERS,
27+
SSM_PARAMETER_TIER,
28+
SSM_PARAMETER_TYPES,
29+
)
2630
from aws_lambda_powertools.utilities.parameters.exceptions import GetParameterError, SetParameterError
2731
from aws_lambda_powertools.warnings import PowertoolsDeprecationWarning
2832

@@ -33,9 +37,6 @@
3337

3438
from aws_lambda_powertools.utilities.parameters.types import TransformOptions
3539

36-
SSM_PARAMETER_TYPES = Literal["String", "StringList", "SecureString"]
37-
SSM_PARAMETER_TIER = Literal["Standard", "Advanced", "Intelligent-Tiering"]
38-
3940
logger = logging.getLogger(__name__)
4041

4142

0 commit comments

Comments
 (0)