From 88665c9ffa82afd9689ed5bd211c7136853e809b Mon Sep 17 00:00:00 2001
From: jvoisin
Date: Sat, 23 Jul 2011 18:16:37 +0200
Subject: Rename misc.py to office.py

---
 lib/office.py | 139 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 139 insertions(+)
 create mode 100644 lib/office.py

(limited to 'lib/office.py')

diff --git a/lib/office.py b/lib/office.py
new file mode 100644
index 0000000..de38129
--- /dev/null
+++ b/lib/office.py
@@ -0,0 +1,139 @@
+import os
+import mimetypes
+import subprocess
+import tempfile
+import glob
+
+import hachoir_core
+
+import pdfrw
+import mat
+import parser
+
+class TorrentStripper(parser.Generic_parser):
+    '''
+        A torrent file looks like:
+        -root
+            -start
+            -announce
+            -announce-list
+            -comment
+            -created_by
+            -creation_date
+            -encoding
+            -info
+            -end
+    '''
+    def remove_all(self):
+        for field in self.editor['root']:
+            if self._should_remove(field):
+                #FIXME : hachoir does not support torrent metadata editing :<
+                del self.editor['/root/' + field.name]
+        hachoir_core.field.writeIntoFile(self.editor,
+            self.filename + parser.POSTFIX)
+        if self.backup is False:
+            mat.secure_remove(self.filename) #remove the old file
+            os.rename(self.filename + parser.POSTFIX, self.filename)
+
+    def is_clean(self):
+        for field in self.editor['root']:
+            if self._should_remove(field):
+                return False
+        return True
+
+    def get_meta(self):
+        metadata = {}
+        for field in self.editor['root']:
+            if self._should_remove(field):
+                try:#FIXME
+                    metadata[field.name] = field.value
+                except:
+                    metadata[field.name] = 'harmful content'
+        return metadata
+
+    def _should_remove(self, field):
+        if field.name in ('comment', 'created_by', 'creation_date', 'info'):
+            return True
+        else:
+            return False
+
+
+class PdfStripper(parser.Generic_parser):
+    '''
+        Represent a pdf file, with the help of pdfrw
+    '''
+    def __init__(self, filename, realname, backup):
+        self.filename = filename
+        self.backup = backup
+        self.realname = realname
+        self.shortname = os.path.basename(filename)
+        self.mime = mimetypes.guess_type(filename)[0]
+        self.trailer = pdfrw.PdfReader(self.filename)
+        self.writer = pdfrw.PdfWriter()
+        self.convert = 'gm convert -antialias -enhance %s %s'
+
+    def remove_all(self):
+        '''
+            Remove all the meta fields that are compromizing
+        '''
+        self.trailer.Info.Title = ''
+        self.trailer.Info.Author = ''
+        self.trailer.Info.Producer = ''
+        self.trailer.Info.Creator = ''
+        self.trailer.Info.CreationDate = ''
+        self.trailer.Info.ModDate = ''
+
+        self.writer.trailer = self.trailer
+        self.writer.write(self.filename + parser.POSTFIX)
+        if self.backup is False:
+            mat.secure_remove(self.filename) #remove the old file
+            os.rename(self.filename + parser.POSTFIX, self.filename)
+
+    def remove_all_ugly(self):
+        '''
+            Transform each pages into a jpg, clean them,
+            then re-assemble them into a new pdf
+        '''
+        output_file = self.realname + parser.POSTFIX + '.pdf'
+        _, self.tmpdir = tempfile.mkstemp()
+        subprocess.call(self.convert % (self.filename, self.tmpdir +
+            'temp.jpg'), shell=True)#Convert pages to jpg
+
+        for current_file in glob.glob(self.tmpdir + 'temp*'):
+        #Clean every jpg image
+            class_file = mat.create_class_file(current_file, False)
+            class_file.remove_all()
+
+        subprocess.call(self.convert % (self.tmpdir +
+            'temp.jpg*', output_file), shell=True)#Assemble jpg into pdf
+
+        for current_file in glob.glob(self.tmpdir + 'temp*'):
+        #remove jpg files
+            mat.secure_remove(current_file)
+
+        if self.backup is False:
+            mat.secure_remove(self.filename) #remove the old file
+            os.rename(output_file, self.filename)#rename the new
+            name = self.realname
+        else:
+            name = output_file
+        class_file = mat.create_class_file(name, False)
+        class_file.remove_all()
+
+    def is_clean(self):
+        '''
+            Check if the file is clean from harmful metadatas
+        '''
+        for field in self.trailer.Info:
+            if field != '':
+                return False
+        return True
+
+    def get_meta(self):
+        '''
+            return a dict with all the meta of the file
+        '''
+        metadata = {}
+        for key, value in self.trailer.Info.iteritems():
+                metadata[key[1:]] = value[1:-1]
+        return metadata
-- 
cgit v1.3