main.py

# -*- coding: utf-8 -*-
import os

from google.appengine.api import memcache

import editors
import jinja2
from bottle import TEMPLATE_PATH as T
from bottle import Bottle, debug, request, response, static_file
from jinja2 import Template

PROJECT_PATH = os.path.join(os.path.abspath(os.path.dirname(__file__)))
TEMPLATE_PATH = os.path.join(PROJECT_PATH, 'templates')
JINJA_ENVIRONMENT = jinja2.Environment(
    loader=jinja2.FileSystemLoader(TEMPLATE_PATH),
    extensions=['jinja2.ext.autoescape'],
    autoescape=True)
STATIC_PATH = os.path.join(PROJECT_PATH, 'assets')

T.insert(0, TEMPLATE_PATH)

app = Bottle()

if os.environ.get('SERVER_SOFTWARE', '').startswith('Development'):
    debug(True)


@app.route('/')
def index():

    def file_exist(file):
        return os.path.isfile("{}/images/logo/{}.png".format(
            STATIC_PATH, file))

    langs = memcache.get('langs')
    if not langs:
        langs = os.listdir("./vim_template/langs")
        memcache.add('langs', langs, 3600)

    template = JINJA_ENVIRONMENT.get_template('index.html')
    return template.render({'langs': langs, 'file_exist': file_exist})


@app.route('/generate.vim', method='POST')
def generate():
    editor_name = request.POST.get("editor", "vim")
    editor = editors.get_editor(editor_name)
    ctx = {"bundle": {}, "vim": {}, "editor": editor}
    select_lang = request.POST.getall('langs')

    for l in select_lang:
        data = memcache.get('vim-{}'.format(l))
        if not data:
            cache = {}
            for ext in ["bundle", "vim"]:
                with open("./vim_template/langs/{0}/{0}.{1}".format(
                        l, ext)) as f:
                    cache[ext] = ctx[ext][l] = f.read()
            memcache.add('vim-{}'.format(l), cache, 3600)
        else:
            ctx["bundle"][l] = data['bundle']
            ctx["vim"][l] = data['vim']

    template = None
    with open("./vim_template/vimrc") as f:
        template = Template(f.read().decode('utf-8'))

    if not template:
        template = Template("")

    response.headers['Content-Type'] = 'application/text'
    response.headers['Content-Disposition'] = 'attachment; \
            filename='.format(os.path.basename(editor.name))
    ctx['select_lang'] = ",".join(select_lang)

    def sh_exist(lang):
        return os.path.isfile("./vim_template/langs/{0}/{0}.sh".format(lang))
    ctx['sh_exist'] = sh_exist

    return template.render(**ctx)


@app.route('/langs')
def langs():

    langs = memcache.get('langs')
    if not langs:
        langs = os.listdir("./vim_template/langs")
        memcache.add('langs', langs, 3600)

    return ",".join(langs)


@app.route('/robots.txt')
def serve_robots():
    return static_file('robots.txt', root=STATIC_PATH)


@app.route('/assets/<path:path>', name='assets')
def static(path):
    yield static_file(path, root=STATIC_PATH)


@app.error(404)
def error_404(error):
    """Return a custom 404 error."""
    return 'Sorry, nothing at this URL.'