在 SQLAlchemy 中填充表的最佳方法是什么？

Question

我正在嘗試創建 Web 應用程序用戶可能的角色列表。 如果我以這種方式定義roles表：

roles = db.Table(
    "roles",
    db.Model.metadata,
    db.Column("role_id", db.Integer, db.ForeignKey("task.id"), primary_key=True),
    db.Column("name", db.String(32)),
)

如果我打算只執行一次（在創建數據庫時），然后再向其中添加任何行，那么填充它的最佳方法是什么？

Answer 1

我相信這種范式稱為“數據庫播種”，當您在谷歌上搜索答案時，這可能會對您有所幫助。

我在網上看了一下，發現了這個： https : //pypi.org/project/Flask-Seeder/

    from flask import Flask
    from flask_sqlalchemy import SQLAlchemy
    from flask_seeder import FlaskSeeder

    def create_app():
      app = Flask(__name__)

      db = SQLAlchemy()
      db.init_app(app)

      seeder = FlaskSeeder()
      seeder.init_app(app, db)

      return app

然后你可以用你的種子創建另一個文件。

    from flask_seeder import Seeder, Faker, generator

    # SQLAlchemy database model
    class User(Base):
      def __init__(self, id_num=None, name=None, age=None):
        self.id_num = id_num
        self.name = name
        self.age = age

      def __str__(self):
        return "ID=%d, Name=%s, Age=%d" % (self.id_num, self.name, self.age)

    # All seeders inherit from Seeder
    class DemoSeeder(Seeder):

      # run() will be called by Flask-Seeder
      def run(self):
        # Create a new Faker and tell it how to create User objects
        faker = Faker(
          cls=User,
          init={
            "id_num": generator.Sequence(),
            "name": generator.Name(),
            "age": generator.Integer(start=20, end=100)
          }
        )

        # Create 5 users
        for user in faker.create(5):
          print("Adding user: %s" % user)
          self.db.session.add(user)

最后，你可以打電話

$ flask seed run

填充數據庫。