views.py 38.1 KB
Newer Older
1
2


Ossi Laine's avatar
Ossi Laine committed
3
4
5
6
from flask_cors import CORS, cross_origin
from app import socketio
from flask_socketio import emit
import embody_plot
7
8
import os
import secrets
9
import json
10
11

from flask import (
Ossi Laine's avatar
Ossi Laine committed
12
13
14
15
16
17
18
    Flask,
    render_template,
    request,
    session,
    flash,
    redirect,
    url_for,
19
20
    Blueprint,
    jsonify
21
22
23
24
25
)

from wtforms import Form
from sqlalchemy import and_, update
from flask_login import login_required
26
from werkzeug import secure_filename
27

Ossi Laine's avatar
Ossi Laine committed
28
from app import app, db
29
30
31
32
33
34
35
from app.routes import APP_ROOT
from app.models import background_question, experiment
from app.models import background_question_answer
from app.models import page, question
from app.models import background_question_option
from app.models import answer_set, answer, forced_id
from app.models import user, trial_randomization
Ossi Laine's avatar
Ossi Laine committed
36
from app.models import embody_answer, embody_question
37
from app.forms import (
Ossi Laine's avatar
Ossi Laine committed
38
39
40
41
    CreateBackgroundQuestionForm,
    CreateQuestionForm, UploadStimuliForm, EditBackgroundQuestionForm,
    EditQuestionForm, EditExperimentForm, UploadResearchBulletinForm,
    EditPageForm, RemoveExperimentForm, GenerateIdForm, CreateEmbodyForm
42
)
Ossi Laine's avatar
Ossi Laine committed
43
from app.utils import get_mean_from_slider_answers, map_answers_to_questions
44

45

Ossi Laine's avatar
Ossi Laine committed
46
# Stimuli upload folder setting
47
48
#APP_ROOT = os.path.dirname(os.path.abspath(__file__))

Ossi Laine's avatar
Ossi Laine committed
49
50
51
52
experiment_blueprint = Blueprint("experiment", __name__,
                                 template_folder='templates',
                                 # static_folder='static',
                                 url_prefix='/experiment')
53

Ossi Laine's avatar
Ossi Laine committed
54
55
56
# Set sliders/embody:
DEFAULT_EMBODY_PICTURE = '/static/img/dummy_600.png'
DEFAULT_EMBODY_QUESTION = 'Color the regions whose activity you feel increasing or getting stronger'
57

Ossi Laine's avatar
Ossi Laine committed
58

59
60
61
@experiment_blueprint.route('/view')
@login_required
def view():
Ossi Laine's avatar
Ossi Laine committed
62
63

    # crap:3lines
64
65
66
    exp_id = request.args.get('exp_id', None)
    media = page.query.filter_by(experiment_idexperiment=exp_id).all()

Ossi Laine's avatar
Ossi Laine committed
67
    # stimulus type
68
    mtype = page.query.filter_by(experiment_idexperiment=exp_id).first()
Ossi Laine's avatar
Ossi Laine committed
69

Ossi Laine's avatar
Ossi Laine committed
70
71
72
73
    # experiment info
    experiment_info = experiment.query.filter_by(idexperiment=exp_id).all()

    # background questions
74
    questions_and_options = {}
Ossi Laine's avatar
Ossi Laine committed
75
76
77
78
79
80
81
82
83
84
85
    questions = background_question.query.filter_by(
        experiment_idexperiment=exp_id).all()

    for q in questions:

        options = background_question_option.query.filter_by(
            background_question_idbackground_question=q.idbackground_question).all()
        options_list = [(o.option, o.idbackground_question_option)
                        for o in options]
        questions_and_options[q.idbackground_question,
                              q.background_question] = options_list
86
87

    questions1 = questions_and_options
Ossi Laine's avatar
Ossi Laine committed
88

Ossi Laine's avatar
Ossi Laine committed
89
    # sliderset
90
91
92
    categories_and_scales = {}
    categories = question.query.filter_by(experiment_idexperiment=exp_id).all()

Ossi Laine's avatar
Ossi Laine committed
93
94
    for cat in categories:

95
        scale_list = [(cat.left, cat.right)]
Ossi Laine's avatar
Ossi Laine committed
96
97
        categories_and_scales[cat.idquestion, cat.question] = scale_list

98
99
    categories1 = categories_and_scales

Ossi Laine's avatar
Ossi Laine committed
100
101
102
103
104
    # embody pictures
    embody_pictures = embody_question.query.filter_by(
        experiment_idexperiment=exp_id).all()

    return render_template('view_experiment.html', exp_id=exp_id, media=media, mtype=mtype, experiment_info=experiment_info, categories1=categories1, questions1=questions1, embody_pictures=embody_pictures)
105
106
107
108
109
110
111
112
113
114
115
116


# Experiment info:

@experiment_blueprint.route('/remove', methods=['GET', 'POST'])
@login_required
def remove():

    exp_id = request.args.get('exp_id', None)
    exp_status = experiment.query.filter_by(idexperiment=exp_id).first()

    if exp_status.status != 'Hidden':
Ossi Laine's avatar
Ossi Laine committed
117

118
119
        flash("Experiment is public. Cannot modify structure.")
        return redirect(url_for('experiment.view', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
120

121
    else:
Ossi Laine's avatar
Ossi Laine committed
122

123
        form = RemoveExperimentForm(request.form)
Ossi Laine's avatar
Ossi Laine committed
124

125
        if request.method == 'POST' and form.validate():
Ossi Laine's avatar
Ossi Laine committed
126

127
            if form.remove.data == 'DELETE':
Ossi Laine's avatar
Ossi Laine committed
128
129
130
131
132
133
134

                # This removes all experiment data from the database!

                # Remove research bulletin if it exists
                empty_filevariable = experiment.query.filter_by(
                    idexperiment=exp_id).first()

135
                if empty_filevariable.research_notification_filename is not None:
Ossi Laine's avatar
Ossi Laine committed
136
137
138
139
                    target = os.path.join(
                        APP_ROOT, empty_filevariable.research_notification_filename)

                    if os.path.exists(target):
140
                        os.remove(target)
Ossi Laine's avatar
Ossi Laine committed
141
142
143
144

                # Tables
                remove_forced_id = forced_id.query.filter_by(
                    experiment_idexperiment=exp_id).all()
Ossi Laine's avatar
Ossi Laine committed
145
                remove_rows(remove_forced_id)
146

Ossi Laine's avatar
Ossi Laine committed
147
148
149
150
151
                # background_question_option & background_question & background question answers:
                remove_background_question = background_question.query.filter_by(
                    experiment_idexperiment=exp_id).all()

                # Remove all background questions and all answers given to each bg question
152
                for a in range(len(remove_background_question)):
Ossi Laine's avatar
Ossi Laine committed
153
154
                    remove_background_question_option = background_question_option.query.filter_by(
                        background_question_idbackground_question=remove_background_question[a].idbackground_question).all()
Ossi Laine's avatar
Ossi Laine committed
155
156
                    remove_rows(remove_background_question_option)

Ossi Laine's avatar
Ossi Laine committed
157
158
                    remove_background_question_answers = background_question_answer.query.filter_by(
                        background_question_idbackground_question=remove_background_question[a].idbackground_question).all()
Ossi Laine's avatar
Ossi Laine committed
159
160
                    remove_rows(remove_background_question_answers)

161
162
                    db.session.delete(remove_background_question[a])
                    db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
163
164
165
166

                # Remove all questions and answers
                remove_question = question.query.filter_by(
                    experiment_idexperiment=exp_id).all()
167
                for a in range(len(remove_question)):
Ossi Laine's avatar
Ossi Laine committed
168
169
                    remove_question_answers = answer.query.filter_by(
                        question_idquestion=remove_question[a].idquestion).all()
Ossi Laine's avatar
Ossi Laine committed
170
                    remove_rows(remove_question_answers)
171
172
                    db.session.delete(remove_question[a])
                    db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
173
174
175
176
177

                # Remove experiment embody pictures, questions and answers
                remove_embody_answers = embody_answer.query.filter(embody_answer.page_idpage.in_(list(map(
                    lambda x: x[0], page.query.with_entities(page.idpage).filter_by(experiment_idexperiment=exp_id).all())))).all()
                remove_rows(remove_embody_answers)
Ossi Laine's avatar
Ossi Laine committed
178
179
                remove_embody_questions = embody_question.query.filter_by(
                    experiment_idexperiment=exp_id).all()
Ossi Laine's avatar
Ossi Laine committed
180
181
182

                for a in range(len(remove_embody_questions)):
                    target = APP_ROOT + remove_embody_questions[a].picture
Ossi Laine's avatar
Ossi Laine committed
183
                    if os.path.exists(target) and DEFAULT_EMBODY_PICTURE != remove_embody_questions[a].picture:
Ossi Laine's avatar
Ossi Laine committed
184
185
                        os.remove(target)

Ossi Laine's avatar
Ossi Laine committed
186
187
                remove_rows(remove_embody_questions)

Ossi Laine's avatar
Ossi Laine committed
188
189
190
191
                # Remove all pages and datafiles
                remove_pages = page.query.filter_by(
                    experiment_idexperiment=exp_id).all()

192
                for a in range(len(remove_pages)):
Ossi Laine's avatar
Ossi Laine committed
193
                    if remove_pages[a].type != 'text':
194
                        target = os.path.join(APP_ROOT, remove_pages[a].media)
Ossi Laine's avatar
Ossi Laine committed
195
                        if os.path.exists(target):
196
                            os.remove(target)
Ossi Laine's avatar
Ossi Laine committed
197
198

                    # Now that the files are removed we can delete the page
199
200
                    db.session.delete(remove_pages[a])
                    db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
201

Ossi Laine's avatar
Ossi Laine committed
202
203
204
205
                # Remove all answer_sets and trial_randomization orders
                remove_answer_set = answer_set.query.filter_by(
                    experiment_idexperiment=exp_id).all()

206
                for a in range(len(remove_answer_set)):
Ossi Laine's avatar
Ossi Laine committed
207
208
                    remove_trial_randomizations = trial_randomization.query.filter_by(
                        answer_set_idanswer_set=remove_answer_set[a].idanswer_set).all()
Ossi Laine's avatar
Ossi Laine committed
209
                    remove_rows(remove_trial_randomizations)
210
211
                    db.session.delete(remove_answer_set[a])
                    db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
212
213
214
215

                # Remove experiment table
                remove_experiment = experiment.query.filter_by(
                    idexperiment=exp_id).first()
216
217
                db.session.delete(remove_experiment)
                db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
218
219
220
221

                # Remove empty directories
                os.rmdir(APP_ROOT + '/static/embody_images/' + str(exp_id))
                os.rmdir(APP_ROOT + '/static/experiment_stimuli/' + str(exp_id))
Ossi Laine's avatar
Ossi Laine committed
222

223
224
                flash("Experiment was removed from database!")
                return redirect(url_for('index'))
Ossi Laine's avatar
Ossi Laine committed
225

226
227
228
            else:
                flash("Experiment was not removed!")
                return redirect(url_for('experiment.view', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
229

230
231
232
233
234
235
236
        return render_template('remove_experiment.html', form=form, exp_id=exp_id)


@experiment_blueprint.route('/publish')
@login_required
def publish():
    exp_id = request.args.get('exp_id', None)
Ossi Laine's avatar
Ossi Laine committed
237
238
    publish_experiment = experiment.query.filter_by(
        idexperiment=exp_id).first()
239
240
241
242
243
244
245
246
247
248
    publish_experiment.status = 'Public'
    flash("Changed status to Public")
    db.session.commit()
    return redirect(url_for('experiment.view', exp_id=exp_id))


@experiment_blueprint.route('/hide')
@login_required
def hide():
    exp_id = request.args.get('exp_id', None)
Ossi Laine's avatar
Ossi Laine committed
249
    hide_experiment = experiment.query.filter_by(idexperiment=exp_id).first()
250
251
252
253
254
255
256
257
258
259
    hide_experiment.status = 'Hidden'
    flash("Changed status to Hidden")
    db.session.commit()
    return redirect(url_for('experiment.view', exp_id=exp_id))


@experiment_blueprint.route('/private')
@login_required
def private():
    exp_id = request.args.get('exp_id', None)
Ossi Laine's avatar
Ossi Laine committed
260
261
    private_experiment = experiment.query.filter_by(
        idexperiment=exp_id).first()
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
    private_experiment.status = 'Private'
    flash("Changed status to Private")
    db.session.commit()
    return redirect(url_for('experiment.view', exp_id=exp_id))


@experiment_blueprint.route('/randomization')
@login_required
def randomization():
    exp_id = request.args.get('exp_id', None)
    status = request.args.get('set')

    if status == 'On':
        flash("Enabled trial randomization")
    elif status == 'Off':
        flash("Disabled trial randomization")

Ossi Laine's avatar
Ossi Laine committed
279
280
    experiment.query.filter_by(
        idexperiment=exp_id).first().randomization = status
281
    db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
282

283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
    return redirect(url_for('experiment.view', exp_id=exp_id))


@experiment_blueprint.route('/set_forced_id')
@login_required
def set_forced_id():
    '''By using forced ID login subjects can only participate to a rating task
    by logging in with a pregenerated ID.'''

    exp_id = request.args.get('exp_id', None)
    status = request.args.get('set')

    if status == 'On':
        flash("Enabled forced ID login")
    elif status == 'Off':
        flash("Disabled forced ID login")

Ossi Laine's avatar
Ossi Laine committed
300
301
    experiment.query.filter_by(
        idexperiment=exp_id).first().use_forced_id = status
302
    db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
303

304
305
306
307
308
309
310
311
312
313
314
    return redirect(url_for('experiment.view', exp_id=exp_id))


@experiment_blueprint.route('/view_forced_id_list', methods=['GET', 'POST'])
@login_required
def view_forced_id_list():
    '''Forced ID login for participants'''

    exp_id = request.args.get('exp_id', None)
    id_list = forced_id.query.filter_by(experiment_idexperiment=exp_id).all()
    form = GenerateIdForm(request.form)
Ossi Laine's avatar
Ossi Laine committed
315

316
317
    if request.method == 'POST' and form.validate():

Ossi Laine's avatar
Ossi Laine committed
318
        for i in range(int(request.form['number'])):
319
            random_id = str(request.form['string']) + str(secrets.token_hex(3))
Ossi Laine's avatar
Ossi Laine committed
320
321
322
323
324
325
326
            check_answer_set = answer_set.query.filter_by(
                session=random_id).first()
            check_forced_id = forced_id.query.filter_by(
                pregenerated_id=random_id).first()

            # here we check if the generated id is found from given answers from the whole database in answer_set table
            # or from forced_id table. If so another id is generated instead to avoid a duplicate
327
            if check_answer_set is not None or check_forced_id is not None:
Ossi Laine's avatar
Ossi Laine committed
328

329
330
                #flash("ID already existed; generated a new one")
                random_id = secrets.token_hex(3)
Ossi Laine's avatar
Ossi Laine committed
331
332
333
334
335
336
337
                check_answer_set = answer_set.query.filter_by(
                    session=random_id).first()
                check_forced_id = forced_id.query.filter_by(
                    pregenerated_id=random_id).first()

            input_id = forced_id(
                experiment_idexperiment=exp_id, pregenerated_id=random_id)
338
339
            db.session.add(input_id)
            db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
340

341
        return redirect(url_for('experiment.view_forced_id_list', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
342

343
344
345
346
347
348
349
350
351
352
    return render_template('view_forced_id_list.html', exp_id=exp_id, id_list=id_list)


@experiment_blueprint.route('/upload_research_notification', methods=['GET', 'POST'])
@login_required
def upload_research_notification():
    '''Upload research bulletin'''

    exp_id = request.args.get('exp_id', None)
    form = UploadResearchBulletinForm(request.form)
Ossi Laine's avatar
Ossi Laine committed
353

354
355
356
    if request.method == 'POST':
        path = 'static/experiment_stimuli/' + str(exp_id)
        target = os.path.join(APP_ROOT, path)
Ossi Laine's avatar
Ossi Laine committed
357

358
359
        if not os.path.isdir(target):
            os.mkdir(target)
Ossi Laine's avatar
Ossi Laine committed
360
361

        # This returns a list of filenames: request.files.getlist("file")
362
        for file in request.files.getlist("file"):
Ossi Laine's avatar
Ossi Laine committed
363
            # save files in the correct folder
364
365
366
            filename = file.filename
            destination = "/".join([target, filename])
            file.save(destination)
Ossi Laine's avatar
Ossi Laine committed
367
368
369

            # add pages to the db
            db_path = path + str('/') + str(filename)
370
371
372
            bulletin = experiment.query.filter_by(idexperiment=exp_id).first()
            bulletin.research_notification_filename = db_path
            db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
373

374
375
376
377
378
379
380
381
382
        return redirect(url_for('experiment.view', exp_id=exp_id))

    return render_template('upload_research_notification.html', exp_id=exp_id, form=form)


@experiment_blueprint.route('/remove_research_notification')
@login_required
def remove_research_notification():
    '''Remove research bulletin'''
Ossi Laine's avatar
Ossi Laine committed
383

384
    exp_id = request.args.get('exp_id', None)
Ossi Laine's avatar
Ossi Laine committed
385
386
387
388
389
390
    empty_filevariable = experiment.query.filter_by(
        idexperiment=exp_id).first()
    target = os.path.join(
        APP_ROOT, empty_filevariable.research_notification_filename)

    if os.path.exists(target):
391
        os.remove(target)
Ossi Laine's avatar
Ossi Laine committed
392

393
394
    empty_filevariable.research_notification_filename = None
    db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
395

396
397
398
399
400
401
402
403
404
    return redirect(url_for('experiment.view', exp_id=exp_id))


@experiment_blueprint.route('/edit', methods=['GET', 'POST'])
@login_required
def edit():
    '''Edit experiment details'''

    exp_id = request.args.get('exp_id', None)
Ossi Laine's avatar
Ossi Laine committed
405
406
407
    current_experiment = experiment.query.filter_by(
        idexperiment=exp_id).first()

408
409
    form = EditExperimentForm(request.form, obj=current_experiment)
    form.language.default = current_experiment.language
Ossi Laine's avatar
Ossi Laine committed
410

411
    if request.method == 'POST' and form.validate():
Ossi Laine's avatar
Ossi Laine committed
412

413
414
415
        form.populate_obj(current_experiment)
        db.session.commit()

Ossi Laine's avatar
Ossi Laine committed
416
        return redirect(url_for('experiment.view', exp_id=exp_id))
417

Ossi Laine's avatar
Ossi Laine committed
418
    return render_template('edit_experiment.html', form=form, exp_id=exp_id)
419
420
421
422
423
424
425


# Background questions:

@experiment_blueprint.route('/add_bg_question', methods=['GET', 'POST'])
@login_required
def add_bg_question():
Ossi Laine's avatar
Ossi Laine committed
426

427
428
429
430
431
432
    exp_id = request.args.get('exp_id', None)
    exp_status = experiment.query.filter_by(idexperiment=exp_id).first()

    if exp_status.status != 'Hidden':
        flash("Experiment is public. Cannot modify structure.")
        return redirect(url_for('experiment.view', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
433

434
435
    else:
        form = CreateBackgroundQuestionForm(request.form)
Ossi Laine's avatar
Ossi Laine committed
436

437
        if request.method == 'POST' and form.validate():
Ossi Laine's avatar
Ossi Laine committed
438
439

            # Split the form data into a list that separates questions followed by the corresponding options
440
            str = form.bg_questions_and_options.data
441
            str_list = str.split('/n')
Ossi Laine's avatar
Ossi Laine committed
442
443

            # Iterate through the questions and options list
444
            for a in range(len(str_list)):
Ossi Laine's avatar
Ossi Laine committed
445
                # Split the list cells further into questions and options
446
                list = str_list[a].split(';')
Ossi Laine's avatar
Ossi Laine committed
447
448

                # Input the first item of the list as a question in db and the items followed by that as options for that question
449
450
                for x in range(len(list)):
                    if x == 0:
Ossi Laine's avatar
Ossi Laine committed
451
452
                        add_bgquestion = background_question(
                            background_question=list[x], experiment_idexperiment=exp_id)
453
454
455
                        db.session.add(add_bgquestion)
                        db.session.commit()
                    else:
Ossi Laine's avatar
Ossi Laine committed
456
457
                        add_bgq_option = background_question_option(
                            background_question_idbackground_question=add_bgquestion.idbackground_question, option=list[x])
458
459
                        db.session.add(add_bgq_option)
                        db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
460
461
462

            return redirect(url_for('experiment.view', exp_id=exp_id))

463
464
465
466
467
468
        return render_template('add_bg_question.html', form=form)


@experiment_blueprint.route('/edit_bg_question', methods=['GET', 'POST'])
@login_required
def edit_bg_question():
Ossi Laine's avatar
Ossi Laine committed
469

470
471
    bg_question_id = request.args.get('idbackground_question', None)

Ossi Laine's avatar
Ossi Laine committed
472
473
474
475
476
    # Search for the right question and for the right options. Form a string of those separated with ";" and insert the
    # formed string into the edit form
    current_bg_question = background_question.query.filter_by(
        idbackground_question=bg_question_id).first()
    exp_id = current_bg_question.experiment_idexperiment
477
    question_string = current_bg_question.background_question
Ossi Laine's avatar
Ossi Laine committed
478
479
480
481
482
483
484
    options = background_question_option.query.filter_by(
        background_question_idbackground_question=bg_question_id).all()

    for o in range(len(options)):
        question_string = str(question_string) + \
            str("; ") + str(options[o].option)

485
486
487
    form = EditBackgroundQuestionForm(request.form)
    form.bg_questions_and_options.data = question_string

Ossi Laine's avatar
Ossi Laine committed
488
    # After user chooses to update the question and options lets replace the old question and options with the ones from the form
489
    if request.method == 'POST' and form.validate():
Ossi Laine's avatar
Ossi Laine committed
490
        # Explode the string with new values from the form
491
492
        form_values = form.new_values.data
        form_values_list = form_values.split(';')
Ossi Laine's avatar
Ossi Laine committed
493
494

        # Check and remove possible whitespaces from string beginnings with lstrip
495
496
        for x in range(len(form_values_list)):
            form_values_list[x] = form_values_list[x].lstrip()
Ossi Laine's avatar
Ossi Laine committed
497
498

        # Cycle through strings and update db
499
        for x in range(len(form_values_list)):
Ossi Laine's avatar
Ossi Laine committed
500
501

            # Replace question and update the object to database
502
            if x == 0:
Ossi Laine's avatar
Ossi Laine committed
503
                current_bg_question.background_question = form_values_list[x]
504
                db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
505
506

                # Delete old options from db
507
508
509
                for o in options:
                    db.session.delete(o)
                    db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
510
511

            # Insert new options to db
512
            else:
Ossi Laine's avatar
Ossi Laine committed
513
514
                new_option = background_question_option(
                    background_question_idbackground_question=current_bg_question.idbackground_question, option=form_values_list[x])
515
516
                db.session.add(new_option)
                db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
517

518
        return redirect(url_for('experiment.view', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
519

520
521
522
523
524
525
526
527
528
529
530
531
532
    return render_template('edit_bg_question.html', form=form, exp_id=exp_id)


@experiment_blueprint.route('/remove_bg_question')
@login_required
def remove_bg_question():

    exp_id = request.args.get('exp_id', None)
    exp_status = experiment.query.filter_by(idexperiment=exp_id).first()

    if exp_status.status != 'Hidden':
        flash("Experiment is public. Cannot modify structure.")
        return redirect(url_for('experiment.view', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
533

534
    else:
535

536
        remove_id = request.args.get('idbackground_question', None)
Ossi Laine's avatar
Ossi Laine committed
537
538
539
540
        remove_options = background_question_option.query.filter_by(
            background_question_idbackground_question=remove_id).all()

        for a in range(len(remove_options)):
541
542
            db.session.delete(remove_options[a])
            db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
543
544
545
546

        remove_question = background_question.query.filter_by(
            idbackground_question=remove_id).first()

547
548
549
        db.session.delete(remove_question)
        db.session.commit()

Ossi Laine's avatar
Ossi Laine committed
550
    return redirect(url_for('experiment.view', exp_id=exp_id))
551
552


553
554
555
556
557
@experiment_blueprint.route('/set_embody')
@login_required
def set_embody():
    '''Enable/disable embody tool'''
    exp_id = request.args.get('exp_id', None)
Ossi Laine's avatar
Ossi Laine committed
558
    exp = experiment.query.filter_by(idexperiment=exp_id).first()
559
560
561
562
    exp.embody_enabled = (True if exp.embody_enabled == False else False)
    return redirect(url_for('experiment.view', exp_id=exp_id))


Ossi Laine's avatar
Ossi Laine committed
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
@experiment_blueprint.route('/add_embody', methods=['GET', 'POST'])
@login_required
def add_embody():

    exp_id = request.args.get('exp_id', None)
    default = request.args.get('default', None)
    exp_info = experiment.query.filter_by(idexperiment=exp_id).first()

    if exp_info.status != 'Hidden':
        flash("Experiment is public. Cannot modify structure.")
        return redirect(url_for('experiment.view', exp_id=exp_id))
    elif default:

        # TODO: check if default image already added

Ossi Laine's avatar
Ossi Laine committed
578
579
580
        default_embody = embody_question(
            experiment_idexperiment=exp_id, picture=DEFAULT_EMBODY_PICTURE, question=DEFAULT_EMBODY_QUESTION)
        db.session.add(default_embody)
Ossi Laine's avatar
Ossi Laine committed
581
        exp_info.embody_enabled = 1
Ossi Laine's avatar
Ossi Laine committed
582
        db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
583
584
585
586
587
588
589
590
591
        return redirect(url_for('experiment.view', exp_id=exp_id))

    else:
        form = CreateEmbodyForm(request.form)

        if request.method == 'POST' and form.validate():
            picture = request.files.get("picture")
            question = request.form.get("question")

Ossi Laine's avatar
Ossi Laine committed
592
            # get filename
Ossi Laine's avatar
Ossi Laine committed
593
594
            filename = secure_filename(picture.filename)
            path = 'static/embody_images/' + str(exp_id)
Ossi Laine's avatar
Ossi Laine committed
595
            db_path = '/' + path + str('/') + str(filename)
Ossi Laine's avatar
Ossi Laine committed
596
597
598
599
600
601
602
603
604
605
            target = os.path.join(APP_ROOT, path)

            # create folder with experiment id (if it does not exist)
            if not os.path.isdir(target):
                os.mkdir(target)

            # save file to server
            destination = "/".join([target, filename])
            picture.save(destination)

Ossi Laine's avatar
Ossi Laine committed
606
607
608
            # add pages to the db
            new_embody = embody_question(
                experiment_idexperiment=exp_id, question=question, picture=db_path)
Ossi Laine's avatar
Ossi Laine committed
609
610
611
            db.session.add(new_embody)
            exp_info.embody_enabled = 1
            db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
612
            return redirect(url_for('experiment.view', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
613
614
615
616

        return render_template('add_embody.html', form=form)


617
618
619
@experiment_blueprint.route('/add_questions', methods=['GET', 'POST'])
@login_required
def add_questions():
Ossi Laine's avatar
Ossi Laine committed
620

621
622
623
624
625
626
627
628
    exp_id = request.args.get('exp_id', None)
    exp_status = experiment.query.filter_by(idexperiment=exp_id).first()

    if exp_status.status != 'Hidden':
        flash("Experiment is public. Cannot modify structure.")
        return redirect(url_for('experiment.view', exp_id=exp_id))
    else:
        form = CreateQuestionForm(request.form)
Ossi Laine's avatar
Ossi Laine committed
629

630
        if request.method == 'POST' and form.validate():
Ossi Laine's avatar
Ossi Laine committed
631

632
633
            str = form.questions_and_options.data
            str_list = str.split('/n')
Ossi Laine's avatar
Ossi Laine committed
634
635

            for a in range(len(str_list)):
636
                list = str_list[a].split(';')
Ossi Laine's avatar
Ossi Laine committed
637
638

                # If there are the right amount of values for the slider input values
639
                if len(list) == 3:
Ossi Laine's avatar
Ossi Laine committed
640
641
                    add_question = question(
                        experiment_idexperiment=exp_id, question=list[0], left=list[1], right=list[2])
642
643
                    db.session.add(add_question)
                    db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
644
645

                # If slider has too many or too litlle parameters give an error and redirect back to input form
646
                else:
Ossi Laine's avatar
Ossi Laine committed
647
648
                    flash(
                        "Error Each slider must have 3 parameters separated by ; Some slider has:")
649
650
                    flash(len(list))
                    return redirect(url_for('create.experiment_questions', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
651
652
653

            return redirect(url_for('experiment.view', exp_id=exp_id))

654
655
656
657
658
659
        return render_template('add_questions.html', form=form)


@experiment_blueprint.route('/edit_question', methods=['GET', 'POST'])
@login_required
def edit_question():
Ossi Laine's avatar
Ossi Laine committed
660

661
662
663
    question_id = request.args.get('idquestion', None)
    current_question = question.query.filter_by(idquestion=question_id).first()
    form = EditQuestionForm(request.form, obj=current_question)
Ossi Laine's avatar
Ossi Laine committed
664

665
    if request.method == 'POST' and form.validate():
Ossi Laine's avatar
Ossi Laine committed
666

667
668
        form.populate_obj(current_question)
        db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
669

670
        return redirect(url_for('experiment.view', exp_id=current_question.experiment_idexperiment))
Ossi Laine's avatar
Ossi Laine committed
671

672
673
674
675
676
677
678
679
680
681
682
683
684
    return render_template('edit_question.html', form=form)


@experiment_blueprint.route('/remove_question')
@login_required
def remove_question():

    exp_id = request.args.get('exp_id', None)
    exp_status = experiment.query.filter_by(idexperiment=exp_id).first()

    if exp_status.status != 'Hidden':
        flash("Experiment is public. Cannot modify structure.")
        return redirect(url_for('experiment.view', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
685

686
687
688
    else:

        remove_id = request.args.get('idquestion', None)
Ossi Laine's avatar
Ossi Laine committed
689
690
        remove_question = question.query.filter_by(
            idquestion=remove_id).first()
Ossi Laine's avatar
Ossi Laine committed
691
692

        # remove answers before removing questions
Ossi Laine's avatar
Ossi Laine committed
693
694
        remove_answers = answer.query.filter_by(
            question_idquestion=remove_question.idquestion).all()
Ossi Laine's avatar
Ossi Laine committed
695
696
697
        for a in range(len(remove_answers)):
            db.session.delete(remove_answers[a])
            db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
698

Ossi Laine's avatar
Ossi Laine committed
699
700
        db.session.delete(remove_question)
        db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
701

Ossi Laine's avatar
Ossi Laine committed
702
703
704
705
706
707
708
709
710
711
712
713
714
    return redirect(url_for('experiment.view', exp_id=exp_id))


@experiment_blueprint.route('/remove_embody')
@login_required
def remove_embody():

    exp_id = request.args.get('exp_id', None)
    exp_status = experiment.query.filter_by(idexperiment=exp_id).first()

    if exp_status.status != 'Hidden':
        flash("Experiment is public. Cannot modify structure.")
        return redirect(url_for('experiment.view', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
715

Ossi Laine's avatar
Ossi Laine committed
716
717
    else:
        remove_id = request.args.get('idembody', None)
Ossi Laine's avatar
Ossi Laine committed
718
719
        remove_question = embody_question.query.filter_by(
            idembody=remove_id).first()
Ossi Laine's avatar
Ossi Laine committed
720
721
722
723
724

        # remove embody image from server
        if DEFAULT_EMBODY_PICTURE != remove_question.picture:
            os.remove(APP_ROOT + remove_question.picture)

Ossi Laine's avatar
Ossi Laine committed
725
726
        remove_answers = embody_answer.query.filter_by(
            embody_question_idembody=remove_question.idembody).all()
Ossi Laine's avatar
Ossi Laine committed
727
        remove_rows(remove_answers)
728
729
        db.session.delete(remove_question)
        db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
730

Ossi Laine's avatar
Ossi Laine committed
731
732
    question_count = embody_question.query.filter_by(
        experiment_idexperiment=exp_id).count()
Ossi Laine's avatar
Ossi Laine committed
733
734
735
736

    if question_count == 0:
        exp_status.embody_enabled = 0
        db.session.commit()
737

Ossi Laine's avatar
Ossi Laine committed
738
    return redirect(url_for('experiment.view', exp_id=exp_id))
739
740
741
742
743
744
745
746
747
748


# Stimuli:

@experiment_blueprint.route('/add_stimuli', methods=['GET', 'POST'])
@login_required
def add_stimuli():

    exp_id = request.args.get('exp_id', None)
    exp_status = experiment.query.filter_by(idexperiment=exp_id).first()
Ossi Laine's avatar
Ossi Laine committed
749

750
751
752
753
    if exp_status.status != 'Hidden':
        flash("Experiment is public. Cannot modify structure.")
        return redirect(url_for('experiment.view', exp_id=exp_id))
    else:
Ossi Laine's avatar
Ossi Laine committed
754
755
756
757
        # If there are no pages set for the experiment lets reroute user to create experiment stimuli upload instead
        is_there_any_stimuli = page.query.filter_by(
            experiment_idexperiment=exp_id).first()

758
759
        if is_there_any_stimuli is None:
            return redirect(url_for('create.experiment_upload_stimuli', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
760

761
762
        stimulus_type = request.args.get('stimulus_type', None)
        form = UploadStimuliForm(request.form)
Ossi Laine's avatar
Ossi Laine committed
763

764
765
766
767
        if request.method == 'POST':
            if stimulus_type == 'text':
                string = form.text.data
                str_list = string.split('/n')
Ossi Laine's avatar
Ossi Laine committed
768

769
                for a in range(len(str_list)):
Ossi Laine's avatar
Ossi Laine committed
770
771
                    add_text_stimulus = page(
                        experiment_idexperiment=exp_id, type='text', text=str_list[a], media='none')
772
773
                    db.session.add(add_text_stimulus)
                    db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
774
775
776

                return redirect(url_for('experiment.view', exp_id=exp_id))

777
            else:
Ossi Laine's avatar
Ossi Laine committed
778
779
                # Upload stimuli into /static/experiment_stimuli/exp_id folder
                # Create the pages for the stimuli by inserting experiment_id, stimulus type, text and names of the stimulus files (as a path to the folder)
780
781
                path = 'static/experiment_stimuli/' + str(exp_id)
                target = os.path.join(APP_ROOT, path)
Ossi Laine's avatar
Ossi Laine committed
782

783
784
                if not os.path.isdir(target):
                    os.mkdir(target)
Ossi Laine's avatar
Ossi Laine committed
785
786

                # This returns a list of filenames: request.files.getlist("file")
787
                for file in request.files.getlist("file"):
Ossi Laine's avatar
Ossi Laine committed
788
                    # save files in the correct folder
789
790
791
                    filename = file.filename
                    destination = "/".join([target, filename])
                    file.save(destination)
Ossi Laine's avatar
Ossi Laine committed
792

Ossi Laine's avatar
Ossi Laine committed
793
794
795
796
                    # add pages to the db
                    db_path = path + str('/') + str(filename)
                    new_page = page(experiment_idexperiment=exp_id,
                                    type=form.type.data, media=db_path)
797
798
                    db.session.add(new_page)
                    db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
799

800
                return redirect(url_for('experiment.view', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
801

802
803
804
805
806
807
808
809
810
811
812
813
            return redirect(url_for('experiment.view', exp_id=exp_id))

        return render_template('add_stimuli.html', form=form, stimulus_type=stimulus_type)


@experiment_blueprint.route('/edit_stimuli', methods=['GET', 'POST'])
@login_required
def edit_stimuli():

    exp_id = request.args.get('exp_id', None)
    page_id = request.args.get('idpage', None)
    edit_page = page.query.filter_by(idpage=page_id).first()
814
815
816
    form = EditPageForm(request.form)
    #form = EditPageForm(request.form, obj=edit_page)

Ossi Laine's avatar
Ossi Laine committed
817
    # TODO: replacing image not working -> form not getting validated for some reaseon !!
818

819
    if request.method == 'POST' and form.validate():
820
        print("POST IMAGE")
Ossi Laine's avatar
Ossi Laine committed
821
        # If the stimulus type is not text, then the old stimulus file is deleted from os and replaced
822
        if edit_page.type != 'text':
Ossi Laine's avatar
Ossi Laine committed
823
            # remove old file
824
825
826
            target = os.path.join(APP_ROOT, edit_page.media)
            os.remove(target)

Ossi Laine's avatar
Ossi Laine committed
827
            # upload new file
828
829
            path = 'static/experiment_stimuli/' + str(exp_id)
            target = os.path.join(APP_ROOT, path)
Ossi Laine's avatar
Ossi Laine committed
830

831
832
            if not os.path.isdir(target):
                os.mkdir(target)
Ossi Laine's avatar
Ossi Laine committed
833
834

            # This returns a list of filenames: request.files.getlist("file")
835
            for file in request.files.getlist("file"):
Ossi Laine's avatar
Ossi Laine committed
836
                # save files in the correct folder
837
838
839
                filename = file.filename
                destination = "/".join([target, filename])
                file.save(destination)
Ossi Laine's avatar
Ossi Laine committed
840
841
842
843

                # update db object
                db_path = path + str('/') + str(filename)
                edit_page.media = db_path
844
845
                db.session.commit()

Ossi Laine's avatar
Ossi Laine committed
846
        # If editing text stimulus no need for filehandling
847
        else:
Ossi Laine's avatar
Ossi Laine committed
848
849
850
            form.populate_obj(edit_page)
            db.session.commit()

851
        return redirect(url_for('experiment.view', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
852

853
854
855
856
857
858
859
860
861
862
863
    return render_template('edit_stimuli.html', form=form, edit_page=edit_page)


@experiment_blueprint.route('/remove_stimuli')
@login_required
def remove_stimuli():

    exp_id = request.args.get('exp_id', None)
    exp_status = experiment.query.filter_by(idexperiment=exp_id).first()

    if exp_status.status != 'Hidden':
Ossi Laine's avatar
Ossi Laine committed
864

865
866
867
        flash("Experiment is public. Cannot modify structure.")

        return redirect(url_for('experiment.view', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
868

869
    else:
Ossi Laine's avatar
Ossi Laine committed
870

871
872
        remove_id = request.args.get('idpage', None)
        remove_page = page.query.filter_by(idpage=remove_id).first()
Ossi Laine's avatar
Ossi Laine committed
873
874
875
876
877
        experiment_pages = page.query.filter_by(
            experiment_idexperiment=exp_id).all()

        # if stimulustype is text, the stimulus itself is text on the database, other stimulus types are real files
        # on the server and need to be deleted
878
        if remove_page.type != 'text':
Ossi Laine's avatar
Ossi Laine committed
879
880

            # helper variable
881
            do_not_delete_file = 'False'
Ossi Laine's avatar
Ossi Laine committed
882
883

            # if the file to be deleted is in duplicate use of another page then we won't delete the file
884
885
886
            for a in range(len(experiment_pages)):
                if experiment_pages[a].media == remove_page.media and experiment_pages[a].idpage != remove_page.idpage:
                    do_not_delete_file = 'True'
Ossi Laine's avatar
Ossi Laine committed
887
888

            # If no other page is using the file then lets remove it
889
890
891
            if do_not_delete_file == 'False':
                target = os.path.join(APP_ROOT, remove_page.media)
                os.remove(target)
Ossi Laine's avatar
Ossi Laine committed
892

Ossi Laine's avatar
Ossi Laine committed
893
            # remove slider answer from this page
Ossi Laine's avatar
Ossi Laine committed
894
895
            remove_question_answers = answer.query.filter_by(
                page_idpage=remove_page.idpage).all()
Ossi Laine's avatar
Ossi Laine committed
896
897
898
            remove_rows(remove_question_answers)

            # remove embody answer from this page
Ossi Laine's avatar
Ossi Laine committed
899
900
            remove_embody_answers = embody_answer.query.filter_by(
                page_idpage=remove_page.idpage).all()
Ossi Laine's avatar
Ossi Laine committed
901
902
            remove_rows(remove_embody_answers)

903
904
            db.session.delete(remove_page)
            db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
905

906
            return redirect(url_for('experiment.view', exp_id=exp_id))
Ossi Laine's avatar
Ossi Laine committed
907

908
        if remove_page.type == 'text':
Ossi Laine's avatar
Ossi Laine committed
909

Ossi Laine's avatar
Ossi Laine committed
910
            # remove slider answer from this page
Ossi Laine's avatar
Ossi Laine committed
911
912
            remove_question_answers = answer.query.filter_by(
                page_idpage=remove_page.idpage).all()
Ossi Laine's avatar
Ossi Laine committed
913
914
915
            remove_rows(remove_question_answers)

            # remove embody answer from this page
Ossi Laine's avatar
Ossi Laine committed
916
917
            remove_embody_answers = embody_answer.query.filter_by(
                page_idpage=remove_page.idpage).all()
Ossi Laine's avatar
Ossi Laine committed
918
919
            remove_rows(remove_embody_answers)

920
921
            db.session.delete(remove_page)
            db.session.commit()
Ossi Laine's avatar
Ossi Laine committed
922

923
924
            return redirect(url_for('experiment.view', exp_id=exp_id))

Ossi Laine's avatar
Ossi Laine committed
925
        return redirect(url_for('experiment.view', exp_id=exp_id))
926
927
928
929
930
931
932


# Misc:

@experiment_blueprint.route('/statistics')
@login_required
def statistics():
933
934

    # TODO: Answers are in normal order although questions might be in randomized order
Ossi Laine's avatar
Ossi Laine committed
935

936
    exp_id = request.args.get('exp_id', None)
Ossi Laine's avatar
Ossi Laine committed
937
938
939
940
941
942

    experiment_info = experiment.query.filter_by(idexperiment=exp_id).all()
    participants = answer_set.query.filter_by(
        experiment_idexperiment=exp_id).all()

    # started and finished ratings counters
943
944
945
946
947
948
949
    started_ratings = answer_set.query.filter_by(
        experiment_idexperiment=exp_id).count()
    experiment_page_count = page.query.filter_by(
        experiment_idexperiment=exp_id).count()
    finished_ratings = answer_set.query.filter(and_(
        answer_set.answer_counter == experiment_page_count, answer_set.experiment_idexperiment == exp_id)).count()

Ossi Laine's avatar
Ossi Laine committed
950
951
952
953
954
955
    # Rating task headers
    question_headers = question.query.filter_by(
        experiment_idexperiment=exp_id).all()
    stimulus_headers = page.query.filter_by(
        experiment_idexperiment=exp_id).all()

956
957
958
959
960
961
    pages = page.query.filter_by(experiment_idexperiment=exp_id).all()
    questions = question.query.filter_by(experiment_idexperiment=exp_id).all()
    pages_and_questions = {}

    for p in pages:
        questions_list = [(p.idpage, a.idquestion) for a in questions]
Ossi Laine's avatar
Ossi Laine committed
962
963
964
965
        pages_and_questions[p.idpage] = questions_list

    # List of answers per participant in format question Stimulus ID/Question ID
    # those are in answer table as page_idpage and question_idquestion respectively
966
    slider_answers = {}
967
    for participant in participants:
968
        if participant.answer_counter > 0:
Ossi Laine's avatar
Ossi Laine committed
969
970
            answers = answer.query.filter_by(
                answer_set_idanswer_set=participant.idanswer_set)\
971
                .order_by(answer.page_idpage)\
Ossi Laine's avatar
Ossi Laine committed
972
973
974
975
976
977
978
979
                .all()

            # flatten pages and questions to list of tuples (page_id, question_id)
            _questions = [
                item for sublist in pages_and_questions.values() for item in sublist]

            slider_answers[participant.session] = map_answers_to_questions(
                answers, _questions)
980

981
    slider_answers['mean'] = get_mean_from_slider_answers(slider_answers)
982

Ossi Laine's avatar
Ossi Laine committed
983
    # Background question answers
Ossi Laine's avatar
Ossi Laine committed
984
985
    bg_questions = background_question.query.filter_by(
        experiment_idexperiment=exp_id).all()
986
987
988
    bg_answers_for_participants = {}

    for participant in participants:
989
        if participant.answer_counter > 0:
990
991
992
993
            bg_answers = background_question_answer.query.filter_by(
                answer_set_idanswer_set=participant.idanswer_set).all()
            bg_answers_list = [(a.answer) for a in bg_answers]
            bg_answers_for_participants[participant.session] = bg_answers_list
Ossi Laine's avatar
Ossi Laine committed
994
995
996
997
998

    # embody questions
    embody_questions = embody_question.query.filter_by(
        experiment_idexperiment=exp_id).all()

Ossi Laine's avatar
Ossi Laine committed
999
1000
1001
1002
1003
1004
1005
1006
1007
1008
1009
1010
    return render_template('experiment_statistics.html',
                           experiment_info=experiment_info,
                           participants_and_answers=slider_answers,
                           pages_and_questions=pages_and_questions,
                           bg_questions=bg_questions,
                           bg_answers_for_participants=bg_answers_for_participants,
                           started_ratings=started_ratings,
                           finished_ratings=finished_ratings,
                           question_headers=question_headers,
                           stimulus_headers=stimulus_headers,
                           embody_questions=embody_questions
                           )
1011
1012


root's avatar
root committed
1013
1014
1015
1016
1017
1018
1019
def remove_rows(rows):
    """Remove list of rows from database"""
    for a in range(len(rows)):
        db.session.delete(rows[a])
        db.session.commit()


1020
1021
1022
1023
@socketio.on('connect', namespace="/create_embody")
def create_embody():
    emit('success', {'connection': 'on'})

Ossi Laine's avatar
Ossi Laine committed
1024

1025
1026
@socketio.on('draw', namespace="/create_embody")
def create_embody(page_id):
1027
1028
1029

    print("DRAW")

1030
    page = page_id["page"]
Ossi Laine's avatar
Ossi Laine committed
1031
    embody = page_id["embody"]
1032

Ossi Laine's avatar
Ossi Laine committed
1033
    img_path = embody_plot.get_coordinates(page, embody)
root's avatar
root committed
1034
    app.logger.info(img_path)
Ossi Laine's avatar
Ossi Laine committed
1035
1036
    emit('end', {'path': img_path})

1037
1038
1039
1040
1041

@socketio.on('end', namespace="/create_embody")
def create_embody():
    print("connection end")
    emit('end', {'connection': 'off'})
Ossi Laine's avatar
Ossi Laine committed
1042
1043


1044
# EOF