comparison utils.py @ 16:c21e96bb68c8

added clean-up after execution
author pieter.lukasse@wur.nl
date Fri, 23 Jan 2015 14:57:55 +0100
parents 8c20185752da
children a394029414c6
comparison
equal deleted inserted replaced
15:76d6d8a5fa29 16:c21e96bb68c8
4 @author: lukas007 4 @author: lukas007
5 ''' 5 '''
6 import shutil 6 import shutil
7 import subprocess 7 import subprocess
8 import csv 8 import csv
9 import os
10 import stat
9 from collections import OrderedDict 11 from collections import OrderedDict
10 12
11 def copy_dir(src, dst): 13 def copy_dir(src, dst):
12 shutil.copytree(src, dst) 14 shutil.copytree(src, dst)
15
16 def _del_rw(action, name, exc):
17 '''
18 ensures the read only files are set to read/write
19 and then deletes them
20 '''
21 os.chmod(name, stat.S_IWRITE)
22 os.remove(name)
23
24 def remove_dir(src):
25 shutil.rmtree(src, onerror=_del_rw)
26
13 27
14 def log_message(log_file, log_message): 28 def log_message(log_file, log_message):
15 with open(log_file, "a") as text_file: 29 with open(log_file, "a") as text_file:
16 text_file.write(log_message + "\n") 30 text_file.write(log_message + "\n")
17 31