首页 > 编程 > Python > 正文

使用python装饰器验证配置文件示例

2020-02-23 05:11:20
字体:
来源:转载
供稿:网友

根据不同配置文件调用不同的验证函数检查输入。可以根据需求更改验证函数的逻辑。

代码如下:
def VerifyData(func):

    def VerifyInt(data):
        assert(int(data) > 0)

    def VerifyString(data):
        assert(len(data) > 10)

    def inner(*args, **kvargs):

        print args
        print kvargs

        assert(len(args) > 1)
        if args[1] == "int.txt":
            VerifyInt(args[0])
        elif args[1] == "string.txt":
            VerifyString(args[0])

        func(*args, **kvargs)

    return inner

@VerifyData
def WriteData(text, filepath):
    print "WriteData"
    print text
    print filepath

WriteData("1234567890", "int.txt")

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表