diff options
-rw-r--r-- | WebcamImages.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/WebcamImages.py b/WebcamImages.py index 0356fbd..d6e3fd8 100644 --- a/WebcamImages.py +++ b/WebcamImages.py @@ -78,7 +78,8 @@ class WebcamImages(): GdkPixbuf.InterpType.BILINEAR) image_widget.set_from_pixbuf(pixbuf) for a in image_list: - os.remove(a) + if os.path.splitext(a)[-1] == '.png': + os.remove(a) self.image_updater_thread = GObject.timeout_add(1000 / fps, self.update_image, @@ -87,7 +88,8 @@ class WebcamImages(): def clean_dir(self): image_list = os.listdir(WEBCAM_IMAGES_DIR) for image in image_list: - os.remove(os.path.join(WEBCAM_IMAGES_DIR, image)) + if os.path.splitext(image)[-1] == '.png': + os.remove(os.path.join(WEBCAM_IMAGES_DIR, image)) def cleanup(self): if self.image_updater_thread: |