summaryrefslogtreecommitdiff
path: root/lib/archive.py
blob: 8fa826ea86785e24956c19dc116d2c181b38f77a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
import tarfile
import sys
import parser
import mat
import shutil
import os
import mimetypes

class TarStripper(parser.Generic_parser):
    def __init__(self, realname, filename, parser, editor, backup):
        super(TarStripper, self).__init__(realname,
            filename, parser, editor, backup)
        self.compression = ''
        self.shortname = os.path.basename(filename)
        self.mime = mimetypes.guess_type(filename)[0]
        self.tarin = tarfile.open(self.filename, 'r' + self.compression)
        self.folder_list = []

    def _remove(self, current_file):
        '''
            remove the meta added by tar itself to the file
        '''
        current_file.mtime = 0
        current_file.uid = 0
        current_file.gid = 0
        current_file.uname = ''
        current_file.gname = ''
        return current_file

    def remove_all(self):
        self.tarin = tarfile.open(self.filename, 'r' + self.compression)
        self.tarout = tarfile.open(self.filename + parser.POSTFIX,
            'w' + self.compression)
        for current_file in self.tarin.getmembers():
            self.tarin.extract(current_file)
            if current_file.type is '0': #is current_file a regular file ?
                #no backup file
                class_file = mat.create_class_file(current_file.name, False)
                class_file.remove_all()
                self.tarout.add(current_file.name, filter=self._remove)
                mat.secure_remove(current_file.name)
            else:
                self.folder_list.insert(0, current_file.name)
        self.tarin.close()
        self.tarout.close()
        print self.folder_list

        self.remove_folder()

        if self.backup is False:
            mat.secure_remove(self.filename)
            os.rename(self.filename + parser.POSTFIX, self.filename)

    def is_file_clean(self, current_file):
        '''
            Check metadatas added by tar
        '''
        if current_file.mtime is not 0:
            return False
        elif current_file.uid is not 0:
            return False
        elif current_file.gid is not 0:
            return False
        elif current_file.uname is not '':
            return False
        elif current_file.gname is not '':
            return False
        else:
            return True

    def is_clean(self):
        self.tarin = tarfile.open(self.filename, 'r' + self.compression)
        for current_file in self.tarin.getmembers():
            self.tarin.extract(current_file)
            if current_file.type is '0': #is current_file a regular file ?
                #no backup file
                class_file = mat.create_class_file(current_file.name, False)
                if not class_file.is_clean():
                    mat.secure_remove(current_file.name)
                    self.remove_folder()
                    return False
                if not self.is_file_clean(current_file):
                    return False
                mat.secure_remove(current_file.name)
            else:
                self.folder_list.insert(0, current_file.name)
        self.tarin.close()
        self.remove_folder()
        return True

    def get_meta(self):
        self.tarin = tarfile.open(self.filename, 'r' + self.compression)
        metadata = {}
        for current_file in self.tarin.getmembers():
            if current_file.type is '0':
                if not self.is_file_clean(current_file):#if there is meta
                    current_meta = {}
                    current_meta['mtime'] = current_file.mtime
                    current_meta['uid'] = current_file.uid
                    current_meta['gid'] = current_file.gid
                    current_meta['uname'] = current_file.uname
                    current_meta['gname'] = current_file.gname
                    metadata[current_file.name] = current_meta
        return metadata

    def remove_folder(self):
        for folder in self.folder_list: #delete remainings folders
            shutil.rmtree(folder)
        self.folder_list = []

class GzipStripper(TarStripper):
    def __init__(self, realname, filename, parser, editor, backup):
        super(GzipStripper, self).__init__(realname,
            filename, parser, editor, backup)
        self.compression = ':gz'

class Bzip2Stripper(TarStripper):
    def __init__(self, realname, filename, parser, editor, backup):
        super(Bzip2Stripper, self).__init__(realname,
            filename, parser, editor, backup)
        self.compression = ':bz2'