diff options
Diffstat (limited to 'lib/misc.py')
| -rw-r--r-- | lib/misc.py | 5 |
1 files changed, 5 insertions, 0 deletions
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 @@ | |||
| 1 | import parser | 1 | import parser |
| 2 | import pdfrw | 2 | import pdfrw |
| 3 | import shutil | ||
| 3 | 4 | ||
| 4 | class PdfStripper(parser.Generic_parser): | 5 | class PdfStripper(parser.Generic_parser): |
| 5 | ''' | 6 | ''' |
| @@ -23,6 +24,10 @@ class PdfStripper(parser.Generic_parser): | |||
| 23 | 24 | ||
| 24 | self.writer.trailer = self.trailer | 25 | self.writer.trailer = self.trailer |
| 25 | self.writer.write(self.filename + parser.POSTFIX) | 26 | self.writer.write(self.filename + parser.POSTFIX) |
| 27 | if self.backup is False: | ||
| 28 | self.secure_remove() #remove the old file | ||
| 29 | shutil.rename(self.filename+ POSTFIX, self.filename)#rename the new | ||
| 30 | |||
| 26 | 31 | ||
| 27 | def is_clean(self): | 32 | def is_clean(self): |
| 28 | ''' | 33 | ''' |
