From 797f4437eb08c463c3d9f49e526d1750ea6a1c8b Mon Sep 17 00:00:00 2001 From: Elrond Date: Mon, 2 Apr 2012 14:02:56 +0200 Subject: [PATCH] Check that the media_data tables are empty. In the analyzing part also check that the media_data tables are empty (as expected) before dropping them. --- mediagoblin/db/sql/convert.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/mediagoblin/db/sql/convert.py b/mediagoblin/db/sql/convert.py index 2a8734f9..ac64cf8d 100644 --- a/mediagoblin/db/sql/convert.py +++ b/mediagoblin/db/sql/convert.py @@ -220,14 +220,21 @@ def cleanup_sql_tables(sql_db): count = session.query(MediaEntry.media_type). \ filter_by(media_type=unicode(mt)).count() print " %s: %d entries" % (mt, count) + if count == 0: + print "\tAnalyzing tables" + for tab in table_list: + cnt2 = session.query(tab).count() + print "\t %s: %d entries" % (tab.__tablename__, cnt2) + assert cnt2 == 0 + print "\tRemoving migration info" mi = session.query(MigrationData).filter_by(name=unicode(mt)).one() session.delete(mi) session.commit() session.close() - print "\tDropping tables %r" % (table_list,) + print "\tDropping tables" tables = [model.__table__ for model in table_list] Base_v0.metadata.drop_all(sql_db.engine, tables=tables)