Handling the absence of db-shm and db-wal files.

Resolves: #43
AmgdGocha · Dec 20, 2024 · c4d3134 · c4d3134
1 parent eb974b6
commit c4d3134
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/drivefs_sleuth/utils.py b/drivefs_sleuth/utils.py
@@ -258,9 +258,10 @@ def get_content_caches_paths(content_cache_dir):
     for root, _, content_caches in os.walk(content_cache_dir):
         for content_cache in content_caches:
             content_caches_paths[content_cache] = os.path.abspath(os.path.join(root, content_cache))
-    del (content_caches_paths['chunks.db'])
-    del (content_caches_paths['chunks.db-shm'])
-    del (content_caches_paths['chunks.db-wal'])
+
+    content_caches_paths.pop('chunks.db', None)
+    content_caches_paths.pop('chunks.db-shm', None)
+    content_caches_paths.pop('chunks.db-wal', None)
 
     return content_caches_paths
 
@@ -271,9 +272,10 @@ def get_thumbnails_paths(thumbnails_dir):
     for root, _, thumbnails in os.walk(thumbnails_dir):
         for thumbnail in thumbnails:
             thumbnails_paths[thumbnail] = os.path.abspath(os.path.join(root, thumbnail))
-    del (thumbnails_paths['chunks.db'])
-    del (thumbnails_paths['chunks.db-shm'])
-    del (thumbnails_paths['chunks.db-wal'])
+
+    thumbnails_paths.pop('chunks.db', None)
+    thumbnails_paths.pop('chunks.db-shm', None)
+    thumbnails_paths.pop('chunks.db-wal', None)
 
     return thumbnails_paths
 

diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
 
 setup(
     name='drivefs_sleuth',
-    version='1.2.0',
+    version='1.2.1',
     description='The ultimate Google Drive File Stream Investigator!',
     long_description=open('README.md', encoding='utf-8').read(),
     long_description_content_type='text/markdown',