1 files changed, 75 insertions, 0 deletions
diff --git a/app/blueprints/tasks/__init__.py b/app/blueprints/tasks/__init__.py
new file mode 100644
index 0000000..8d002db
--- /dev/null
+++ b/app/blueprints/tasks/__init__.py
@@ -0,0 +1,75 @@
+# Content DB
+# Copyright (C) 2018  rubenwardy
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+
+from flask import *
+from flask_user import *
+import flask_menu as menu
+from app import csrf
+from app.models import *
+from app.tasks import celery, TaskError
+from app.tasks.importtasks import getMeta
+from app.utils import shouldReturnJson
+from app.utils import *
+
+bp = Blueprint("tasks", __name__)
+
+@csrf.exempt
+@bp.route("/tasks/getmeta/new/", methods=["POST"])
+@login_required
+def start_getmeta():
+	author = request.args.get("author")
+	author = current_user.forums_username if author is None else author
+	aresult = getMeta.delay(request.args.get("url"), author)
+	return jsonify({
+		"poll_url": url_for("tasks.check", id=aresult.id),
+	})
+
+@bp.route("/tasks/<id>/")
+def check(id):
+	result = celery.AsyncResult(id)
+	status = result.status
+	traceback = result.traceback
+	result = result.result
+
+	info = None
+	if isinstance(result, Exception):
+		info = {
+				'id': id,
+				'status': status,
+			}
+
+		if current_user.is_authenticated and current_user.rank.atLeast(UserRank.ADMIN):
+			info["error"] = str(traceback)
+		elif str(result)[1:12] == "TaskError: ":
+			info["error"] = str(result)[12:-1]
+		else:
+			info["error"] = "Unknown server error"
+	else:
+		info = {
+				'id': id,
+				'status': status,
+				'result': result,
+			}
+
+	if shouldReturnJson():
+		return jsonify(info)
+	else:
+		r = request.args.get("r")
+		if r is not None and status == "SUCCESS":
+			return redirect(r)
+		else:
+			return render_template("tasks/view.html", info=info)