From c4eeea45de6a3233e55caaa2eaf948355df30fcc Mon Sep 17 00:00:00 2001 From: Alan Pinkert Date: Tue, 30 Apr 2024 20:10:54 -0400 Subject: [PATCH] Use string enums for natural gas company (#169) * use string enums for natural gas company * use StrEnum --- rules-engine/src/rules_engine/parser.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/rules-engine/src/rules_engine/parser.py b/rules-engine/src/rules_engine/parser.py index e90ac7fd..54aff093 100644 --- 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: