exercise.cgi

#!/usr/bin/python3

import sys, cgi, cgitb, random, sqlite3, json, glob
from exercise import load_exercise_file, load_exercise_data, eval_exercise
cgitb.enable()
conn = sqlite3.connect('exercise.db')
conn.row_factory = sqlite3.Row
form = cgi.FieldStorage()

print("Content-Type: text/json\n")
datapath = "data"

def get_exercise_sets(lang):
    return set(glob.glob(datapath + "/*/" + lang + "-*"))

def get_next_exercise(c, exercise_sets):
    ex_set = random.choice(exercise_sets)
    ex = load_exercise_file (ex_set)
#    outlier = random.choice(ex["outliers"])
    outlier = get_rarest_outlier(c, ex_set)
    ex["words"].append(outlier)
    random.shuffle(ex["words"])
    return ex_set, ex["words"]

def action_next(c, ex_id):
    ex = c.execute("SELECT * FROM exercises WHERE id=?", (ex_id,)).fetchone()
    results = json.loads(ex["results"])
    response = {"completed": False}
    all_ex = get_exercise_sets(ex["lang"])
    todo = list(all_ex - set(results.keys()))
    response["progress"] = (len(all_ex) - len(todo)) / len(all_ex)
    response["todo"] = len(todo)
    if "current" in results:
        response["words"] = results["current"]["words"]
        return response
    if len(todo) == 0:
        response["completed"] = True
        return response
    filename, response["words"] = get_next_exercise(c, todo)
    results["current"] = {"file": filename, "words": response["words"]}
    c.execute("UPDATE exercises SET results=? WHERE id=?", (json.dumps(results), ex_id))
    return response

def save_result(c, ex_id, outlier):
    ex = c.execute("SELECT results FROM exercises WHERE id=?", (ex_id,)).fetchone()
    results = json.loads(ex["results"])
    if not "current" in results:
        return
    if outlier != "=SKIP=" and outlier not in results["current"]["words"]:
        return
    filename = results["current"]["file"]
    from datetime import datetime
    dt = datetime.now().isoformat(" ", "seconds")
    results[filename] = {"words": results["current"]["words"], "outlier": outlier, "when": dt}
    results["previous"] = results.pop("current")
    c.execute("UPDATE exercises SET results=? WHERE id=?", (json.dumps(results), ex_id))

def undo_previous(c, ex_id):
    ex = c.execute("SELECT results FROM exercises WHERE id=?", (ex_id,)).fetchone()
    results = json.loads(ex["results"])
    if not "previous" in results:
        return
    filename = results["previous"]["file"]
    del results[filename]
    results["current"] = results.pop("previous")
    c.execute("UPDATE exercises SET results=? WHERE id=?", (json.dumps(results), ex_id))

def get_rarest_outlier(c, target_set_name):
    stat = {x: 0 for x in load_exercise_file(target_set_name)['outliers']}
    lang = target_set_name.split('/')[-1][:2]
    for row in c.execute('SELECT * FROM exercises WHERE lang=?', (lang,)):
        results = json.loads(row['results'])
        if target_set_name in results:
            for word in results[target_set_name]['words']:
                if word in stat:
                    stat[word] += 1
    return sorted([(y, x) for x, y in stat.items()])[0][1]


if len(sys.argv) > 1: # for testing
    c = conn.cursor()
    if sys.argv[1] == 'next':
        ex_id = sys.argv[2]
        response = action_next(c, ex_id)
    exit(0)


action = form["action"].value

if action == "new":
    name = form["name"].value
    lang = form["lang"].value
    c = conn.cursor()
    while True:
        try:
            c.execute("INSERT INTO exercises VALUES(LOWER(HEX(RANDOMBLOB(16))), ?, ?, '{}', DATETIME())", (name, lang))
        except sqlite3.IntegrityError:
            continue
        break
    new = c.execute("SELECT * FROM exercises WHERE ROWID=?", (c.lastrowid,))
    conn.commit()
    print('{"id": "%s"}' % new.fetchone()["id"])
elif action == "next":
    c = conn.cursor()
    ex_id = form["id"].value.lower()
    if "chosen" in form:
        save_result(c, ex_id, form["chosen"].value)
    response = action_next(c, ex_id)
    conn.commit()
    print(json.dumps(response))
elif action == "undo":
    c = conn.cursor()
    ex_id = form["id"].value.lower()
    undo_previous(c, ex_id)
    response = action_next(c, ex_id)
    conn.commit()
    print(json.dumps(response))
elif action == "eval":
    c = conn.cursor()
    ex_id = form["id"].value.lower()
    r = eval_exercise (c, load_exercise_data(datapath), "ALL", True, None, None, ex_id)[0]
    print(json.dumps(r))