copyvios/copyvios/settings.py

# -*- coding: utf-8  -*-

from flask import g
from markupsafe import escape

from .cookies import set_cookie, delete_cookie
from .misc import Query

def process_settings():
    query = Query(method="POST")
    if query.action == "set":
        status = _do_set(query)
    elif query.action == "delete":
        status = _do_delete(query)
    else:
        status = None
    return status

def _do_set(query):
    cookies = g.cookies
    changes = set()
    if query.lang:
        key = "CopyviosDefaultLang"
        if key not in cookies or cookies[key].value != query.lang:
            set_cookie(key, query.lang, 1095)
            changes.add("site")
    if query.project:
        key = "CopyviosDefaultProject"
        if key not in cookies or cookies[key].value != query.project:
            set_cookie(key, query.project, 1095)
            changes.add("site")
    if query.background:
        key = "CopyviosBackground"
        if key not in cookies or cookies[key].value != query.background:
            set_cookie(key, query.background, 1095)
            delete_cookie("EarwigBackgroundCache")
            changes.add("background")
    if changes:
        changes = ", ".join(sorted(list(changes)))
        return "Updated {0}.".format(changes)
    return None

def _do_delete(query):
    cookies = g.cookies
    if query.cookie in cookies:
        delete_cookie(query.cookie.encode("utf8"))
        template = u'Deleted cookie <b><span class="mono">{0}</span></b>.'
        return template.format(escape(query.cookie))
    elif query.all:
        number = len(cookies)
        for cookie in cookies.values():
            delete_cookie(cookie.key)
        return "Deleted <b>{0}</b> cookies.".format(number)
    return None