initial working state

utils.py

@@ -0,0 +1,191 @@
+from datetime import datetime, timezone
+from flask import current_app as app
+from flask import g
+import os
+from werkzeug.utils import secure_filename
+
+def create_result_table(key):
+    cur = g.db.cursor()
+    cur.execute("""
+    CREATE TABLE IF NOT EXISTS `%s` (
+        question TEXT,
+        answer TEXT,
+        answer_type TEXT
+    );
+    """%(key, )
+    )
+    g.db.commit()
+
+def create_voter_table(db, name):
+    table_name = get_voter_table_name(name)
+    cur = db.cursor()
+
+    cur.execute("""
+    CREATE TABLE IF NOT EXISTS `%s` (
+        token TEXT PRIMARY KEY,
+        answered BOOLEAN
+    );
+    """%(table_name, )
+    )
+    db.commit()
+
+def get_voter_table_name(key):
+    return key + "__voters"
+
+
+def get_result_table_name(key):
+    return key
+
+
+def has_voted(key, token):
+    cur = g.db.cursor()
+    cur.execute(
+        "SELECT token FROM %s WHERE token = ? AND answered = 'true'"%(
+            get_voter_table_name(key),
+        ),
+        (
+            token,
+        )
+    )
+    return len(cur.fetchall()) > 0
+
+
+def is_draft(form):
+    return form['draft']
+
+def is_expired(form):
+    if form['expires'] == None:
+        return False
+
+    return datetime.now(timezone.utc) > form['expires']
+
+def is_key(key, cli_opts = False):
+    key = secure_filename(key)
+
+    if cli_opts:
+        root_path = cli_opts.questions
+    else:
+        root_path = app.config['QUESTIONS']
+
+    return os.path.exists(
+        os.path.join(
+            root_path,
+            key + ".txt"
+        )
+    )
+
+def parse_form(key):
+    form = {
+        'expires': None,
+        'draft': False,
+        'questions': []
+    }
+    key = secure_filename(key)
+    try:
+        current_question = None
+        with open(os.path.join(app.config['QUESTIONS'], key + ".txt"), "rt") as fp:
+            for row in fp:
+                if row.strip() == "":
+                    continue
+                if row.startswith("#"):
+                    continue
+                if row.lower().startswith("expires: "):
+                    form['expires'] = parse_row_date(row)
+                    continue
+                if row.lower().startswith("draft: "):
+                    if row.lower().rstrip() == "draft: true":
+                        form['draft'] = True
+                    continue
+                if row.startswith("- "):
+                    if current_question == None:
+                        continue
+                    form['questions'][current_question]['choices'].append(
+                        row[2:].strip()
+                    )
+                    continue
+                if row.startswith("+ "):
+                    if current_question == None:
+                        continue
+                    form['questions'][current_question]['multichoices'].append(
+                        row[2:].strip()
+                    )
+                    continue
+                current_question = len(form['questions'])
+                form['questions'].append({
+                    'choices': [],
+                    'multichoices': [],
+                    'index': len(form['questions']) + 1,
+                    'name': row.strip().rstrip("_:").rstrip(),
+                    'open_question': row.strip().endswith("___")
+                })
+        return form
+    except Exception as err:
+        if app.config['DEBUG']:
+            raise err
+        return False
+
+
+def parse_row_date(row):
+    row = row[9:].strip()
+    if row.lower() == "none":
+        return None
+    try:
+        return datetime.strptime(
+            row,
+            '%Y-%m-%d %H:%M %z'
+        )
+    except Exception as err:
+        if app.config['DEBUG']:
+            print(row)
+            raise err
+        return None
+
+def write_vote(key, token, answers, form):
+
+    cur = g.db.cursor()
+    for question in form['questions']:
+        answer = None
+        answer_type = None
+        if 'QC%d'%( question['index'], ) in answers:
+            answer = (answers['QC%d'%( question['index'], )],)
+            answer_type = "single"
+        if 'QM%d'%( question['index'], ) in answers:
+            answer = answers.getlist('QM%d'%( question['index'], ))
+            answer_type = "multiple"
+        if 'QO%d'%( question['index'], ) in answers:
+            answer = (answers['QO%d'%( question['index'], )],)
+            answer_type = "open"
+        if answer == None:
+            continue
+        for single in answer:
+            cur.execute(
+                "INSERT INTO %s VALUES (?, ?, ?)"%(
+                    key,
+                ),
+                (
+                    question['name'],
+                    single.strip(),
+                    answer_type
+                )
+            )
+
+    cur.execute(
+        "UPDATE %s SET answered = 'true' WHERE token = ?"%(
+            get_voter_table_name(key),
+        ),
+        (
+            token,
+        )
+    )
+    g.db.commit()
+
+
+def time_to_expiry(form):
+    if form['expires'] == None:
+        return "Never"
+
+    time_to_go = form['expires'] - datetime.now(timezone.utc)
+    time_to_go = ".".join(str(time_to_go).split('.')[0:-1])[0:-3]
+
+    #time_to_go.microseconds = 0
+    return "%s (%s to go)"%( form['expires'], time_to_go )