Add download button and URL

1 files changed, 12 insertions, 0 deletions

diff --git a/app/models.py b/app/models.py
index 9aab1b5..4b16b04 100644
--- a/app/models.py
+++ b/app/models.py
@@ -147,6 +147,18 @@ class Package(db.Model):
 				type=self.type.toName(),
 				author=self.author.username, name=self.name)
 
+	def getDownloadURL(self):
+		return url_for("package_download_page",
+				type=self.type.toName(),
+				author=self.author.username, name=self.name)
+
+	def getDownloadRelease(self):
+		for rel in self.releases:
+			if rel.approved:
+				return rel
+
+		return None
+
 	def checkPerm(self, user, perm):
 		if not user.is_authenticated:
 			return False