From 7ad9ff08ad52d16b05e689ef383e395b7731d594 Mon Sep 17 00:00:00 2001
From: jvoisin
Date: Tue, 20 Mar 2018 23:35:02 +0100
Subject: Add a test for PNG files

---
 src/parsers/png.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

(limited to 'src')

diff --git a/src/parsers/png.py b/src/parsers/png.py
index 3c0a507..20cd234 100644
--- a/src/parsers/png.py
+++ b/src/parsers/png.py
@@ -7,12 +7,21 @@ from . import abstract
 
 class PNGParser(abstract.AbstractParser):
     mimetypes = {'image/png', }
-    meta_list = set()
+    meta_whitelist = {'SourceFile', 'ExifToolVersion', 'FileName',
+            'Directory', 'FileSize', 'FileModifyDate', 'FileAccessDate',
+            "FileInodeChangeDate", 'FilePermissions', 'FileType',
+            'FileTypeExtension', 'MIMEType', 'ImageWidth', 'BitDepth', 'ColorType',
+            'Compression', 'Filter', 'Interlace', 'BackgroundColor', 'ImageSize',
+            'Megapixels', 'ImageHeight'}
 
     def get_meta(self):
         out = subprocess.check_output(['exiftool', '-json', self.filename])
-        return json.loads(out)[0]
+        meta = json.loads(out)[0]
+        for key in self.meta_whitelist:
+            meta.pop(key, None)
+        return meta
 
     def remove_all(self):
         surface = cairo.ImageSurface.create_from_png(self.filename)
         surface.write_to_png(self.output_filename)
+        return True
-- 
cgit v1.3