Change locking boundaries for shared dict. Add a unit test.

[python_utils.git] / directory_filter.py

diff --git a/directory_filter.py b/directory_filter.py
index 03602d156a0929ba6b60ac094d264359e3426d6d..508baf3bc888cf49e832ca72feddeda1797890a0 100644 (file)
--- a/directory_filter.py
+++ b/directory_filter.py
@@ -30,9 +30,11 @@ class DirectoryFileFilter(object):
     >>> os.remove(testfile)
 
     """
+
     def __init__(self, directory: str):
         super().__init__()
         import file_utils
+
         if not file_utils.does_directory_exist(directory):
             raise ValueError(directory)
         self.directory = directory
@@ -49,6 +51,7 @@ class DirectoryFileFilter(object):
 
     def _update_file(self, filename: str, mtime: Optional[float] = None):
         import file_utils
+
         assert file_utils.does_file_exist(filename)
         if mtime is None:
             mtime = file_utils.get_file_raw_mtime(filename)
@@ -97,12 +100,14 @@ class DirectoryAllFilesFilter(DirectoryFileFilter):
 
     >>> os.remove(testfile)
     """
+
     def __init__(self, directory: str):
         self.all_md5s = set()
         super().__init__(directory)
 
     def _update_file(self, filename: str, mtime: Optional[float] = None):
         import file_utils
+
         assert file_utils.does_file_exist(filename)
         if mtime is None:
             mtime = file_utils.get_file_raw_mtime(filename)
@@ -122,4 +127,5 @@ class DirectoryAllFilesFilter(DirectoryFileFilter):
 
 if __name__ == '__main__':
     import doctest
+
     doctest.testmod()