diff --git a/auth.py b/auth.py index 8ec5c08..5a92a52 100644 --- a/auth.py +++ b/auth.py @@ -1,4 +1,4 @@ -from flask import Flask, render_template, request, session +from flask import Flask, render_template, request, redirect, url_for, session from functools import wraps from datetime import datetime, date, timedelta import io @@ -78,7 +78,6 @@ def require_secret(f): return decorated_function - @require_secret def mylinks(): allowed_secrets = session.get('allowed_secrets', []) @@ -107,6 +106,19 @@ def mylinks(): else: secret_folders[secret] = [] + return render_template('mylinks.html', + allowed_secrets=allowed_secrets, + secret_qr_codes=secret_qr_codes, + secret_folders=secret_folders) + +def remove_secret(): + secret_to_remove = request.form.get('secret') + allowed_secrets = session.get('allowed_secrets', []) + if secret_to_remove in allowed_secrets: + allowed_secrets.remove(secret_to_remove) + session['allowed_secrets'] = allowed_secrets + return redirect(url_for('mylinks')) + return render_template('mylinks.html', allowed_secrets=allowed_secrets, secret_qr_codes=secret_qr_codes, diff --git a/templates/mylinks.html b/templates/mylinks.html index 9a52968..ad3b899 100644 --- a/templates/mylinks.html +++ b/templates/mylinks.html @@ -46,6 +46,10 @@ https://{{ request.host }}?secret={{ secret }}
+ {% if secret_folders[secret] %}