给web.py加点仿TG的功能

最近忙，本来计划的一篇技术文章写了很久没写完，先发一篇短文凑数。

以前用Turbogears 1.x的时候觉得它有一些功能真是很方便，比如expose decorator。不过现在用web.py就没有这种东西了。

话说我改用web.py就是图它比较清爽这个优点，但能增加点方便还是好的。其实这个功能实现起来挺简单的，于是这两天就给web.py写了一个。

另外，现在RESTful的API都喜欢提供多种格式的结果返回（貌似这是twitter引领的风潮），于是顺便也给自己写的这个东东增加了这样的功能。

代码如下（其中obj2xml的代码来自这里 ）：

import json
import web
from web.contrib.template import render_mako
class TGBaseHandler:
    def __init__(self, templates_path):
        self.mako_render = render_mako(
            directories=[templates_path,],
            input_encoding='utf-8',
            output_encoding='utf-8'
        )
    # 默认支持两种 format render
    # html render 使用 mako 模板
    def html_render(self, page, result):
        return getattr(self.mako_render, page)(**result)
    def json_render(self, page, result):
        return json.dumps(result)
    # 自动调用相应的 format render
    def render(self, page, format, result):
        try:
            render_func=getattr(self, format + '_render')
        except AttributeError:
            raise "Invalide render format!"
        return render_func(page, result)
# format : html/json/xml...
# 默认为 html format，只需要提供 page 参数，其它 format 可以不需要 page 参数
def expose(page='', format='html'):
    def expose_wrapper(fn):
        def format_wrapper(self, *args, **kwargs):
            result=fn(self, *args, **kwargs)
            if 'format' in result.keys(): # 自动选择相应的 format
                format = result['format']
                del result['format']
            return self.render(page, format, result)
        return format_wrapper
    return expose_wrapper

用法如下：

#...
import obj2xml
urls = (
        "/", "index",
        "/index", "index",
        "/jsondata", "jsondata",
        "/timeline\.?([^/?\.]+)", "timeline", # 自动取得 format 参数
        )
#...
class BaseHandler(TGBaseHandler):
    def __init__(self):
        TGBaseHandler.__init__(self, ‘pathto/templates’) # 初始化 mako
    # 可以在这里添加各种自定义的 format render
    def xml_render(self, page, result):
        return obj2xml.getXML(result)
class index(BaseHandler):
    @expose("index") # 默认 format 为 html
    def GET(self):
        return dict(url="/")
class jsondata(BaseHandler):
    @expose(format='json') # 指定用 json format render
    def GET(self):
        return dict(tweets=[{'id':'1234','text':'hello'},{'id':'2345','text':'world'}])
class timeline(BaseHandler):
    @expose() # 自动选择 format render
    def GET(self, format):
        return dict(format=format, tweets=[{'id':'1234','text':'hello'},{'id':'2345','text':'world'}]) # 注意，要在这里返回 format 参数

这样访问 index 页面时返回的是用mako render出来的HTML，访问 jsondata 页面返回的就是一个 JSON 文本，访问 timeline.xml 和 timeline.json 就可以分别返回 XML 和 JSON 文本。

另外Turbogears 1.x的identity也很赞，回头有空也弄一个给web.py。

推送到[go4pro.org]