From e163520a130145496ceeeeeb94d950f17e7fd76e Mon Sep 17 00:00:00 2001
From: jvoisin
Date: Thu, 23 Jun 2011 19:00:44 +0200
Subject: Implementation of the --backup option, to keep a backup of the
 treated file (disabled by default)

---
 lib/misc.py | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'lib/misc.py')

diff --git a/lib/misc.py b/lib/misc.py
index 56c2274..a8070f1 100644
--- a/lib/misc.py
+++ b/lib/misc.py
@@ -1,5 +1,6 @@
 import parser
 import pdfrw
+import shutil
 
 class PdfStripper(parser.Generic_parser):
     '''
@@ -23,6 +24,10 @@ class PdfStripper(parser.Generic_parser):
 
         self.writer.trailer = self.trailer
         self.writer.write(self.filename + parser.POSTFIX)
+        if self.backup is False:
+            self.secure_remove() #remove the old file
+            shutil.rename(self.filename+ POSTFIX, self.filename)#rename the new
+
 
     def is_clean(self):
         '''
-- 
cgit v1.3