49.tornado表单验证

最近在学习Python web框架tornado的使用，话说tornado的RequestHandler类的get_argument方法可以获得get/post传过来的 值，要做表单验证的话，python也有其它的专门的类库。不过我的要求比较简单，就没必要动用那些库了，于是自己简单弄了个功能简单的表单验证。

整个思路就是让RequestHandler在调用self.render_string的时候自动注册一个显示表单验证提示的模板函数.上代码吧：

from tornado.web import RequestHandler



class ObjectDict(object):



    def __getattr__(self, key):

        if key in self:

            return self[key]

        return None



    def __setattr__(self, key ,value):

        self[key] = value



class BaseHandler(RequestHandler):



    def prepare(self):

        self._prepare_context()

        self._prepare_filters()



    def _prepare_context(self):

        self._context = ObjectDict()

        self._context.form = []



    def _prepare_filters(self):

        self._filters = ObjectDict()

        self._filters.form_error = self._form_error



    def _form_error(self, field, msg, error_class='form-error'):

        if field and hasattr(self._context.form, field):

            ret = self.locale.translate(getattr(self._context.form, field))

            if ret:

                return '<span class="' + error_class + '">' + ret + '</span>'

        return ''



    def form_error(self, key, msg):

        setattr(self._context.form, field, msg)



    def render_string(self, template_name, **kwargs):

        kwargs.update(self._filters) # 注册模板函数

        assert "context" not in kwargs, "context is a reserved keyword."

        kwargs["context"] = self._context

        return super(BaseHandler, self).render_string(template_name,**kwargs)

之后在定义的其它handlers的时候只要继承 BaseHandler ，当表单的某个选项验证出错的时候即可调用 self.form_error(field,msg) 来注册验证错误信息，比如 username 选项不能为空，则可以调用 self.form_error('username','Username can not be empty') ，然后在模板中用 {{ form_error('username') }} 来显示信息，并且不影响tornado的多语言国际化。