From 79ce29a7d5e41bb3bb2499bc7eb99164c423aa1d Mon Sep 17 00:00:00 2001
From: jvoisin
Date: Sun, 22 Feb 2015 10:10:50 +0100
Subject: Preliminary implementation of PIL for images.

PIL (Python Image Library) is used to open, then save
images, in order to remove unknown metadata.
---
 libmat/images.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

(limited to 'libmat/images.py')

diff --git a/libmat/images.py b/libmat/images.py
index 67c710f..2daea88 100644
--- a/libmat/images.py
+++ b/libmat/images.py
@@ -7,9 +7,21 @@ References:
 '''
 
 import parser
+import pillow
 
 
-class JpegStripper(parser.GenericParser):
+class ImageStripper(parser.GenericParser, pillow.PillowStripper):
+    ''' Common stripper for images.
+    Its purpose is to open then save
+    images with PIL, the goal being to remove
+    unknown metadata.
+    '''
+    def remove_all(self):
+        self.open_and_save()
+        super(ImageStripper, self).remove_all()
+
+
+class JpegStripper(ImageStripper):
     ''' Represents a jpeg file.
         Custom Huffman and Quantization tables
         are stripped: they may leak
@@ -34,7 +46,7 @@ class JpegStripper(parser.GenericParser):
         return True
 
 
-class PngStripper(parser.GenericParser):
+class PngStripper(ImageStripper):
     ''' Represents a png file
     '''
     def _should_remove(self, field):
-- 
cgit v1.3