nathanwoodburn/hns-links
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: Add image before checking files

Browse Source
This commit is contained in:
Nathan Woodburn 2024-02-25 14:23:45 +11:00
parent ed222714ce
commit 96f7f0fece
Signed by: nathanwoodburn
GPG Key ID: 203B000478AD0EF1
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
main.py
View File

@ -204,6 +204,7 @@ def site_post():
address = [i for i in address if i['address'] != ''] address = [i for i in address if i['address'] != '']
data['address'] = address data['address'] = address
data['image'] = ''
if 'image' in request.files: if 'image' in request.files:
if request.files['image'].filename != '' and request.files['image'].filename != None: if request.files['image'].filename != '' and request.files['image'].filename != None:
# Make sure the file is an image # Make sure the file is an image
@ -211,9 +212,7 @@ def site_post():
extension = file.filename.split('.')[-1] extension = file.filename.split('.')[-1]
file.save(f'avatars/{i["name"]}.' + extension) file.save(f'avatars/{i["name"]}.' + extension)
data['image'] = f'{i["name"]}.' + extension data['image'] = f'{i["name"]}.' + extension
else:
data['image'] = ''
with open(f'sites/{i["name"]}.json', 'w') as file: with open(f'sites/{i["name"]}.json', 'w') as file: