app.py

# encoding: utf-8

import os
import jinja2
import webapp2
from google.appengine.api import memcache
import wsd
from glob import glob

jinja_environment = jinja2.Environment(
    loader=jinja2.FileSystemLoader(
        os.path.join(os.path.dirname(os.path.abspath(__file__)), 'html')
    )
)


def get_rulesets():
    rulesets = memcache.get('rulesets')
    if rulesets is not None:
        return rulesets
    else:
        rulesets = []
        for rule_name in (fn[:-9] for fn in glob('*_rules.py')):
            try:
                mod = __import__('{0}_rules'.format(rule_name))
                rulesets.extend(mod.get_rulesets())
                wsd.LOG.info('Loaded: {0}'.format(rule_name))
            except Exception, e:
                wsd.LOG.warn('Not loaded: {0}: {1}'.format(rule_name, e))
        if not memcache.add('rulesets', rulesets, 3600):
            wsd.LOG.error('Memcache set failed.')
        return rulesets


class MainPage(webapp2.RequestHandler):
    def get(self):
        rulesets = get_rulesets()
        disabled_rulesets = self.request.cookies.get('d')
        if disabled_rulesets:
            disabled_rulesets = disabled_rulesets.split('|')
        else:
            disabled_rulesets = []
        template_values = {
            'rulesets': rulesets,
            'disabled': disabled_rulesets
        }
        template = jinja_environment.get_template('index.html')
        self.response.out.write(template.render(template_values))


class Process(webapp2.RequestHandler):
    def post(self):
        text = self.request.get('text')
        checked_rulesets = self.request.POST.getall('ruleset')
        rulesets = []
        disabled = []
        for ruleset in get_rulesets():
            if ruleset.uid in checked_rulesets:
                rulesets.append(ruleset)
            else:
                disabled.append(ruleset.uid)
        prulesets = wsd.ProcessedRulesets(rulesets, text)
        self.response.set_cookie('d', '|'.join(disabled), max_age=31556926)
        self.response.out.write(prulesets.to_html(True))


handler = webapp2.WSGIApplication(
    [
        ('/', MainPage),
        ('/process', Process)
    ],
    debug=True
)