implement directory not found exception

deuteronomy-works · Feb 20, 2021 · 04440a7 · 04440a7
1 parent dc62880
commit 04440a7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/recursive_size/__init__.py b/recursive_size/__init__.py
@@ -35,14 +35,18 @@ def get_size(folder_name):
     try:
         conts = os.listdir(folder_name)
 
-    except:
+    except NotADirectoryError:
         try:
             return os.stat(folder_name).st_size
 
         except Exception as exc:
             # Send early error back to user
             raise RuntimeError(f'Unable to read {folder_name}') from exc
 
+    except Exception as exc:
+            # Send early error back to user
+            raise RuntimeError(f'Unable to read {folder_name}') from exc
+
     for x in conts:
         y = os.path.join(folder_name, x)
         total += looper(y)