From 0fae3a6fcdf9b5af1c64d2f7626474d98a223354 Mon Sep 17 00:00:00 2001 From: rubenwardy Date: Tue, 20 Mar 2018 03:32:14 +0000 Subject: [PATCH] Fix bug with PackageType to name form --- app/models.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/app/models.py b/app/models.py index 4ea8d3c..3629b16 100644 --- a/app/models.py +++ b/app/models.py @@ -24,6 +24,9 @@ class UserRank(enum.Enum): def atLeast(self, min): return self.value >= min.value + def getTitle(self): + return self.name.replace("_", " ").title() + class User(db.Model, UserMixin): id = db.Column(db.Integer, primary_key=True) @@ -71,13 +74,16 @@ class PackageType(enum.Enum): GAME = "Game" TXP = "Texture Pack" + def toName(self): + return self.value.lower().replace(" ", "") + @staticmethod def fromName(name): if name == "mod": return PackageType.MOD elif name == "game": return PackageType.GAME - elif name == "texturepacks": + elif name == "texturepack": return PackageType.TXP else: return None @@ -118,12 +124,12 @@ class Package(db.Model): def getDetailsURL(self): return url_for("package_page", - type=self.type.value.lower(), + type=self.type.toName(), author=self.author.username, name=self.name) def getEditURL(self): return url_for("edit_package_page", - type=self.type.value.lower(), + type=self.type.toName(), author=self.author.username, name=self.name) def checkPerm(self, user, perm):