mcvsoftware/socializer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added a safe-filename function so we will be sure the suggested filename won't break anything in the operating system

Browse Source
This commit is contained in:
Manuel Cortez 2019-10-14 17:25:06 -05:00
parent 9e6e72c0a1
commit 0cdfc253d2
3 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/controller/buffers.py
View File

@ -594,7 +594,7 @@ class documentBuffer(feedBuffer):
def download(self, *args, **kwargs): def download(self, *args, **kwargs):
post = self.get_post() post = self.get_post()
filename = post["title"] filename = utils.safe_filename(post["title"])
# If document does not end in .extension we must fix it so the file dialog will save it properly later. # If document does not end in .extension we must fix it so the file dialog will save it properly later.
if filename.endswith(post["ext"]) == False: if filename.endswith(post["ext"]) == False:
filename = filename+ "."+post["ext"] filename = filename+ "."+post["ext"]

2
src/presenters/displayPosts/audio.py
View File

@ -73,7 +73,7 @@ class displayAudioPresenter(base.basePresenter):
def get_suggested_filename(self, audio_index): def get_suggested_filename(self, audio_index):
post = self.post[audio_index] post = self.post[audio_index]
return "{0} - {1}.mp3".format(post["title"], post["artist"]) return utils.safe_filename("{0} - {1}.mp3".format(post["title"], post["artist"]))
def download(self, audio_index, path): def download(self, audio_index, path):
post = self.post[audio_index] post = self.post[audio_index]

6
src/sessionmanager/utils.py
View File

@ -90,4 +90,8 @@ def transform_audio_url(url):
url = url.replace("/"+parts[-2], "") url = url.replace("/"+parts[-2], "")
else: else:
url = url.replace("/"+parts[-3], "") url = url.replace("/"+parts[-3], "")
return url return url
def safe_filename(filename):
allowed_symbols = ["_", ".", ",", "-", "(", ")"]
return "".join([c for c in filename if c.isalpha() or c.isdigit() or c==' ' or c in allowed_symbols]).rstrip()