如何为烧瓶 sql 炼金术数据库播种

Question

我是python的新手，我刚刚学会了如何使用flask restless和flask sql-alchemy创建一个api。但是，我想用随机值为数据库播种。我如何实现这一点？请帮忙。这是 api 代码...

import flask
import flask.ext.sqlalchemy
import flask.ext.restless
import datetime
DATABASE = 'sqlite:///tmp/test.db'
#Create the Flask application and the FLask-SQLALchemy object
app = flask.Flask(__name__)
app.config ['DEBUG'] = True
app.config['SQLALCHEMY_DATABASE_URI'] =  DATABASE
db = flask.ext.sqlalchemy.SQLAlchemy(app)
#create Flask-SQLAlchemy models
class TodoItem(db.Model):
    id = db.Column(db.Integer, primary_key = True)
    todo = db.Column(db.Unicode)
    priority = db.Column(db.SmallInteger)
    due_date = db.Column(db.Date)
#Create database tables
db.create_all()
#Create Flask restless api manager
manager = flask.ext.restless.APIManager(app, flask_sqlalchemy_db = db)
#Create api end points
manager.create_api(TodoItem, methods = ['GET','POST','DELETE','PUT'], results_per_page = 20)
#Start flask loop
app.run()

Answer 1

我有一个类似的问题，做了一些研究，发现了一些有效的东西。

我看到的模式基于注册 Flask CLI 自定义命令，例如：flask seed .

根据您的示例，这看起来像这样。首先，将以下内容导入您的 api 代码文件(假设您将其命名为 server.py(：

from flask.cli import with_appcontext

(我看到你做import flask但我只想补充你应该把它们改成from flask import what_you_need(

接下来，创建一个为项目执行种子设定的函数：

@with_appcontext
def seed():
   """Seed the database."""
   todo1 = TodoItem(...).save()
   todo2 = TodoItem(...).save()
   todo3 = TodoItem(...).save()

最后，将以下命令注册到烧瓶应用中：

def register_commands(app):
    """Register CLI commands."""
    app.cli.add_command(seed)

配置应用程序后，请确保调用 register_commands 来注册命令：

register_commands(app)

此时，您应该能够运行：flask seed .您可以使用相同的模式添加更多函数(可能是flask reset(。

Answer 2

从另一个新手那里，forgerypy 和 forgerypy3 库可用于此目的(尽管它们看起来有点没有被触及(。

通过将它们添加到模型中来使用它们的简单示例：

class TodoItem(db.Model):
    ....
    @staticmethod
    def generate_fake_data(records=10):
        import forgery_py
        from random import randint
        for record in records:
            todo = TodoItem(todo=forgery_py.lorem_ipsum.word(),
                            due_date=forgery_py.date.date(),
                            priority=randint(1,4))
            db.session.add(todo)
        try:
            db.session.commit()
        except:
            db.session.rollback()

然后，您将在 shell 会话中调用 generate_fake_data 方法。

Miguel Grinberg的Flask Web Development(O'Reilly的书，而不是博客(第11章是一个很好的资源。