diff options
author | Aneesh Dogra <lionaneesh@gmail.com> | 2012-12-26 05:36:22 (GMT) |
---|---|---|
committer | Aneesh Dogra <lionaneesh@gmail.com> | 2012-12-26 05:36:22 (GMT) |
commit | 1dfb0e67a6af44f65d6e563f83ca0936a383751c (patch) | |
tree | 590278a915e007a268656d0931ba385d8b549a48 | |
parent | 75151314656db1ed5ee26cacbe444b95e58e3748 (diff) |
-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: |