From 082341090e6e16054e7ba5108c0894211f92ff21 Mon Sep 17 00:00:00 2001 From: "Antonio J. Delgado" Date: Fri, 3 Feb 2023 09:26:27 +0200 Subject: [PATCH] fix result when exception --- find_duplicate_files/find_duplicate_files.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/find_duplicate_files/find_duplicate_files.py b/find_duplicate_files/find_duplicate_files.py index 1d590a4..344946d 100644 --- a/find_duplicate_files/find_duplicate_files.py +++ b/find_duplicate_files/find_duplicate_files.py @@ -61,14 +61,15 @@ class find_duplicate_files: def _check_file_cache(self, file): file_sql = file.replace("'", "''") query = f"SELECT hash FROM files WHERE file = '{file_sql}'" + row = False if isinstance(query, bytes): query = query.decode('utf-8') try: result = self.cur.execute(query) + row = result.fetchone() except Exception as error: self._log.error(f"Error executing query '{query}'. {error}") #sys.exit(2) - row = result.fetchone() if row and len(row) > 0: return row[0] else: @@ -77,14 +78,15 @@ class find_duplicate_files: def _cache_file(self, file, hash): file_sql = file.replace("'", "''") query = f"INSERT INTO files (file, hash) VALUES ('{file_sql}', '{hash}')" + result = False if isinstance(query, bytes): query = query.decode('utf-8') try: result = self.cur.execute(query) + self.cache_db.commit() except Exception as error: self._log.error(f"Error executing query '{query}'. {error}") #sys.exit(3) - self.cache_db.commit() return result def _cache_size(self):