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
