sfa/storage/migrations/versions/001_slice_researchers.py

   1 # this move is about adding a slice x users many to many relation ship for modelling
   2 # regular "membership" of users in a slice
   3
   4 from sqlalchemy import Table, MetaData, Column, ForeignKey
   5 from sqlalchemy import Integer, String
   6
   7 metadata = MetaData()
   8
   9 # this is needed by migrate so it can locate 'records.record_id'
  10 records = \
  11     Table('records', metadata,
  12           Column('record_id', Integer, primary_key=True),
  13           )
  14
  15 # slice x user (researchers) association
  16 slice_researcher_table = \
  17     Table('slice_researcher', metadata,
  18           Column('slice_id', Integer, ForeignKey(
  19               'records.record_id'), primary_key=True),
  20           Column('researcher_id', Integer, ForeignKey(
  21               'records.record_id'), primary_key=True),
  22           )
  23
  24
  25 def upgrade(migrate_engine):
  26     metadata.bind = migrate_engine
  27     slice_researcher_table.create()
  28
  29
  30 def downgrade(migrate_engine):
  31     metadata.bind = migrate_engine
  32     slice_researcher_table.drop()