feat: Add cookie for referral
All checks were successful
Build Docker / Build Image (push) Successful in 18s

This commit is contained in:
Nathan Woodburn 2023-11-08 13:33:17 +11:00
parent 7c3d101782
commit 784455fe1d
Signed by: nathanwoodburn
GPG Key ID: 203B000478AD0EF1

13
main.py
View File

@ -22,9 +22,17 @@ def send_report(path):
@app.route('/')
def index():
params = request.args
if 'r' in request.cookies:
print("Referer: " + request.cookies['r'])
return render_template('index.html', hidden=request.cookies['r'],address=address)
if 'r' in params:
print("Referer: " + params['r'])
return render_template('index.html', hidden=params['r'],address=address)
# Set cookie
resp = make_response(render_template('index.html', hidden=params['r'],address=address))
resp.set_cookie('r', params['r'], max_age=60*60*24)
return resp
return render_template('index.html',address=address)
@ -45,6 +53,9 @@ def submit():
if hidden == '':
hidden = 'None'
if 'r' in request.cookies:
hidden = request.cookies['r']
status = gift.gift(name, email, hidden, ip)
print(status,flush=True)