Use string enums for natural gas company (codeforboston#169)

* use string enums for natural gas company * use StrEnum
AdamFinkle · May 1, 2024 · c4eeea4 · c4eeea4
1 parent 07ad54b
commit c4eeea4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/rules-engine/src/rules_engine/parser.py b/rules-engine/src/rules_engine/parser.py
@@ -5,14 +5,14 @@
 import csv
 import io
 from datetime import datetime, timedelta
-from enum import Enum
+from enum import StrEnum
 
 from .pydantic_models import NaturalGasBillingInput, NaturalGasBillingRecordInput
 
 
-class NaturalGasCompany(Enum):
-    EVERSOURCE = 1
-    NATIONAL_GRID = 2
+class NaturalGasCompany(StrEnum):
+    EVERSOURCE = "eversource"
+    NATIONAL_GRID = "national_grid"
 
 
 class _GasBillRowEversource: