From cfb92273ff9b3fb22bdfa215fb2912a62f6f0baa Mon Sep 17 00:00:00 2001
From: rubenwardy <rw@rubenwardy.com>
Date: Tue, 15 May 2018 20:05:35 +0100
Subject: [PATCH] Add import screenshots admin action

---
 app/templates/admin/list.html |  1 +
 app/views/admin.py            | 11 ++++++++++-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/app/templates/admin/list.html b/app/templates/admin/list.html
index 1db3247f..b54201bb 100644
--- a/app/templates/admin/list.html
+++ b/app/templates/admin/list.html
@@ -17,6 +17,7 @@
 			<input type="hidden" name="csrf_token" value="{{ csrf_token() }}" />
 			<select name="action">
 				<option value="importusers" selected>Create users from mod list</option>
+				<option value="importscreenshots">Import screenshots from VCS</option>
 			</select>
 			<input type="submit" value="Start" />
 		</form>
diff --git a/app/views/admin.py b/app/views/admin.py
index 03dcc759..0f308b67 100644
--- a/app/views/admin.py
+++ b/app/views/admin.py
@@ -3,7 +3,8 @@ from flask_user import *
 from flask.ext import menu
 from app import app
 from app.models import *
-from app.tasks.forumtasks import importUsersFromModList
+from app.tasks.importtasks import importRepoScreenshot
+from app.tasks.forumtasks  import importUsersFromModList
 from flask_wtf import FlaskForm
 from wtforms import *
 from app.utils import loginUser, rank_required
@@ -18,6 +19,14 @@ def admin_page():
 		if action == "importusers":
 			task = importUsersFromModList.delay()
 			return redirect(url_for("check_task", id=task.id, r=url_for("user_list_page")))
+		elif action == "importscreenshots":
+			packages = Package.query \
+				.outerjoin(PackageScreenshot, Package.id==PackageScreenshot.package_id) \
+				.filter(PackageScreenshot.id==None).all()
+			for package in packages:
+				importRepoScreenshot.delay(package.id)
+
+			return redirect(url_for("admin_page"))
 		else:
 			flash("Unknown action: " + action, "error")