flask 后台表单验证模块

python flask 表单验证

我不想直接用flask的wtf模块，太大，功能太多，用不了。但后台也不能不做验证吧，我比较懒，不想一行一行写代码验证，所以就写了一个验证模块，对于小项目也够用了

# encoding=utf-8
# 这是一个简单的后台验证框架,因为前台的话要用ajax处理,那么后台就不需要复杂的表单处理了
# 基础表单类,具体验证表单继承它
from flask import session
class BaseForm(object):
    def __init__(self,form=None):
        self.form = form
        self.fields = []
        self.errors = []  # 错误信息存放的地方
        self.init()
    def init(self):
        if not self.form:
            return False
        # 给表单赋值
        for field_name in dir(self):
            field = self.__getattribute__(field_name)
            if isinstance(field,Field):
                if field_name in self.form:
                    field.data = self.form[field_name]
                    self.fields.append((field_name,field.validate))
        # 验证自定义的以validate_开头的方法
        for validate_method in dir(self):
            if validate_method.startswith('validate_'):
                self.fields.append((validate_method,self.__getattribute__(validate_method)))
    # 验证入口
    def validate(self):
        status = True
        for field_name,validate_method in self.fields:
            status = validate_method() and status
            if not status:
                print field_name,status
                return False
        return status
# 字段类,主要提供字段的各种验证和约束
class Field(object):
    def __init__(self,label=None,type='text',validators={},description=None,data=''):
        self.data = data
        self.label = label
        self.type = type
        self.validators = validators
        self.description = description
    # 验证入口,验证该字段所有的约束条件
    def validate(self):
        status = True
        for method_name,param in self.validators.items():
            status = self.__getattribute__(method_name)(param) and status
            if not status:
                print method_name,status
                return True
        return status
    # 验证数据是否复合正则表达式
    def __rule(self,re_str):
        import re
        pattern = re.compile(re_str)
        match = pattern.match(self.data)
        if match:
            return True
        return False
    # 对开提供的正则表达式模块
    def regex(self,re_str):
        return self.__rule(re_str)
    # 是否为数字
    def digits(self,status=True):
        return self.__rule(r"\d+")
    # 是否是链接
    def url(self,status=True):
        re_str = r"^(https?|s?ftp):\/\/(((([a-z]|\d|-|\.|_|~|[\u00A0-\uD7FF\uF900-\uFDCF\uFDF0-\uFFEF])|(%[\da-f]{2})|[!\$&'\(\)\*\+,;=]|:)*@)?(((\d|[1-9]\d|1\d\d|2[0-4]\d|25[0-5])\.(\d|[1-9]\d|1\d\d|2[0-4]\d|25[0-5])\.(\d|[1-9]\d|1\d\d|2[0-4]\d|25[0-5])\.(\d|[1-9]\d|1\d\d|2[0-4]\d|25[0-5]))|((([a-z]|\d|[\u00A0-\uD7FF\uF900-\uFDCF\uFDF0-\uFFEF])|(([a-z]|\d|[\u00A0-\uD7FF\uF900-\uFDCF\uFDF0-\uFFEF])([a-z]|\d|-|\.|_|~|[\u00A0-\uD7FF\uF900-\uFDCF\uFDF0-\uFFEF])*([a-z]|\d|[\u00A0-\uD7FF\uF900-\uFDCF\uFDF0-\uFFEF])))\.)+(([a-z]|[\u00A0-\uD7FF\uF900-\uFDCF\uFDF0-\uFFEF])|(([a-z]|[\u00A0-\uD7FF\uF900-\uFDCF\uFDF0-\uFFEF])([a-z]|\d|-|\.|_|~|[\u00A0-\uD7FF\uF900-\uFDCF\uFDF0-\uFFEF])*([a-z]|[\u00A0-\uD7FF\uF900-\uFDCF\uFDF0-\uFFEF])))\.?)(:\d*)?)(\/((([a-z]|\d|-|\.|_|~|[\u00A0-\uD7FF\uF900-\uFDCF\uFDF0-\uFFEF])|(%[\da-f]{2})|[!\$&'\(\)\*\+,;=]|:|@)+(\/(([a-z]|\d|-|\.|_|~|[\u00A0-\uD7FF\uF900-\uFDCF\uFDF0-\uFFEF])|(%[\da-f]{2})|[!\$&'\(\)\*\+,;=]|:|@)*)*)?)?(\?((([a-z]|\d|-|\.|_|~|[\u00A0-\uD7FF\uF900-\uFDCF\uFDF0-\uFFEF])|(%[\da-f]{2})|[!\$&'\(\)\*\+,;=]|:|@)|[\uE000-\uF8FF]|\/|\?)*)?(#((([a-z]|\d|-|\.|_|~|[\u00A0-\uD7FF\uF900-\uFDCF\uFDF0-\uFFEF])|(%[\da-f]{2})|[!\$&'\(\)\*\+,;=]|:|@)|\/|\?)*)?$"
        return self.__rule(re_str)
    # 是否是邮箱
    def email(self,status=True):
        re_str = r"^[a-zA-Z0-9.!#$%&'*+\/=?^_`{|}~-]+@[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?(?:\.[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?)*$"
        return self.__rule(re_str)
    # 是否为空
    def not_null(self,status):
        if not status:
            return True
        if not self.data:
            return False
        if hasattr(self.data,'replace') and len(self.data.replace(' ','')) < 0:
            return False
        if len(self.data) == 0:
            return False
        return True
    # 最小长度
    def min_length(self,min=-1):
        return self.not_null(True) and len(self.data) >= min
    # 长度范围
    def length_range(self,range=[]):
        if self.not_null(True) and len(range) == 2:
            data_length = len(self.data)
            return data_length >= range[0] and data_length <= range[1]
        return False
    # 最大长度
    def max_length(self,max):
        return self.not_null(True) and len(self.data) <= max
    # 最小值
    def min(self,min):
        try:
            return int(self.data) >= min
        except Exception,e:
            print e
            return False
    # 最大值
    def max(self,max):
        try:
            return int(self.data) <= max
        except Exception,e:
            print e
            return False
    # 值范围
    def data_range(self,range=[]):
        try:
            data = int(self.data)
            if len(range) == 2:
                return data >= range[0] and data <= range[1]
            return False
        except Exception,e:
            print e
            return False
    # 是否等于field
    def equal_to(self,field):
        return self.data == field.data
    # 是否为规定的值
    def choices(self,args=[]):
        return self.data in args
    # 检查多选框
    def checked(self,status=True):
        if not status:
            return True
        return self.data in [1,'y','yes','true']
if __name__ == '__main__':
    class Test(BaseForm):
        name = Field()
        password = Field()
        email = Field()