gryf/slack-backup
1
0
Fork 0
mirror of https://github.com/gryf/slack-backup.git synced 2025-12-17 11:30:25 +01:00
Code Issues Projects Releases Wiki Activity

File extension rework for downloaded files

Browse Source
This commit is contained in:
gryf
2019-01-05 16:52:36 +01:00
parent 2a2f58680b
commit 9ab0dd0da0
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
slack_backup/download.py
View File

@@ -60,7 +60,7 @@ class Download(object):
self._hier_created = False self._hier_created = False
self.cookies = {} self.cookies = {}
def download(self, url, filetype): def download(self, url, filetype, ext=None):
""" """
Download asset, return local path to it Download asset, return local path to it
""" """
@@ -68,7 +68,7 @@ class Download(object):
if not self._hier_created: if not self._hier_created:
self._create_assets_dir() self._create_assets_dir()
filepath = self.get_filepath(url, filetype) filepath = self.get_filepath(url, filetype, ext)
temp_file = utils.get_temp_name() temp_file = utils.get_temp_name()
self._download(url, temp_file) self._download(url, temp_file)
@@ -95,7 +95,7 @@ class Download(object):
self._hier_created = True self._hier_created = True
def get_filepath(self, url, filetype): def get_filepath(self, url, filetype, ext=None):
"""Get full path and filename for the file""" """Get full path and filename for the file"""
typemap = {'avatar': self._images, typemap = {'avatar': self._images,
@@ -123,6 +123,9 @@ class Download(object):
utils.makedirs(os.path.join(path, part)) utils.makedirs(os.path.join(path, part))
path = os.path.join(path, part) path = os.path.join(path, part)
if ext and not fname.endswith(ext):
fname = fname + "." + ext
return os.path.join(path, fname) return os.path.join(path, fname)
def calculate_new_filename(self, path, filetype): def calculate_new_filename(self, path, filetype):