Skip to content

Commit 94ae313

Browse files
committed
chore: update CLI to accept log level as string
1 parent a5f594a commit 94ae313

1 file changed

Lines changed: 7 additions & 6 deletions

File tree

  • src/aws_durable_execution_sdk_python_testing

src/aws_durable_execution_sdk_python_testing/cli.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ class CliConfig:
4444
# Server configuration
4545
host: str = "0.0.0.0" # noqa:S104
4646
port: int = 5000
47-
log_level: int = 20 # INFO level
47+
log_level: str = "info"
4848
lambda_endpoint: str = "http://127.0.0.1:3001"
4949
local_runner_endpoint: str = "http://0.0.0.0:5000"
5050
local_runner_region: str = "us-west-2"
@@ -56,7 +56,7 @@ def from_environment(cls) -> CliConfig:
5656
return cls(
5757
host=os.getenv("AWS_DEX_HOST", "0.0.0.0"), # noqa:S104
5858
port=int(os.getenv("AWS_DEX_PORT", "5000")),
59-
log_level=int(os.getenv("AWS_DEX_LOG_LEVEL", "20")),
59+
log_level=os.getenv("AWS_DEX_LOG_LEVEL", "info"),
6060
lambda_endpoint=os.getenv(
6161
"AWS_DEX_LAMBDA_ENDPOINT", "http://127.0.0.1:3001"
6262
),
@@ -90,9 +90,9 @@ def run(self, args: list[str] | None = None) -> int:
9090

9191
# Configure logging based on log level
9292
logging.basicConfig(
93-
level=parsed_args.log_level
93+
level=getattr(logging, parsed_args.log_level.upper())
9494
if hasattr(parsed_args, "log_level")
95-
else self.config.log_level,
95+
else getattr(logging, self.config.log_level.upper()),
9696
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
9797
)
9898

@@ -151,9 +151,10 @@ def _create_start_server_parser(self, subparsers) -> None:
151151
)
152152
start_server_parser.add_argument(
153153
"--log-level",
154-
type=int,
154+
type=str,
155+
choices=["debug", "info", "warning", "error", "critical"],
155156
default=self.config.log_level,
156-
help=f"Logging level as integer (default: {self.config.log_level}, env: AWS_DEX_LOG_LEVEL)",
157+
help=f"Logging level (default: {self.config.log_level}, env: AWS_DEX_LOG_LEVEL)",
157158
)
158159
start_server_parser.add_argument(
159160
"--lambda-endpoint",

0 commit comments

Comments
 (0)