欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2023-10-20 16:14:43
根据不同配置文件调用不同的验证函数检查输入。可以根据需求更改验证函数的逻辑。 复制代码 代码如下:def verifydata(func):   &...

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

复制代码 代码如下:

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")