From a8195f5d8034f82aef04d2da223fc0a9c0dc9d38 Mon Sep 17 00:00:00 2001
From: jvoisin
Date: Mon, 27 Jan 2014 22:19:50 +0000
Subject: Non-writtables test files are now chmod'ed to be properly removed.

---
 MAT/mat.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'MAT/mat.py')

diff --git a/MAT/mat.py b/MAT/mat.py
index 4d64c2d..32f55d8 100644
--- a/MAT/mat.py
+++ b/MAT/mat.py
@@ -112,6 +112,13 @@ class XMLParser(xml.sax.handler.ContentHandler):
 def secure_remove(filename):
     ''' Securely remove the file
     '''
+    # I want the file removed, even if it's ro
+    try:
+        os.chmod(filename, 0o777)
+    except OSError:
+        logging.error('Unable to add write rights to %s' % filename)
+        raise MAT.exceptions.UnableToWriteFile
+
     try:
         if not subprocess.call(['shred', '--remove', filename]):
             return True
@@ -122,11 +129,12 @@ def secure_remove(filename):
 
     try:
         os.remove(filename)
-        return True
     except OSError:
         logging.error('Unable to remove %s' % filename)
         raise MAT.exceptions.UnableToRemoveFile
 
+    return True
+
 
 def create_class_file(name, backup, **kwargs):
     ''' Return a $FILETYPEStripper() class,
-- 
cgit v1.3