如何在Peewee中初始化db?

时间:2017-03-01 21:09:54

标签: python flask peewee flask-peewee

我是Flask堆栈的新手。我需要运行一些项目。我创建了virtualenv并安装了所有要求

Flask==0.10.1
Flask-Login==0.3.1
Flask-Testing==0.4.2
Flask-WTF==0.12
Jinja2==2.8
MarkupSafe==0.23
WTForms==2.0.2
Werkzeug==0.10.4
argparse==1.2.1
flask-peewee==0.6.6
funcsigs==0.4
itsdangerous==0.24
mock==1.3.0
nose==1.3.7
pbr==1.8.0
peewee==2.6.4
psycopg2==2.6.1
requests==2.7.0
six==1.9.0
wsgiref==0.1.2
wtf-peewee==0.2.3

我还有 runserver.py 脚本启动一个项目:

from myproject import create_app
app = create_app()

if __name__ == '__main__':
    app.run(debug=True, port=5000, host='0.0.0.0')

现在我想从 models.py

初始化db
from peewee import Model, CharField, IntegerField, DateTimeField, DecimalField, TextField, datetime as peewee_datetime
from playhouse.pool import PooledPostgresqlExtDatabase

from .config import BaseConfig


db = PooledPostgresqlExtDatabase(**BaseConfig.DATABASE)
db.commit_select = True
db.autorollback = True


class BaseModel(Model):
    class Meta:
        database = db

    def save(self, **kwds):
        with db.transaction():
            Model.save(self, **kwds)


class Payment(BaseModel):
    class Meta:
        db_table = "payments"

    card_number = CharField()
    amount = DecimalField()
    ....


def init_db():
    try:
        db.connect()
        map(lambda l: db.drop_table(l, True), (Payment,))
        print "tables dropped"
        map(lambda l: db.create_table(l, True), (Payment,))
        print "tables created"
    except:
        db.rollback()
        raise

如何创建表格? Peewee是否有像Django命令" python manage.py migrate "? 以及如何在项目中运行python shell,例如" python manage.py shell "在Django?

1 个答案:

答案 0 :(得分:0)

db.create_tables([Table1, Table2, Table3])

文档:http://docs.peewee-orm.com/en/latest/peewee/models.html#creating-model-tables