diff options
| author | rubenwardy <rw@rubenwardy.com> | 2018-03-20 00:58:44 +0000 |
|---|---|---|
| committer | rubenwardy <rw@rubenwardy.com> | 2018-03-20 00:58:44 +0000 |
| commit | 07a9b79396fd260b2d9aa918751823c70ba709c2 (patch) | |
| tree | e9eebd6881afe69ff3a0e75b86f3eab8e43fe374 /app/models.py | |
| parent | bc88027fbe2039aa509de6c958a1ab3939909539 (diff) | |
| download | cheatdb-07a9b79396fd260b2d9aa918751823c70ba709c2.tar.xz | |
Check type and author in package details
Diffstat (limited to 'app/models.py')
| -rw-r--r-- | app/models.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/app/models.py b/app/models.py index 177e4c8..c5ce4b6 100644 --- a/app/models.py +++ b/app/models.py @@ -71,6 +71,17 @@ class PackageType(enum.Enum): else: return "TXP" + @staticmethod + def fromName(name): + if name == "mod": + return PackageType.MOD + elif name == "game": + return PackageType.GAME + elif name == "texturepacks": + return PackageType.TXP + else: + return None + class Package(db.Model): id = db.Column(db.Integer, primary_key=True) |