Skip to content

Commit 4088136

Browse files
committed
refactor: use StrEnum for string enum classes
Replace custom string enum classes with Python's enum.StrEnum for better type safety and clarity.
1 parent e1f727f commit 4088136

File tree

4 files changed

+15
-15
lines changed

4 files changed

+15
-15
lines changed

cardano_node_tests/utils/cluster_nodes.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
"""Functionality for cluster setup and interaction with cluster nodes."""
22

33
import dataclasses
4+
import enum
45
import functools
56
import json
67
import logging
@@ -43,10 +44,10 @@ class ServiceStatus:
4344
message: str = ""
4445

4546

46-
class Testnets:
47-
preview: tp.Final[str] = "preview"
48-
preprod: tp.Final[str] = "preprod"
49-
mainnet: tp.Final[str] = "mainnet"
47+
class Testnets(enum.StrEnum):
48+
preview = "preview"
49+
preprod = "preprod"
50+
mainnet = "mainnet"
5051

5152

5253
class ClusterType:

cardano_node_tests/utils/clusterlib_utils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,10 +60,10 @@ class ChainAccount:
6060
treasury: int
6161

6262

63-
class BuildMethods:
64-
BUILD: tp.Final[str] = "build"
65-
BUILD_RAW: tp.Final[str] = "build_raw"
66-
BUILD_EST: tp.Final[str] = "build_estimate"
63+
class BuildMethods(enum.StrEnum):
64+
BUILD = "build"
65+
BUILD_RAW = "build_raw"
66+
BUILD_EST = "build_estimate"
6767

6868

6969
class KeyGenMethods(enum.StrEnum):

cardano_node_tests/utils/requirements.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,14 @@
44
import json
55
import logging
66
import pathlib as pl
7-
import typing as tp
87

98
from cardano_node_tests.utils import helpers
109

1110
LOGGER = logging.getLogger(__name__)
1211

1312

14-
class GroupsKnown:
15-
CHANG_US: tp.Final[str] = "chang_us"
13+
class GroupsKnown(enum.StrEnum):
14+
CHANG_US = "chang_us"
1615

1716

1817
class Statuses(enum.Enum):

cardano_node_tests/utils/submit_utils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,15 @@
1+
import enum
12
import shutil
2-
import typing as tp
33

44
import pytest
55
from cardano_clusterlib import clusterlib
66

77
from cardano_node_tests.utils import submit_api
88

99

10-
class SubmitMethods:
11-
API: tp.Final[str] = "api"
12-
CLI: tp.Final[str] = "cli"
10+
class SubmitMethods(enum.StrEnum):
11+
API = "api"
12+
CLI = "cli"
1313

1414

1515
# The "submit_method" is a fixtrue defined in `conftest.py`.

0 commit comments

Comments
 (0)