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 base_test import fixture
   7 from daklib.dbconn import DBConn
   8
   9 from sqlalchemy import create_engine, __version__
  10
  11 import pickle
  12
  13 pickle_filename = fixture('db-metadata-%s.pkl' % __version__)
  14 pickle_file = open(pickle_filename, 'w')
  15 metadata = DBConn().db_meta
  16 pickle.dump(metadata, pickle_file)
  17 pickle.dump(metadata.ddl_listeners, pickle_file)
  18 pickle_file.close()
  19 print "File %s has been updated successfully." % pickle_filename