Hide VCS release option when not available

1 files changed, 21 insertions, 0 deletions

diff --git a/app/models.py b/app/models.py
index dad0a03..eac8d8f 100644
--- a/app/models.py
+++ b/app/models.py
@@ -18,6 +18,7 @@
 from flask import Flask, url_for
 from flask_sqlalchemy import SQLAlchemy
 from flask_migrate import Migrate
+from urllib.parse import urlparse
 from app import app
 from datetime import datetime
 from sqlalchemy.orm import validates
@@ -339,6 +340,26 @@ class Package(db.Model):
 
 		return None
 
+	def canImportScreenshot(self):
+		if self.repo is None:
+			return False
+
+		url = urlparse(self.repo)
+		if url.netloc == "github.com":
+			return True
+
+		return False
+
+	def canMakeReleaseFromVCS(self):
+		if self.repo is None:
+			return False
+
+		url = urlparse(self.repo)
+		if url.netloc == "github.com":
+			return True
+
+		return False
+
 	def checkPerm(self, user, perm):
 		if not user.is_authenticated:
 			return False