forked from Scouterna/skojjt
-
Notifications
You must be signed in to change notification settings - Fork 0
/
imports.py
83 lines (68 loc) · 2.92 KB
/
imports.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
# -*- coding: utf-8 -*-
import time
from dataimport import UserPrefs, ndb, Semester, TaskProgress, RunScoutnetImport
from flask import Blueprint, render_template, request, make_response, redirect
from google.appengine.ext import deferred
import_page = Blueprint('import_page', __name__, template_folder='templates')
@import_page.route('/', methods = ['POST', 'GET'])
def import_():
user = UserPrefs.current()
if not user.canImport():
return "denied", 403
breadcrumbs = [{'link':'/', 'text':'Hem'},
{'link':'/import', 'text':'Import'}]
currentSemester = Semester.getOrCreateCurrent()
semesters=[currentSemester]
semesters.extend(Semester.query(Semester.key!=currentSemester.key))
if request.method != 'POST':
return render_template('updatefromscoutnetform.html', heading="Import", breadcrumbs=breadcrumbs, user=user, semesters=semesters)
api_key = request.form.get('apikey').strip()
groupid = request.form.get('groupid').strip()
semester_key=ndb.Key(urlsafe=request.form.get('semester'))
return startAsyncImport(api_key, groupid, semester_key, user, request)
progress = Blueprint('progress_page', 'progress', template_folder='templates')
@progress.route('/<progress_url>')
@progress.route('/<progress_url>/')
@progress.route('/<progress_url>/<update>')
@progress.route('/<progress_url>/<update>/')
def importProgress(progress_url, update=None):
if update is not None:
taskProgress = None
for i in range(1, 2):
taskProgress = ndb.Key(urlsafe=progress_url).get()
if taskProgress is not None:
break
time.sleep(1)
if taskProgress is not None:
s = taskProgress.toJson()
else:
s = '{"messages": ["Error: Hittar inte uppgiften"], "failed": "true", "running": "false"}'
response = make_response(s)
response.headers['Content-Type'] = 'application/json'
return response
breadcrumbs = [{'link':'/', 'text':'Hem'}, {'link':'/import', 'text':'Import'}]
return render_template('importresult.html', tabletitle="Importresultat", rowtitle='Result', breadcrumbs=breadcrumbs)
def startAsyncImport(api_key, groupid, semester_key, user, request):
taskProgress = TaskProgress(name='Import', return_url=request.url)
taskProgress.put()
deferred.defer(importTask, api_key, groupid, semester_key, taskProgress.key, user.key)
return redirect('/progress/' + taskProgress.key.urlsafe())
def importTask(api_key, groupid, semester_key, taskProgress_key, user_key):
semester=semester_key.get()
user = user_key.get()
progress = None
for i in range(1, 3):
progress = taskProgress_key.get()
if progress is not None:
break
time.sleep(1) # wait for the eventual consistency
try:
success = RunScoutnetImport(groupid, api_key, user, semester, progress)
if not success:
progress.info("Importen misslyckades")
progress.failed = True
else:
progress.info("Import klar")
except Exception as e: # catch all exceptions so that defer stops running it again (automatic retry)
progress.info("Importfel: " + str(e))
progress.done()