feat: Add delete image button

This commit is contained in:
2024-02-25 07:59:07 +11:00
parent c24ab6ad48
commit 2b38a128a8
2 changed files with 24 additions and 2 deletions

22
main.py
View File

@@ -216,6 +216,28 @@ def site_post():
response.set_cookie('auth', '', expires=0)
return response
@app.route('/image/delete')
def delete_image():
if 'auth' not in request.cookies:
return redirect('/')
auth = request.cookies['auth']
for i in cookies:
if i['cookie'] == auth:
# Get site content
if os.path.isfile(f'sites/{i["name"]}.json'):
with open(f'sites/{i["name"]}.json') as file:
data = json.load(file)
if 'image' in data:
data['image'] = ''
with open(f'sites/{i["name"]}.json', 'w') as file:
json.dump(data, file)
return redirect('/site')
response = make_response(redirect('/'))
response.set_cookie('auth', '', expires=0)
return response
@app.route('/preview')
def site_preview():
if 'auth' not in request.cookies: