From f8b290fc45842e67e86a28e837bdac5fa536d0eb Mon Sep 17 00:00:00 2001 From: rubenwardy Date: Sat, 30 Mar 2019 15:41:38 +0000 Subject: Add badges next to packages awaiting approval list --- app/models.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) (limited to 'app/models.py') diff --git a/app/models.py b/app/models.py index ed9a21f..be4cdea 100644 --- a/app/models.py +++ b/app/models.py @@ -417,6 +417,22 @@ class Package(db.Model): for e in PackagePropertyKey: setattr(self, e.name, getattr(package, e.name)) + def getState(self): + if self.approved: + return "approved" + elif self.review_thread_id: + return "thread" + elif (self.type == PackageType.GAME or \ + self.type == PackageType.TXP) and \ + self.screenshots.count() == 0: + return "wip" + elif not self.getDownloadRelease(): + return "wip" + elif "Other" in self.license.name or "Other" in self.media_license.name: + return "license" + else: + return "ready" + def getAsDictionaryShort(self, base_url, version=None, protonum=None): tnurl = self.getThumbnailURL(1) release = self.getDownloadRelease(version=version, protonum=protonum) -- cgit v1.2.3