Home | History | Annotate | Download | only in migrations 
      1 import common
      2 from autotest_lib.database import db_utils
      3 
      4 
      5 ORIG_NAMES = (
      6         'aborted_host_queue_entries',
      7         'acl_groups',
      8         'acl_groups_hosts',
      9         'acl_groups_users',
     10         'atomic_groups',
     11         'autotests',
     12         'autotests_dependency_labels',
     13         'host_attributes',
     14         'host_queue_entries',
     15         'hosts',
     16         'hosts_labels',
     17         'ineligible_host_queues',
     18         'jobs',
     19         'jobs_dependency_labels',
     20         'labels',
     21         'profilers',
     22         'recurring_run',
     23         'special_tasks',
     24         'users',
     25         )
     26 
     27 RENAMES_UP = dict((name, 'afe_' + name) for name in ORIG_NAMES)
     28 
     29 RENAMES_DOWN = dict((value, key) for key, value in RENAMES_UP.iteritems())
     30 
     31 
     32 def migrate_up(manager):
     33     db_utils.rename(manager, RENAMES_UP)
     34 
     35 
     36 def migrate_down(manager):
     37     db_utils.rename(manager, RENAMES_DOWN)
     38