From b2dce6bac00c4a562aaf99439d5a8cc375f859a2 Mon Sep 17 00:00:00 2001 From: Nikolay Simakov Date: Sat, 11 Jul 2020 21:46:06 -0400 Subject: [PATCH] added check on table presence on update --- akrr/update.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/akrr/update.py b/akrr/update.py index f40707a6..02d66a6d 100644 --- a/akrr/update.py +++ b/akrr/update.py @@ -768,11 +768,18 @@ def _update_db_populate_new_db(self): ak_con, ak_cur = self.get_old_ak_db_con() tables_to_load = self.tables_to_drop['mod_appkernel']['tables'] tables_to_load.reverse() + + cursor_execute(ak_cur, "show tables") + ak_tables = tuple((r[0] for r in akrr_cur.fetchall())) + for table_name in tables_to_load: table_pkl_name = self._get_table_pkl_name("mod_appkernel", table_name) if not os.path.isfile(table_pkl_name): log.debug("Table %s was not saved (might not present in previous version)", table_name) continue + if table_name not in ak_tables: + log.debug("Table %s is not in current mod_appkernel", table_name) + continue log.debug("Populating: mod_appkernel.%s" % table_name) cursor_execute(ak_cur, "show columns from %s" % table_name)