not smart bune

This commit is contained in:
Lynne Megido 2018-11-14 01:16:29 +10:00
parent 4f1561105d
commit 10041f9790
Signed by: lynnesbian
GPG key ID: FB7B970303ACE499

5
web.py
View file

@ -43,7 +43,7 @@ def db_reconnect():
c = db.cursor() c = db.cursor()
dc = db.cursor(dictionary=True) dc = db.cursor(dictionary=True)
return (db, c, dc) return (db, c, dc)
db, c, dc = db_reconnect() db, c, dc = db_reconnect()
c.execute("CREATE TABLE IF NOT EXISTS `data` (username VARCHAR(64) NOT NULL, instance VARCHAR(128) NOT NULL, password TINYTEXT NOT NULL, avi TEXT NOT NULL, secret TINYTEXT NOT NULL, client_id VARCHAR(128) NOT NULL, client_secret TINYTEXT NOT NULL, cc TINYTEXT, ccavi VARCHAR(128) DEFAULT 'https://lynnesbian.space/res/ceres/cc-smol.png', latest_post TINYTEXT, last_check INT DEFAULT 0 NOT NULL, time_between_checks INT DEFAULT %s NOT NULL, settings VARCHAR(4096) DEFAULT %s, PRIMARY KEY(username, instance))", (cfg['min_time_between_checks'], json.dumps(settings),)) c.execute("CREATE TABLE IF NOT EXISTS `data` (username VARCHAR(64) NOT NULL, instance VARCHAR(128) NOT NULL, password TINYTEXT NOT NULL, avi TEXT NOT NULL, secret TINYTEXT NOT NULL, client_id VARCHAR(128) NOT NULL, client_secret TINYTEXT NOT NULL, cc TINYTEXT, ccavi VARCHAR(128) DEFAULT 'https://lynnesbian.space/res/ceres/cc-smol.png', latest_post TINYTEXT, last_check INT DEFAULT 0 NOT NULL, time_between_checks INT DEFAULT %s NOT NULL, settings VARCHAR(4096) DEFAULT %s, PRIMARY KEY(username, instance))", (cfg['min_time_between_checks'], json.dumps(settings),))
db.close() db.close()
@ -190,11 +190,12 @@ def do_login():
def create_password(): def create_password():
db, c, dc = db_reconnect() db, c, dc = db_reconnect()
c.execute("SELECT COUNT(*) FROM data WHERE username = %s AND instance = %s", (session['username'], session['instance'])) c.execute("SELECT COUNT(*) FROM data WHERE username = %s AND instance = %s", (session['username'], session['instance']))
db.close()
if c.fetchone()[0] == 0: if c.fetchone()[0] == 0:
db.close()
return render_template("create_password.html", bg = "background-image:url('{}')".format(session['avi'])) return render_template("create_password.html", bg = "background-image:url('{}')".format(session['avi']))
else: else:
#user already exists in database, so they already have a password #user already exists in database, so they already have a password
db.close()
return redirect(url_for('main')) return redirect(url_for('main'))
@app.route('/internal/create_account', methods=['POST']) @app.route('/internal/create_account', methods=['POST'])