tornado表单验证

工作中看到有一些不好的代码，想了个办法优化了一下，于是有了这篇博文。本文主要讲tornado的表单验证，使用的是wtforms。本文最终达到的效果是使用self.xx获取post传递过来的值。

首先是将wtforms嵌入tornado。做法可以参考https://github.com/truemped/tornadotools/blob/master/tornadotools/forms.py，我的做法类似，可以看最下面的代码。这样继承一下就能够使用wtforms的表单验证了。能够使用的结果是很多人写这样的代码

self.xx1 = self.form.xx1.data
self.xx2 = self.form.xx2.data
self.xx3 = self.form.xx3.data
:
:

然后这样写了20行(不黑，项目里真有这样的)。写的人也确实是很认真啊！，改进的方法很简单。调用self.xx1的时候自动取值self.form.xx1.data。能做成这样的方法挺多。很多时候会导致self.xxx使用IDE无法补全。我个人认为好的代码肯定能够方便的补全和提示，毕竟这样能实打实的提升码代码的心情，试想一下你要输入20个单词，很可能一不小心就输入错误了。改进的方法就是使用__getattribute__,配合类继承，兼顾自动补全也不用跑去写self.form.xxx.data了

1. __getattr__，当属性不存在的时候会调用
2. __getattribute__,无论属性是否存在都会被调用

import tornado.ioloop
import tornado.web

from wtforms import validators, fields
from wtforms import Form
from wtforms.fields.core import UnboundField


class MultiDict(dict):
    def getlist(self, key):
        return self[key]

    def setlist(self, key, value):
        self[key] = value


class BaseForm(Form):
    def __init__(self, handler=None, obj=None, prefix='', **kwargs):
        if handler is None:
            return
        formdata = MultiDict()
        if handler.request.method == 'POST':
            for name in handler.request.arguments.keys():
                formdata.setlist(name, handler.get_arguments(name))
        else:
            for name in handler.request.query_arguments.keys():
                formdata.setlist(name, handler.request.query_arguments[name])
        Form.__init__(self, formdata, obj=obj, prefix=prefix, **kwargs)


class LoginForm(BaseForm):
    email = fields.StringField(
        validators=[validators.Email(), validators.required(), validators.length(min=5, max=64)])
    password = fields.PasswordField(validators=[validators.required()])


class MainHandler(tornado.web.RequestHandler, LoginForm):
    def __getattribute__(self, item):
        ret = object.__getattribute__(self, item)
        if isinstance(ret, UnboundField):
            form = object.__getattribute__(self, 'form')
            return getattr(form, item).data
        return ret

    def post(self):
        self.form = LoginForm(self)
        print(self.email)
        self.write("Hello, world")


def make_app():
    return tornado.web.Application([
        (r"/", MainHandler),
    ])


if __name__ == "__main__":
    app = make_app()
    app.listen(8888)
    tornado.ioloop.IOLoop.current().start()

完美解决问题,心情瞬间舒畅了┑(￣Д ￣)┍