diff --git a/app/db/__init__.py b/app/db/__init__.py index a2da361..4412bf0 100644 --- a/app/db/__init__.py +++ b/app/db/__init__.py @@ -9,8 +9,10 @@ import os import sys import requests +from ..decorators import time_it +@time_it def download(): res = requests.get('https://markusjoe.github.io/static_file_hosting/data.sqlite') with open('./app/db/data.sqlite', 'wb') as fp: @@ -22,7 +24,6 @@ if __name__ != '__main__': print('数据库文件未找到, 正在下载中') try: download() - print('下载完成') except Exception as error: print('下载失败, 错误原因:', error) print('请手动下载数据库文件到 ./app/db/data.sqlite') diff --git a/app/decorators.py b/app/decorators.py new file mode 100644 index 0000000..d1d4dfc --- /dev/null +++ b/app/decorators.py @@ -0,0 +1,25 @@ +#!/usr/bin/env python3 +# -- coding:utf-8 -- +# @Author: markushammered@gmail.com +# @Development Tool: PyCharm +# @Create Time: 2022/4/5 +# @File Name: decorators.py + +import time +from functools import wraps + + +def time_it(func): + @wraps(func) + def wrapper(*args, **kwargs): + print('数据库文件开始下载') + start = time.time() + result = func(*args, **kwargs) + end = time.time() + print('下载结束, 耗时 {} 秒'.format(round(end - start, 2))) + return result + + return wrapper + + +