diff --git a/web.py b/web.py index 5b7c12b..706841e 100755 --- a/web.py +++ b/web.py @@ -55,12 +55,14 @@ def home(): if 'acct' in session: if 'cc' not in session: session['cc'] = "None" - if session['cc'] == "None": + if session['cc'] == "None" or 'ccavi' not in session: #every time home is rendered without cc being set - c.execute("SELECT cc FROM `data` WHERE client_id LIKE %s AND instance LIKE %s", (session['client_id'], session['instance'])) - cc = c.fetchone()[0] - if cc != '': - session['cc'] = cc + c.execute("SELECT cc, ccavi FROM `data` WHERE client_id LIKE %s AND instance LIKE %s", (session['client_id'], session['instance'])) + cc = c.fetchone() + if cc[0] != '': + session['cc'] = cc[0] + session['ccavi'] = cc[1] + if 'last_avi_update' not in session or session['last_avi_update'] + (24 * 60 * 60) < time.time(): #avatars haven't been updated for over 24 hours, update them now @@ -75,7 +77,7 @@ def home(): c.execute("UPDATE data SET avi = %s, ccavi = %s WHERE client_id LIKE %s AND instance LIKE %s", (session['avi'], session['ccavi'], session['client_id'], session['instance'])) else: c.execute("UPDATE data SET avi = %s WHERE client_id LIKE %s AND instance LIKE %s", (session['avi'], session['client_id'], session['instance'])) - return render_template("home.html") + return render_template("home.html", mabg="background-image:url('{}')".format(session['avi']), ccbg="background-image:url('{}')".format(session['ccavi'])) else: return redirect(url_for('main'))