Add support for multiple types in packages list

1 files changed, 7 insertions, 0 deletions

diff --git a/app/models.py b/app/models.py
index f002abc..8f71146 100644
--- a/app/models.py
+++ b/app/models.py
@@ -212,6 +212,13 @@ class PackageType(enum.Enum):
 		return self.name
 
 	@classmethod
+	def get(cls, name):
+		try:
+			return PackageType[name.upper()]
+		except KeyError:
+			return None
+
+	@classmethod
 	def choices(cls):
 		return [(choice, choice.value) for choice in cls]