#54: Use EnumAsInteger Column type

ral-facilities · Sep 12, 2019 · ba49d1e · ba49d1e
1 parent 10f974e
commit ba49d1e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/common/models/db_models.py b/common/models/db_models.py
@@ -577,7 +577,6 @@ class JOB(Base, EntityHelper):
                                   backref='JOB')
 
 
-
 class KEYWORD(Base, EntityHelper):
     __tablename__ = 'KEYWORD'
     __table_args__ = (
@@ -624,7 +623,7 @@ class ValueTypeEnum(enum.Enum):
     NAME = Column(String(255), nullable=False)
     UNITS = Column(String(255), nullable=False)
     UNITSFULLNAME = Column(String(255))
-    VALUETYPE = Column(Integer, nullable=False)
+    VALUETYPE = Column(EnumAsInteger(ValueTypeEnum), nullable=False)
     VERIFIED = Column(Integer, server_default=FetchedValue())
     FACILITY_ID = Column(ForeignKey('FACILITY.ID'), nullable=False)