fix drop column

This commit is contained in:
j 2016-01-17 13:49:31 +05:30
commit b00647e3f3
2 changed files with 9 additions and 6 deletions

View file

@ -783,11 +783,11 @@ def remove_unused_names():
state.db.session.commit()
def update_sort_table():
current = db.get_table_columms('sort')
current = db.get_table_columns('sort')
drop_columns = list(set(current) - set(Item.sort_keys+['item_id']))
add_columns = list(set(Item.sort_keys+['item_id'])-set(current))
if drop_columns:
db.drop_columns('sort', drop_columns)
add_columns = list(set(Item.sort_keys)-set(current+['item_id']))
if add_columns:
create_table = str(CreateTable(Sort.__table__).compile(db.engine)).split('\n')
sql = []