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

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

Browse Source
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
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
main.py
View File

@ -22,9 +22,17 @@ def send_report(path):
@app.route('/') @app.route('/')
def index(): def index():
params = request.args 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: if 'r' in params:
print("Referer: " + params['r']) 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) return render_template('index.html',address=address)
@ -45,6 +53,9 @@ def submit():
if hidden == '': if hidden == '':
hidden = 'None' hidden = 'None'
if 'r' in request.cookies:
hidden = request.cookies['r']
status = gift.gift(name, email, hidden, ip) status = gift.gift(name, email, hidden, ip)
print(status,flush=True) print(status,flush=True)