From 42af39d105892de1d16fb1e4ba41e39c9956cf07 Mon Sep 17 00:00:00 2001 From: Victor Date: Thu, 15 Apr 2021 00:38:10 +0300 Subject: [PATCH] Add png processing --- picsorter.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/picsorter.py b/picsorter.py index 2041136..6fc673e 100644 --- a/picsorter.py +++ b/picsorter.py @@ -41,7 +41,8 @@ class PicSorter: library = Library(self.dir_library) metadata = Metadata(self.dir_tmp) db = Database() - for filename in self.dir_input.rglob('*.jpg'): + files = {p for p in self.dir_input.iterdir() if p.suffix in [".jpg", ".png"]} + for filename in files: print("Process ", filename) try: url = iqdb.search(filename)