Adds migration for OAuth1 tables

This commit is contained in:
xray7224 2013-07-18 20:21:35 +01:00
parent 89d5b44e0a
commit 8ddd7769de

View File

@ -26,7 +26,9 @@ from sqlalchemy.sql import and_
from migrate.changeset.constraint import UniqueConstraint
from mediagoblin.db.migration_tools import RegisterMigration, inspect_table
from mediagoblin.db.models import MediaEntry, Collection, User, MediaComment
from mediagoblin.db.models import (MediaEntry, Collection, User, MediaComment,
Client, RequestToken, AccessToken,
NonceTimestamp)
MIGRATIONS = {}
@ -379,3 +381,15 @@ def pw_hash_nullable(db):
constraint.create()
db.commit()
@RegisterMigration(14, MIGRATIONS)
def create_oauth1_tables(db):
""" Creates the OAuth1 tables """
Client.__table__.create(db.bind)
RequestToken.__table__.create(db.bind)
AccessToken.__table__.create(db.bind)
NonceTimestamp.__table__.create(db.bind)
db.commit()