30 lines
903 B
Python
30 lines
903 B
Python
from sqlalchemy import MetaData, Table, Column, Integer, Unicode, ForeignKey
|
|
from migrate import migrate_engine
|
|
from migrate.changeset.schema import create_column
|
|
from gnuviechadmin.config import config
|
|
|
|
dbschema = None
|
|
if 'database.schema' in config:
|
|
dbschema = config['database.schema']
|
|
|
|
meta = MetaData(migrate_engine)
|
|
client = Table('client', meta, schema = dbschema, autoload = True)
|
|
organization = Table(
|
|
'organization', meta,
|
|
Column('organizationid', Integer, primary_key = True),
|
|
Column('name', Unicode(200), nullable = False, unique = True),
|
|
schema = dbschema,
|
|
useexisting = True)
|
|
|
|
def upgrade():
|
|
organization.create()
|
|
col = Column('organizationid', Integer,
|
|
ForeignKey(organization.c.organizationid),
|
|
nullable = True)
|
|
create_column(col, client)
|
|
|
|
|
|
def downgrade():
|
|
client.c.organizationid.drop()
|
|
organization.drop()
|