helper.py

#!/usr/bin/env python
# coding=utf-8
#
# Copyright 2012 F2E.im
# Do have a faith in what you're doing.
# Make your life a story worth telling.

import json
import re

from lib.variables import *
from lib.superjson import dumps
from jinja2 import evalcontextfilter, Markup, escape
from markdown import markdown

class Filters():
    def __init__(self, jinja2_env):
        self.jinja2 = jinja2_env

    def register(self):
        self.jinja2.filters["dump_errors"] = self.dump_errors
        self.jinja2.filters["pagination"] = self.pagination
        self.jinja2.filters["nl2br"] = self.nl2br
        self.jinja2.filters["build_uri"] = build_uri
        #self.jinja2.filters["tojson"] = json.JSONEncoder().encode
        self.jinja2.filters["tojson"] = dumps
        self.jinja2.filters["pretty_date"] = self.pretty_date
        self.jinja2.filters["content_process"] = self.content_process
        self.jinja2.filters["markdown"] = self.markdown
        return self.jinja2

    def dump_errors(self, errors):
        t = self.jinja2.from_string("""
            {% if errors %}
            <ul class="errors alert alert-error">
                {% for error in errors %}
                    <li>{{ ','.join(errors[error]) }}</li>
                {% endfor %}
            </ul>
            {% endif %}
            """)

        return t.render(errors = errors)

    def pagination(self, page, uri, list_rows = 10):
        def gen_page_list(current_page = 1, total_page = 1, list_rows = 10):
            if(total_page <= list_rows):
                return range(1, total_page + 1)

            if(current_page + list_rows > total_page):
                return range(total_page - list_rows + 1, list_rows + 1)

            return range(current_page, list_rows + 1)

        t = self.jinja2.from_string("""
            {% if page and not page.pages == 1 %}
                <ul>
                    <li {% if page.current == page.prev %}class="disabled"{% endif %}><a href="{{ uri|build_uri('p', page.prev) }}">«</a></li>
                    {% for p in gen_page_list(page.current, page.pages, list_rows) %}
                        <li {% if page.current == p %}class="active"{% endif %}>
                            {% if not page.current == p %}
                                <a href="{{ uri|build_uri('p', p) }}">{{ p }}</a>
                            {% else %}
                                <a href="javascript:;">{{ p }}</a>
                            {% endif %}
                        </li>
                    {% endfor %}
                    <li {% if page.current == page.next %}class="disabled"{% endif %}><a href="{{ uri|build_uri('p', page.next) }}">»</a></li>
                </ul>
            {% endif %}
            """)

        return t.render(page = page, uri = uri, gen_page_list = gen_page_list, list_rows = list_rows)

    @evalcontextfilter
    def nl2br(self, eval_ctx, value):
        _paragraph_re = re.compile(r'(?:\r\n|\r|\n){2,}')
        result = u'\n\n'.join(u'<p>%s</p>' % p.replace('\n', '<br>\n') for p in _paragraph_re.split(escape(value)))
        if eval_ctx.autoescape:
            result = Markup(result)
        return result

    def pretty_date(self, time = False):
        """
        Get a datetime object or a int() Epoch timestamp and return a
        pretty string like 'an hour ago', 'Yesterday', '3 months ago',
        'just now', etc
        """
        if time == None:
            return time

        from datetime import datetime
        now = datetime.now()
        if type(time) is str or type(time) is unicode:
            time = datetime.strptime(time, '%Y-%m-%d %H:%M:%S')
        elif type(time) is int:
            diff = now - datetime.fromtimestamp(time)
        elif isinstance(time, datetime):
            diff = now - time 
        elif not time:
            diff = now - now
        second_diff = diff.seconds
        day_diff = diff.days

        if day_diff < 0:
            return ''

        if day_diff == 0:
            if second_diff < 10:
                return "刚刚"
            if second_diff < 60:
                return str(second_diff) + " 秒前"
            if second_diff < 120:
                return  "1 分钟前"
            if second_diff < 3600:
                return str(second_diff / 60) + " 分钟前"
            if second_diff < 7200:
                return "1 小时前"
            if second_diff < 86400:
                return str(second_diff / 3600) + " 小时前"
        if day_diff == 1:
            return "昨天"
        if day_diff < 7:
            return str(day_diff) + " 天前"
        if day_diff < 31:
            return str(day_diff / 7) + " 周前"
        if day_diff < 365:
            return str(day_diff / 30) + " 月前"
        return str(day_diff / 365) + " 天前"

    def content_process(self, content):
        # render content included gist
        content = re.sub(r'http(s)?:\/\/gist.github.com\/(\d+)(.js)?', r'<script src="http://gist.github.com/\2.js"></script>', content)
        # render sinaimg pictures
        content = re.sub(r'(http:\/\/\w+.sinaimg.cn\/.*?\.(jpg|gif|png))', r'<img src="\1" />', content)
        # render @ mention links
        content = re.sub(r'@(\w+)(\s|)', r'@<a href="/u/\1">\1</a> ', content)
        # render youku videos
        content = re.sub(r'http://v.youku.com/v_show/id_(\w+).html', r'<embed src="http://player.youku.com/player.php/sid/\1/v.swf" quality="high" width="480" height="400" align="middle" allowScriptAccess="sameDomain" allowFullscreen="true" type="application/x-shockwave-flash"></embed>', content)
        return content

    def markdown(self, content):
        if not content:
            return ""
        return markdown(content, extensions = ['codehilite', 'fenced_code', 'mathjax'], safe_mode = 'escape')