from pre_request import pre
from pre_request import Rule

field = {
    "age": Rule(direct_type=int, enum=[1, 2]),
    "name": Rule(gt=6, lt=12),
    "email": Rule(email=True),
    "mobile": Rule(mobile=True),
    "empty": Rule(allow_empty=True, default="sssss_empty"),
    "range": Rule(direct_type=int, gt=10, lt=30),
    "reg": Rule(reg=r'^h\w{3,5}o$', key_map="reg_exp"),
    "trim": Rule(trim=True, json=True),
    "call": Rule(direct_type=int, callback=lambda x: x+100)
}

# 不指定get和post时，不论get请求或者post请求都会使用同一个过滤参数
# 如果指定了get或者post时，直接设置的过滤参数会被覆盖
@app.route("/test", methods=['get', 'post'])
@pre.catch(field)
def test_handler(params=None):
    return str(params)


# 单独设置get请求的过滤参数
@app.route("/get", methods=['get'])
@pre.catch(get=get_field)
def get_handler(params=None):
    return str(params)

# 单独设置post请求的过滤参数
@app.route("/post", methods=['post'])
@pre.catch(post=post_field)
def post_handler(params=None):
    return str(params)

# 同时设置get和post的过滤参数
@app.route("/all", methods=['get', 'post'])
@pre.catch(get=get_field, post=post_field)
def all_handler(params=None):
    return str(params)

# 方法视图
@pre.catch(get=get_field)
def get(self, params=None):
return str(params)

@pre.catch(post=post_field)
def post(self, params=None):
return str(params)

# 字段目标数据类型
self.direct_type = kwargs.get("direct_type", str)

# 当前字段是否允许为空
self.allow_empty = kwargs.get("allow_empty", False)
# 当前字段默认值，如果不允许为空，则次字段无意义
self.default = kwargs.get("default", None)
# 去除前后的空格
self.trim = kwargs.get("trim", False)

# 字段枚举值设置
self.enum = kwargs.get("enum", list())

# 正则表达式
self.reg = kwargs.get("reg", None)
# Email判断
self.email = kwargs.get("email", False)
# 手机号判断
self.mobile = kwargs.get("mobile", False)

# 等于
self.eq = kwargs.get("eq", None)
# 不等于
self.neq = kwargs.get("neq", None)

# 范围限定 direct_type 为数字时限定数字大小，为字符串时限定字符串长度
# 大于
self.gt = kwargs.get("gt", None)
# 大于等于
self.gte = kwargs.get("gte", None)
# 小于
self.lt = kwargs.get("lt", None)
# 小于等于
self.lte = kwargs.get("lte", None)

# key映射
self.key_map = kwargs.get("key_map", None)

# 是否需要进行json解析
self.json_load = kwargs.get("json", False)

# 自定义处理callback, 在所有的filter处理完成后，通过callback回调给用户进行自定义处理
self.callback = kwargs.get("callback", None)