0

我正在尝试创建 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)),
)

如果我打算只做一次（在数据库创建时），然后再向其中添加任何行，那么填充它的最佳方法是什么？

4

1 回答 1

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

填充数据库。

于 2020-11-23T18:16:49.747 回答