From f8b290fc45842e67e86a28e837bdac5fa536d0eb Mon Sep 17 00:00:00 2001
From: rubenwardy <rw@rubenwardy.com>
Date: Sat, 30 Mar 2019 15:41:38 +0000
Subject: [PATCH] Add badges next to packages awaiting approval list

---
 app/models.py                | 16 ++++++++++++++++
 app/templates/todo/list.html | 10 ++++++++++
 2 files changed, 26 insertions(+)

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)
diff --git a/app/templates/todo/list.html b/app/templates/todo/list.html
index 2a6fa5e..3eb3251 100644
--- a/app/templates/todo/list.html
+++ b/app/templates/todo/list.html
@@ -15,6 +15,16 @@
 				<div class="list-group list-group-flush">
 					{% for p in packages %}
 						<a href="{{ p.getDetailsURL() }}" class="list-group-item list-group-item-action">
+							{% if p.getState() == "thread" %}
+								<span class="mr-2 badge badge-danger">Thread</span>
+							{% elif p.getState() == "ready" %}
+								<span class="mr-2 badge badge-success">Ready</span>
+							{% elif p.getState() == "wip" %}
+								<span class="mr-2 badge badge-warning">WIP</span>
+							{% elif p.getState() == "license" %}
+								<span class="mr-2 badge badge-info">WIP</span>
+							{% endif %}
+
 							{{ p.title }} by {{ p.author.display_name }}
 						</a>
 					{% else %}