Skip to content

Commit a3b9eb4

Browse files
troyswansonrubenfonseca
authored andcommitted
Add overload signatures for get_parameter function
1 parent 76217cd commit a3b9eb4

File tree

1 file changed

+37
-1
lines changed
  • aws_lambda_powertools/utilities/parameters

1 file changed

+37
-1
lines changed

aws_lambda_powertools/utilities/parameters/ssm.py

Lines changed: 37 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -545,9 +545,45 @@ def _raise_if_errors_key_is_present(parameters: Dict, reserved_parameter: str, r
545545
)
546546

547547

548+
@overload
548549
def get_parameter(
549550
name: str,
550-
transform: Optional[str] = None,
551+
transform: None = None,
552+
decrypt: Optional[bool] = None,
553+
force_fetch: bool = False,
554+
max_age: Optional[int] = None,
555+
**sdk_options,
556+
) -> str:
557+
...
558+
559+
560+
@overload
561+
def get_parameter(
562+
name: str,
563+
transform: Literal["json"],
564+
decrypt: Optional[bool] = None,
565+
force_fetch: bool = False,
566+
max_age: Optional[int] = None,
567+
**sdk_options,
568+
) -> dict:
569+
...
570+
571+
572+
@overload
573+
def get_parameter(
574+
name: str,
575+
transform: Literal["binary"],
576+
decrypt: Optional[bool] = None,
577+
force_fetch: bool = False,
578+
max_age: Optional[int] = None,
579+
**sdk_options,
580+
) -> bytes:
581+
...
582+
583+
584+
def get_parameter(
585+
name: str,
586+
transform: Optional[Literal["json", "binary"]] = None,
551587
decrypt: Optional[bool] = None,
552588
force_fetch: bool = False,
553589
max_age: Optional[int] = None,

0 commit comments

Comments
 (0)