ClockworkAprilFools2017/website/webinterface.py

from flask import Flask, render_template, request, session, redirect, url_for
from os import urandom as rand
import pprint
import ts3

pp = pprint.PrettyPrinter(indent=4)



app = Flask(__name__)
app.secret_key = rand(24)

def teamspeakClientAdd(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:
            clientdbid = ts3conn.clientdbfind(pattern=clientName)[0]['cldbid']
        except ts3.query.TS3QueryError:
            return False

        try:
            ts3conn.servergroupaddclient(sgid=9,cldbid=clientdbid)
            ts3conn.servergroupaddclient(sgid=10,cldbid=clientdbid)
            ts3conn.servergroupaddclient(sgid=11,cldbid=clientdbid)
            ts3conn.servergroupaddclient(sgid=12,cldbid=clientdbid)
            return True
        except:
            print("ERROR")

@app.route("/")
def index():
    return render_template("index.html")

@app.route("/collectsunlight")
def wait():
    return render_template("wait.html")


@app.route("/pay")
def payment():
    if "username" in session:
        return render_template("payment.html")
    else:
        return redirect(url_for("buy"))

@app.route("/activate")
def activate():
    if "username" in session:
        if teamspeakClientAdd(session['username']):
            return redirect(url_for('index'))
        else:
            return "Adam fucked up! go nag him"
    else:
        return redirect(url_for("buy"))


@app.route("/buy", methods=['GET', 'POST'])
def buy():
    if request.method == 'POST':
        session['username'] = request.form['username']
        return redirect(url_for("payment"))
    return render_template("buy.html")

if __name__ == '__main__':
    app.run(host='0.0.0.0',debug=True)