flask - 如何使用 SQLalchemy 和 Alembic 自动初始化数据库？

Question

目前，我跑

$ flask db init
$ flask db migrate -m "initialization"
$ flask db upgrade

如果数据库不存在。我想在 Python 中运行它，例如就像是

app.create_db()

这样我就不必关心设置数据库了。那可能吗？

我使用 flask-sqlalchemy和 flask-migrations插件

Answer 1

您可以使用 SQLAlchemy-Utils为了这。

from sqlalchemy import create_engine
from sqlalchemy_utils import database_exits,create_database

def validate_database():
     engine = create_engine('postgres://postgres@localhost/name')
     if not database_exists(engine.url): # Checks for the first time  
         create_database(engine.url)     # Create new DB    
         print("New Database Created"+database_exists(engine.url)) # Verifies if database is there or not.
     else:
         print("Database Already Exists")

在您的 __init__.py 文件中调用此方法，以便它在您的服务器每次启动时进行检查。