tests/create_pickle_file.py

   1 #!/usr/bin/python
   2
   3 # recreate the pickle file db-metadata-*.pkl that needs to be updated
   4 # after a database upgrade
   5
   6 from daklib.dbconn import DBConn
   7
   8 from sqlalchemy import create_engine, __version__
   9
  10 import pickle
  11 from os.path import abspath, dirname
  12
  13 DAK_TEST_DIR = dirname(abspath(__file__))
  14 pickle_filename = '%s/fixtures/db-metadata-%s.pkl' % (DAK_TEST_DIR, __version__)
  15 pickle_file = open(pickle_filename, 'w')
  16 metadata = DBConn().db_meta
  17 pickle.dump(metadata, pickle_file)
  18 pickle.dump(metadata.ddl_listeners, pickle_file)
  19 pickle_file.close()
  20 print "File %s has been updated successfully." % pickle_filename