+ api

models/enums.py

@@ -1,19 +1,43 @@
 from enum import Enum
 
 
-class AspectRatios(Enum):
-    NINESIXTEEN = '9:16'
-    SIXTEENNINE = '16:9'
-    THREEFOUR = '3:4'
-    FOURTHREE = '4:3'
+class AspectRatios(str, Enum):
+    NINESIXTEEN = "NINESIXTEEN"
+    SIXTEENNINE = "SIXTEENNINE"
+    THREEFOUR = "THREEFOUR"
+    FOURTHREE = "FOURTHREE"
+
+    @property
+    def value_ratio(self) -> str:
+        return {
+            AspectRatios.NINESIXTEEN: "9:16",
+            AspectRatios.SIXTEENNINE: "16:9",
+            AspectRatios.THREEFOUR: "3:4",
+            AspectRatios.FOURTHREE: "4:3",
+        }[self]
 
 
-class Quality(Enum):
-    ONEK = '1K'
-    TWOK = '2K'
-    FOURK = '4K'
+class Quality(str, Enum):
+    ONEK = 'ONEK'
+    TWOK = 'TWOK'
+    FOURK = 'FOURK'
+
+    @property
+    def value_quality(self) -> str:
+        return {
+            Quality.ONEK: '1K',
+            Quality.TWOK: '2K',
+            Quality.FOURK: '4K'
+        }[self]
 
 
-class GenType(Enum):
+class GenType(str, Enum):
     TEXT = 'Text'
-    IMAGE = 'Image'
+    IMAGE = 'Image'
+
+    @property
+    def value_type(self) -> str:
+        return {
+            GenType.TEXT: 'Text',
+            GenType.IMAGE: 'Image'
+        }[self]