You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
31 lines
616 B
31 lines
616 B
# -*- 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("[数据库]初始化数据库连接成功")
|
|
|