RequestCounter/app/db/__init__.py

#!/usr/bin/env python3
# -- coding:utf-8 --
# @Author: markushammered@gmail.com
# @Development Tool: PyCharm
# @Create Time: 2022/3/25
# @File Name: __init__.py.py


import os
import time
import sys
import requests
from functools import wraps


def time_it(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        print('I: 数据库文件开始下载')
        start = time.time()
        result = func(*args, **kwargs)
        end = time.time()
        print('I: 下载结束, 耗时 {} 秒'.format(round(end - start, 2)))
        return result

    return wrapper


@time_it
def download():
    res = requests.get('https://filebase.vercel.app/download/data.db')
    with open('./app/db/data.sqlite', 'wb') as fp:
        fp.write(res.content)


if not os.path.exists('./app/db/data.sqlite'):
    print('C: 数据库文件未找到, 正在下载中')
    try:
        download()
    except Exception as error:
        print('C: 下载失败, 错误原因:', error)
        print('I: 请手动下载数据库文件到 ./app/db/data.sqlite')
        sys.exit(1)

__all__ = [
    'db'
]