diff options
| author | rubenwardy <rw@rubenwardy.com> | 2018-05-15 20:05:35 +0100 |
|---|---|---|
| committer | rubenwardy <rw@rubenwardy.com> | 2018-05-15 20:05:35 +0100 |
| commit | cfb92273ff9b3fb22bdfa215fb2912a62f6f0baa (patch) | |
| tree | 87502500e1497a4f362ee78ca60aeca46247a104 | |
| parent | dbc56283ebb0d850fff65ecaf6753c0fa31e0c53 (diff) | |
| download | cheatdb-cfb92273ff9b3fb22bdfa215fb2912a62f6f0baa.tar.xz | |
Add import screenshots admin action
| -rw-r--r-- | app/templates/admin/list.html | 1 | ||||
| -rw-r--r-- | app/views/admin.py | 11 |
2 files changed, 11 insertions, 1 deletions
diff --git a/app/templates/admin/list.html b/app/templates/admin/list.html index 1db3247..b54201b 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 03dcc75..0f308b6 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") |