AdamAnt/ClockworkAprilFools2017
1
0
Fork 0
mirror of https://github.com/Adam-Ant/ClockworkAprilFools2017 synced 2024-12-20 11:04:35 +00:00
Code Issues Releases Wiki Activity

Added some sanity checking to username

Browse Source
This commit is contained in:
Adam Dodman 2017-03-31 17:24:10 +01:00
parent c59e2e87fc
commit 979f23b2de
1 changed files with 37 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
website/webinterface.py
View File

@ -1,4 +1,4 @@
from flask import Flask, render_template, request, session, redirect, url_for from flask import Flask, render_template, request, session, redirect, url_for, flash
from os import urandom as rand from os import urandom as rand
import pprint import pprint
import ts3 import ts3
@ -33,14 +33,46 @@ def teamspeakClientAdd(clientName):
except: except:
print("ERROR") print("ERROR")
def teamspeakClientCheck(clientName):
with ts3.query.TS3Connection('magic.adam-ant.co.uk', '10011') as ts3conn:
try:
ts3conn.login(client_login_name='serveradmin',client_login_password='DE0xWKTx')
except ts3.query.TS3QueryError as err:
print(err)
ts3conn.use(sid=1)
try:
print('before')
client = ts3conn.clientfind(pattern=clientName)
print('after')
clientid = client[0]['clid']
groups = ts3conn.clientinfo(clid=clientid)[0]['client_servergroups']
except ts3.query.TS3QueryError as ts3err:
if ts3err.resp.error['id'] == '512':
session.pop('username', None)
flash("ERROR: Username not found. Type it carefully cos Clockwork PLUS! is gonna cost ya...","danger")
return redirect(url_for('index'))
return "Ooops! Tell Adam this: <br>" + ts3err.resp.error['msg']
if "9," in groups:
print("FeckOffMate!")
flash("ERROR: You already have Clockwork PLUS, ya dingus","danger")
session.pop('username', None)
return redirect(url_for('index'))
else:
return redirect(url_for("payment"))
@app.route("/") @app.route("/")
def index(): def index():
return render_template("index.html") return render_template("index.html")
@app.route("/collectsunlight") @app.route("/collectsunlight")
def wait(): def wait():
if "username" in session:
return render_template("wait.html") return render_template("wait.html")
else:
return redirect(url_for("buy"))
@app.route("/pay") @app.route("/pay")
def payment(): def payment():
@ -53,6 +85,7 @@ def payment():
def activate(): def activate():
if "username" in session: if "username" in session:
if teamspeakClientAdd(session['username']): if teamspeakClientAdd(session['username']):
flash("Success! You have been upgraded to Clockwork PLUS! Enjoy!","success")
return redirect(url_for('index')) return redirect(url_for('index'))
else: else:
return "Adam fucked up! go nag him" return "Adam fucked up! go nag him"
@ -64,7 +97,7 @@ def activate():
def buy(): def buy():
if request.method == 'POST': if request.method == 'POST':
session['username'] = request.form['username'] session['username'] = request.form['username']
return redirect(url_for("payment")) return teamspeakClientCheck(session['username'])
return render_template("buy.html") return render_template("buy.html")
if __name__ == '__main__': if __name__ == '__main__':