[关闭]
@bergus 2015-12-02T14:33:04.000000Z 字数 5118 阅读 1699

python表单校验摘录

python flask wtf

  1. # encoding=utf-8
  2. from app.models import Student
  3. from flask import g
  5. import re
  6. from flask.ext.wtf import Form
  7. from wtforms import StringField, RadioField, PasswordField, TextAreaField, BooleanField, DateField, ValidationError, \
  8.     IntegerField
  9. from wtforms.validators import DataRequired, Length, Regexp, Email, EqualTo
  10. from wtforms.widgets import ListWidget, HTMLString
  13. class BSListWidget(ListWidget):
  14.     def __call__(self, field, **kwargs):
  15.         kwargs.setdefault('id', field.id)
  16.         html = ""
  17.         for subfield in field:
  18.             html += u'<label class="radio-inline"> %s%s </label>' % (subfield(), subfield.label.text)
  19.         return HTMLString(html)
  22. class Fields(object):
  23.     notnull = u'该项输入不能为空'
  25.     def get_len_str(min=None, max=None):
  26.         if min and not max:
  27.             return u"该项输入的最小长度必须是%d" % min
  28.         elif max and not min:
  29.             return u"该项输入的最大长度必须是%d" % max
  30.         else:
  31.             return u'该输入的长度必须大于%d,小于%d' % (min, max)
  33.     username = StringField(label=u'请输入您的用户名',
  34.                            validators=[DataRequired(message=notnull),
  35.                                        Length(min=0, max=15, message=get_len_str(0, 16)),
  36.                                        ])
  38.     password = PasswordField(label=u'请输入密码', description=u'请输入密码',
  39.                              validators=[DataRequired(message=notnull),
  40.                                          Length(min=0, max=60, message=get_len_str(min=0, max=61)),
  41.                                          ])
  42.     confirm_password = PasswordField(label=u'请确认密码',
  43.                                      description=u'请确认密码',
  45.                                      validators=[DataRequired(message=notnull),
  46.                                                  Length(min=5, max=60, message=get_len_str(min=4, max=61)),
  47.                                                  EqualTo(u'confirm_password', message=u'两次输入的密码不一致'), ]
  48.                                      )
  49.     student_amount = StringField(label=u'请输入您指导的学生数量',
  50.                                  validators=[Regexp(re.compile(r"\d"))])
  52.     is_active = RadioField(label=u'是否激活账户',
  53.                            coerce=int,
  54.                            choices=[(0, u'否'), (1, u'是')],
  55.                            default=0,
  56.                            widget=BSListWidget())
  58.     notice = TextAreaField(label=u'请填写对学生的通知')
  60.     attachment = StringField(label=u'添加附加',
  61.                              validators=[Length(min=0, max=32, message=get_len_str(min=0, max=33))], )
  63.     is_comment_teacher = RadioField(label=u'是否有评价功能',
  64.                                     coerce=int,
  65.                                     choices=[(0, u'否'), (1, u'是')],
  66.                                     default=0,
  67.                                     widget=BSListWidget())
  69.     student_name = StringField(label=u'请输入您的姓名',
  70.                                description='',
  71.                                validators=[DataRequired(message=notnull),
  72.                                            Length(min=0, max=15, message=get_len_str(0, 16)),
  73.                                            ])
  75.     sex = RadioField(label=u'您的性别',
  76.                      coerce=int,
  77.                      choices=[(0, u'男'), (1, u'女')],
  78.                      default=0,
  79.                      widget=BSListWidget())
  81.     user_type = RadioField(label=u'您是',
  82.                            coerce=str,
  83.                            choices=[(u'student', u'学生'), (u'teacher', u'老师'), (u'admin', u'管理员')],
  84.                            default=u'student',
  85.                            widget=BSListWidget())
  87.     mark = StringField(label=u'您的分数',
  88.                        default=0,
  89.                        validators=[DataRequired(message=notnull),
  90.                                    Length(min=0, max=100, message=get_len_str(0, 101)),
  91.                                    ])
  92.     comment = TextAreaField(label=u'请填写您对学生的评语',
  93.                             validators=[
  94.                                 Length(min=0, max=128, message=get_len_str(0, 129)),
  95.                             ])
  97.     title = StringField(label=u'请填写毕业设计的题目',
  98.                         validators=[
  99.                             Length(min=0, max=128, message=get_len_str(0, 129)),
  100.                         ])
  101.     description = TextAreaField(label=u'请填写毕业设计的描述')
  103.     task_start_date = DateField(label=u'开始时间')
  104.     task_end_date = DateField(label=u'结束时间')
  106.     comment_start_date = DateField(label=u'开始时间')
  107.     comment_end_date = DateField(label=u'结束时间')
  110. class LoginForm(Form):
  111.     username = Fields.username
  112.     password = Fields.password
  114.     user_type = Fields.user_type
  116.     remember_me = BooleanField(label=u'记住我', default=u'y')
  118.     errors = {}
  121.     def validate_fields(self):
  122.         status = True
  123.         status = status and self.username.validate(self)
  124.         status = status and self.username.validate(self)
  125.         self.password.validate()
  127.         # def validate(self):
  128.         self._fields
  129.         self.validate_on_submit()
  130.         return self.username.validate(self)
  132.         # def validate_username(self, field):
  133.         # user = Student.get_user(field.data)
  134.         # if not user:
  135.         # print 'not user'
  136.         # self.errors['username'] = u'用户名不存在'
  137.         # raise ValidationError(message=u'该用户名已被注册')
  138.         # else:
  139.         # print 'user'
  140.         # return True
  142.         #
  143.         # def validate_password(self,field):
  144.         # if g.user:
  145.         # if field.data != g.user.username:
  146.         # field.errors.append(message=u'用户密码不正确')
  147.         # return False
  148.         #
  149.         # def validate_user_type(self, field):
  150.         # print field.data
  151.         #
  152.         # def validate_remember_me(self,field):
  153.         # if field.data:
  154.         # print field.data
  157. class StuInfo(Form):
  158.     username = Fields.username
  159.     student_name = Fields.student_name
  160.     sex = Fields.sex
  161.     attachment = Fields.attachment
  162.     mark = Fields.mark
  163.     comment = Fields.comment
  166. class BaseForm(object):
  167.     def __init__(self, form):
  168.         self.username = Field(label=u'用户名', type='text', validators={'min_length': 5})
  170.         for field_name, data in form.items():
  171.             getattr(self, field_name).data = data
  173.     def validate(self):
  174.         status = True
  175.         for field_name in self.__dict__:
  176.             status = status and getattr(self, field_name).validate()
  177.         return status
  180. class Field(object):
  181.     def __init__(self, label=None, type='text', validators={}, description=None, data=''):
  182.         self.data = data
  183.         self.label = label
  184.         self.type = type
  185.         self.validators = validators
  186.         self.description = description
  188.     def validate(self):
  189.         status = True
  190.         for method_name, param in self.validators.items():
  191.             print method_name, param
  192.             status = status and getattr(self, method_name)(param)
  193.         return status
  195.     def is_null(self, status):
  196.         if status:
  197.             return True
  198.         if not self.data:
  199.             return False
  200.         if hasattr(self.data, 'replace') and len(self.data.replace(' ', '')) > 0:
  201.             return False
  202.         if len(self.data) == 0:
  203.             return False
  204.         return True
  206.     def min_length(self, min=-1):
  207.         if not self.is_null(False) and len(self.data) < min:
  208.             return False
  209.         return True
  211.     def max_length(self, max):
  212.         if not self.is_null(False) and len(self.data) > max:
  213.             return False
  214.         return True
  216.     def min(self, min):
  217.         try:
  218.             if int(self.data) < min:
  219.                 return False
  220.             return True
  221.         except Exception, e:
  222.             print e
  223.             return False
  225.     def max(self, max):
  226.         try:
  227.             if int(self.data) > max:
  228.                 return False
  229.             return True
  230.         except Exception, e:
  231.             print e
  232.             return False
  234.     def equal_to(self, data):
  235.         if self.data != data:
  236.             return False
  237.         return True
  239.     def select_from(self, *args):
  240.         if self.data not in args:
  241.             return False
  242.         return True
  245. if __name__ == '__main__':
  246.     b = BaseForm({'username': 'o'})
  247.     print b.validate()
  248.     print 'ok'
添加新批注
在作者公开此批注前，只有你和作者可见。
回复批注