From e387e30983844bd5bc5b18a6c55de9399539a871 Mon Sep 17 00:00:00 2001 From: Inex Code Date: Thu, 22 Sep 2022 18:34:33 +0300 Subject: [PATCH] Fix handling of FileNotFoundError during size calculation --- selfprivacy_api/dependencies.py | 2 +- selfprivacy_api/services/generic_size_counter.py | 7 ++++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/selfprivacy_api/dependencies.py b/selfprivacy_api/dependencies.py index fa101d2..5a0ba67 100644 --- a/selfprivacy_api/dependencies.py +++ b/selfprivacy_api/dependencies.py @@ -27,4 +27,4 @@ async def get_token_header( def get_api_version() -> str: """Get API version""" - return "2.0.7" + return "2.0.8" diff --git a/selfprivacy_api/services/generic_size_counter.py b/selfprivacy_api/services/generic_size_counter.py index 4a706fb..71dff82 100644 --- a/selfprivacy_api/services/generic_size_counter.py +++ b/selfprivacy_api/services/generic_size_counter.py @@ -12,5 +12,10 @@ def get_storage_usage(path: str) -> int: for iter_path in pathlib.Path(path).rglob("**/*"): if iter_path.is_dir(): continue - storage_usage += iter_path.stat().st_size + try: + storage_usage += iter_path.stat().st_size + except FileNotFoundError: + pass + except Exception as error: + print(error) return storage_usage