models.py 文件源码

def setup_portfolio_database():
    """
    Create database tables if required
    """
    try:
        TableStockExchange.create_table()
    except peewee.OperationalError:
        print("Stock Exchange table already exists!")

    try:
        TableUserPortfolio.create_table()
    except peewee.OperationalError:
        print("User Portfolio table already exists!")

    try:
        TableLogger.create_table()
    except peewee.OperationalError:
        print("Logger table already exists!")