def file_type(self, filename): """ Retrieve the file type :param filename: file's name Return the file type ('.zip'/'.gz'/'.other'...) """ f = os.path.splitext(filename) ft = f[1] if ft not in type_id.keys(): ft = 'other' return ft
def get_file_type(self, filename=None): if not filename: filename = self.filename else: self.filename = filename extension = os.path.splitext(filename)[1] if filename.endswith('tar.gz'): self.file_type = 'tar.gz' elif extension in type_id.keys(): self.file_type = extension else: self.file_type = 'other' return self.file_type