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 custom command，类似于：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章是一个很好的资源。

我如何播种烧瓶sql-alchemy数据库

2 个答案: