Use string enums for natural gas company (#169)

* use string enums for natural gas company * use StrEnum
codeforboston · May 1, 2024 · 0653a98 · 0653a98
1 parent ac0b3d7
commit 0653a98
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: