From 962838424251685fe3ed18fd446c2ab5ecc6c32b Mon Sep 17 00:00:00 2001 From: Oleg Broytman Date: Sun, 5 Mar 2023 17:05:35 +0300 Subject: [PATCH] Feat(reload_db): Update existing row if the full name was changed Count updated rows. --- reload_db.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/reload_db.py b/reload_db.py index d60e3ec..fe25e86 100755 --- a/reload_db.py +++ b/reload_db.py @@ -19,10 +19,10 @@ def convert_str(s): return int(float(s)) -fs_encoding = xsetbg_conf.get("images", "fs_encoding") dump_file = open(sys.argv[1], 'r') xsetbg_db = recreate_db() -count_new = count_old = 0 +count_new = count_old = count_updated = 0 +fs_encoding = xsetbg_conf.get("images", "fs_encoding") with SQLiteMassInsert() as txn: for line in dump_file: @@ -47,9 +47,14 @@ with SQLiteMassInsert() as txn: assert id is None or row.id == id if row.last_shown is not None: assert row.last_shown == timestamp - count_old += 1 + if row.full_name == filename: + count_old += 1 + else: + row.full_name = filename + count_updated += 1 dump_file.close() print "New images:", count_new print "Existing images:", count_old +print "Updated images:", count_updated -- 2.39.5