Preliminary implementation of PIL for images.

PIL (Python Image Library) is used to open, then save
images, in order to remove unknown metadata.

1 files changed, 14 insertions, 2 deletions

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):