From bd3585ffcd4bf6800b3b02060ee28f47341e8d12 Mon Sep 17 00:00:00 2001 From: termie Date: Tue, 26 Nov 2013 17:11:50 -0800 Subject: [PATCH] add admin for apikey and migration --- alembic/env.py | 4 +-- alembic/versions/449461dbc725_add_apikey.py | 31 +++++++++++++++++++++ refstack/admin.py | 1 + 3 files changed, 34 insertions(+), 2 deletions(-) create mode 100644 alembic/versions/449461dbc725_add_apikey.py diff --git a/alembic/env.py b/alembic/env.py index 6449b1da..b28f97ae 100755 --- a/alembic/env.py +++ b/alembic/env.py @@ -38,8 +38,8 @@ fileConfig(config.config_file_name) # from myapp import mymodel # target_metadata = mymodel.Base.metadata # target_metadata = None -from refstack.models import * -target_metadata = Base.metadata +from refstack.models import db +target_metadata = db.metadata # other values from the config, defined by the needs of env.py, # can be acquired: diff --git a/alembic/versions/449461dbc725_add_apikey.py b/alembic/versions/449461dbc725_add_apikey.py new file mode 100644 index 00000000..9becd096 --- /dev/null +++ b/alembic/versions/449461dbc725_add_apikey.py @@ -0,0 +1,31 @@ +"""empty message + +Revision ID: 449461dbc725 +Revises: 59e15d864941 +Create Date: 2013-11-26 16:57:16.062788 + +""" + +# revision identifiers, used by Alembic. +revision = '449461dbc725' +down_revision = '59e15d864941' + +from alembic import op +import sqlalchemy as sa + + +def upgrade(): + op.create_table('apikey', + sa.Column('id', sa.Integer(), nullable=False), + sa.Column('name', sa.String(length=60), nullable=True), + sa.Column('key', sa.String(length=200), nullable=True), + sa.Column('openid', sa.String(length=200), nullable=True), + sa.Column('user_id', sa.Integer(), nullable=False), + sa.Column('timestamp', sa.DateTime(), nullable=True), + sa.ForeignKeyConstraint(['user_id'], ['user.id'], ), + sa.PrimaryKeyConstraint('id'), + ) + + +def downgrade(): + op.drop_Table('apikey') diff --git a/refstack/admin.py b/refstack/admin.py index d756a65e..39074dbf 100644 --- a/refstack/admin.py +++ b/refstack/admin.py @@ -22,6 +22,7 @@ def init_app(app): def configure_admin(app): + admin.add_view(SecureView(models.ApiKey, db.session)) admin.add_view(SecureView(models.Cloud, db.session)) admin.add_view(SecureView(models.User, db.session)) admin.add_view(SecureView(models.Vendor, db.session))