X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=directory_filter.py;h=508baf3bc888cf49e832ca72feddeda1797890a0;hb=865825894beeedd47d26dd092d40bfee582f5475;hp=03602d156a0929ba6b60ac094d264359e3426d6d;hpb=52290f9c9e0eeaba3d5a067043f5ba98c9b386e5;p=python_utils.git diff --git a/directory_filter.py b/directory_filter.py index 03602d1..508baf3 100644 --- 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()