Revision 8c574358 image_creator/os_type/unix.py

b/image_creator/os_type/unix.py
1 1
#!/usr/bin/env python
2 2

  
3 3
import re
4
import sys
4 5

  
5 6
from image_creator.os_type import OSBase
6 7

  
8

  
7 9
class Unix(OSBase):
8 10

  
9 11
    sensitive_userdata = ['.bash_history']
......
36 38
        self.cleanup_log()
37 39

  
38 40
    def cleanup_tmp(self):
39
        files = []
40
        files.extend(self.ls('/tmp/'))
41
        files.extend(self.ls('/var/tmp/'))
42
    
43
        for filename in files:
44
            self.g.rm_rf(filename)
41
        self.foreach_file('/tmp', self.g.rm_rf, maxdepth=1)
45 42

  
46 43
    def cleanup_log(self):
47
        files = self.find( '/var/log/')
44
        self.foreach_file('/var/log', self.g.truncate, ftype='r')
48 45

  
49
        for filename in filter(self.g.is_file, files):
50
            self.g.truncate(filename)
51
        
52 46
    def cleanup_userdata(self):
53 47
        homedirs = ['/root'] + self.ls('/home/')
54 48

  
55 49
        for homedir in homedirs:
56 50
            for data in self.sensitive_userdata:
57 51
                fname = "%s/%s" % (homedir, data)
58
                print "Filename: %s\n" % fname
59 52
                if self.g.is_file(fname):
60 53
                    self.g.scrub_file(fname)
61 54

  

Also available in: Unified diff