db_sync.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from models.business import Base as BusinessBase
from models.base import Base as ABase
from models.cloud import Base as CloudBase
from models.asset import Base as ServerBase
from models.event import Base as EventBase
from models.tag import Base as TagBase
from models.tree import Base as TreeBase
from models.cloud_region import Base as CloudRegionBase
from models.domain import Base as DomainBase
from models.order_model import Base as OrderBase
from models.autdit import Base as AuditBase
from websdk2.consts import const
from settings import settings as app_settings

# ORM创建表结构
from sqlalchemy import create_engine

default_configs = app_settings[const.DB_CONFIG_ITEM][const.DEFAULT_DB_KEY]
engine = create_engine('mysql+pymysql://%s:%s@%s:%s/%s?charset=utf8mb4' % (
    default_configs.get(const.DBUSER_KEY),
    default_configs.get(const.DBPWD_KEY),
    default_configs.get(const.DBHOST_KEY),
    default_configs.get(const.DBPORT_KEY),
    default_configs.get(const.DBNAME_KEY),
), encoding='utf-8', echo=True)


def create():
    ABase.metadata.create_all(engine)
    CloudBase.metadata.create_all(engine)
    ServerBase.metadata.create_all(engine)
    EventBase.metadata.create_all(engine)
    BusinessBase.metadata.create_all(engine)
    TagBase.metadata.create_all(engine)
    TreeBase.metadata.create_all(engine)
    CloudRegionBase.metadata.create_all(engine)
    DomainBase.metadata.create_all(engine)
    OrderBase.metadata.create_all(engine)
    AuditBase.metadata.create_all(engine)
    print('[Success] 表结构创建成功!')


def drop():
    ABase.metadata.drop_all(engine)
    CloudBase.metadata.drop_all(engine)
    ServerBase.metadata.drop_all(engine)
    EventBase.metadata.drop_all(engine)
    BusinessBase.metadata.drop_all(engine)
    TagBase.metadata.drop_all(engine)
    TreeBase.metadata.drop_all(engine)
    CloudRegionBase.metadata.drop_all(engine)
    DomainBase.metadata.drop_all(engine)


if __name__ == '__main__':
    create()