nathanwoodburn/shakecities
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: Cleanup file name on upload
All checks were successful
Build Docker / Build SLDs Image (push) Successful in 19s
Details
Build Docker / Build Main Image (push) Successful in 21s
Details

Browse Source
This commit is contained in:
Nathan Woodburn 2023-11-18 11:52:51 +11:00
parent eb53b955b8
commit d879aa32d6
Signed by: nathanwoodburn
GPG Key ID: 203B000478AD0EF1
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
avatar.py
View File

@ -4,6 +4,7 @@ import db
import random import random
import json import json
import varo import varo
import re
IMAGE_LOCATION = os.getenv('IMAGE_LOCATION') IMAGE_LOCATION = os.getenv('IMAGE_LOCATION')
@ -25,7 +26,10 @@ def allowed_file(filename):
return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
def save_avatar(file,owner): def save_avatar(file,owner):
filename = file.filename filename = re.sub(r'[^a-zA-Z0-9]', '', filename).lower()
while os.path.exists(f"{IMAGE_LOCATION}/{filename}"): while os.path.exists(f"{IMAGE_LOCATION}/{filename}"):
filename = f"{random.randint(0,1000000)}-{filename}" filename = f"{random.randint(0,1000000)}-{filename}"
file.save(f"{IMAGE_LOCATION}/{filename}") file.save(f"{IMAGE_LOCATION}/{filename}")