From 529a01269d0732e5439aa137be3f734be044ffca Mon Sep 17 00:00:00 2001 From: j Date: Thu, 7 Jan 2016 15:53:41 +0530 Subject: [PATCH] only migrate if user exists --- oml/setup.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/oml/setup.py b/oml/setup.py index 0bff8c9..8114d1c 100644 --- a/oml/setup.py +++ b/oml/setup.py @@ -248,21 +248,22 @@ def upgrade_db(old, new=None): if old <= '20160106-497-c86ba8a': with db.session() as session: u = User.get(settings.USER_ID) - l = u.library - for i in u.items.all(): - if not i in l.items: - l.items.append(i) - session.add(l) - u.clear_list_cache() - for u in User.query.filter_by(peered=True): + if u: l = u.library for i in u.items.all(): if not i in l.items: l.items.append(i) session.add(l) u.clear_list_cache() - l.items_count() - session.commit() + for u in User.query.filter_by(peered=True): + l = u.library + for i in u.items.all(): + if not i in l.items: + l.items.append(i) + session.add(l) + u.clear_list_cache() + l.items_count() + session.commit() if old <= '20160107-508-a0c1970': with db.session() as session: for l in List.query.filter_by(name=' [2]'):