]> git.decadent.org.uk Git - dak.git/commitdiff
Merge branch 'dbtests' of ftp-master.debian.org:public_html/dak into dbtests
authorTorsten Werner <twerner@debian.org>
Tue, 11 Jan 2011 20:25:26 +0000 (21:25 +0100)
committerTorsten Werner <twerner@debian.org>
Tue, 11 Jan 2011 20:25:26 +0000 (21:25 +0100)
tests/create_pickle_file.py [new file with mode: 0755]

diff --git a/tests/create_pickle_file.py b/tests/create_pickle_file.py
new file mode 100755 (executable)
index 0000000..55843a9
--- /dev/null
@@ -0,0 +1,19 @@
+#!/usr/bin/python
+
+# recreate the pickle file db-metadata-*.pkl that needs to be updated
+# after a database upgrade
+
+from base_test import fixture
+from daklib.dbconn import DBConn
+
+from sqlalchemy import create_engine, __version__
+
+import pickle
+
+pickle_filename = fixture('db-metadata-%s.pkl' % __version__)
+pickle_file = open(pickle_filename, 'w')
+metadata = DBConn().db_meta
+pickle.dump(metadata, pickle_file)
+pickle.dump(metadata.ddl_listeners, pickle_file)
+pickle_file.close()
+print "File %s has been updated successfully." % pickle_filename