使用SQLAlchemy在使用db.create_all()创建数据库后如何填充行

我在SQLAlchemy app.py中有一个create_all调用

@app.before_first_request
def create_tables():
    db.create_all()

并在user.py模型文件中定义一个基本用户：

class UserModel(db.Model):
    __tablename__ = 'users'

    id = db.Column(db.Integer, primary_key = True)
    username = db.Column(db.String(80))
    password = db.Column(db.String(80))

    def __init__(self, username, password):
        self.username = username
        self.password = password

我想预填充一些默认数据,在此示例中,一些基本用户：

    admin = self('admin', 'test')
    guest = self('guest', 'test')

我可以通过某种方式在app.py中输入它们吗？还是在另一个create_data.py类型的文件中？

解决方法:

您可以创建一个example_data.py文件,如下所示：

from user import UserModel

def db_load_example_data(app, db):
    admin = UserModel('admin', 'test')
    guest = UserModel('guest', 'test')
    with app.app_context():
        db.session.add(admin)
        db.session.add(guest)
        db.commit()

然后在您的启动脚本中可以调用此

from example_data import db_load_example_data

db_load_example_data(app, db)

标签：sqlalchemy,flask,flask-sqlalchemy,python
来源： https://codeday.me/bug/20191026/1936431.html