heckyel/mediagoblin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Check that the media_data tables are empty.

Browse Source 
In the analyzing part also check that the media_data tables
are empty (as expected) before dropping them.
This commit is contained in:
Elrond 2012-04-02 14:02:56 +02:00
parent f9d62eccf4
commit 797f4437eb
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mediagoblin/db/sql/convert.py
View File

@ -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)