# peewee db close 装饰器
from model import ZaneTest


def peewee_db_close(func):
    async def wrapper():
        # print(args)
        ZaneTest._schema.database.close()
        return await func()
    return wrapper