feat: Add cookie for referral
All checks were successful
Build Docker / Build Image (push) Successful in 18s
All checks were successful
Build Docker / Build Image (push) Successful in 18s
This commit is contained in:
parent
7c3d101782
commit
784455fe1d
13
main.py
13
main.py
@ -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)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user