# -*- coding: utf-8 -*-

from tortoise import Tortoise, run_async
from config import *
from utils.utils import logger

"""
数据库操作
"""


def get_db_url():
    """
    拼接数据库url
    :return:
    """
    return F"mysql://{DB_USER}:{DB_PASSWORD}@{DB_HOST}:{DB_PORT}/{DB_DATABASE}"


async def init():
    """
    初始化数据库连接
    :return:
    """
    await Tortoise.init(
        db_url=get_db_url(),
        modules={"models": ['models.monitor_result_db', 'models.monitor_task_db']}
    )
    await Tortoise.generate_schemas()
    logger.info("[数据库]初始化数据库连接成功")