setup_db.py 699 Bytes
# -*- coding: utf-8 -*-


from server.database import Base, engine
from server.database import db_adapter
from server.database.models import User, UserToken


def setup_db():
    """Initialize db tables

    make sure database and user correctly created in mysql
    in case upgrade the table structure, the origin table need be dropped firstly
    """
    Base.metadata.create_all(bind=engine)

    # init REQUIRED db data.
    db_adapter.add_object_kwargs(User,
                                 id=1,
                                 name='super_admin',
                                 password='123456',
                                 role='admin'
                                 )
setup_db()