diff --git a/image_classifier/image_classifier.py b/image_classifier/image_classifier.py index 0fdfd99..f8657f2 100755 --- a/image_classifier/image_classifier.py +++ b/image_classifier/image_classifier.py @@ -71,11 +71,17 @@ class image_classifier: entries = self.recursive_scandir(directory) else: entries = os.scandir(directory) - #self._log.debug(f"Processing {len(entries)} files...") + self._log.debug(f"Processing {self.count(entries)} files...") for entry in entries: if not entry.name.startswith('.') and entry.is_file(): self.process_file(entry.path) + def count(self, iterable): + count = 0 + for item in iterable: + count += 1 + return count + def recursive_scandir(self, path, ignore_hidden_files=True): files = [] try: