def advertise(): with dbapi2.connect(app.config['dsn']) as connection: if(request.method == 'GET'): _advertiseList = dbmanager.getAdvertises(connection) return render_template('advertise.html', advertiseList = _advertiseList) if(request.form["action"] == "Add Advertise"): dbmanager.addAdvertise(request.form['advertise_imageurl'], request.form['advertise_exturl'], request.form['advertise_size'], connection) return redirect(url_for('advertise')) if(request.form["action"] == "Delete"): dbmanager.deleteAdvertise(request.form['id'], connection) return redirect(url_for('advertise')) return render_template('advertise.html')
def login(): with dbapi2.connect(app.config['dsn']) as connection: if(request.method == 'GET'): _info = layoutInfo('All about Snooker','Daily News for Snooker','static/img/home-bg.jpg') _advertiseList = dbmanager.getAdvertises(connection) if(session.get('loginStatus') == 'WrongPass'): _info = layoutInfo('All about Snooker','But you Enter Wrong Username or Password!!','static/img/home-bg.jpg') if(session.get('loginStatus') == 'SameUsername'): _info = layoutInfo('All about Snooker','But Username is used by Another User!!','static/img/home-bg.jpg') return render_template('login.html', info = _info,advertiseList=_advertiseList) if(request.form["action"] == "Log In"): user = dbmanager.checkUserLogin(request.form['username'], request.form['password'], connection) if user is not None: session['loggedUser'] = user.Username session['loggedUserID'] = user.getID() session['loginStatus'] = 'OK' if(user.getAccountType() == 'King'): session['loginStatus'] = 'King' return redirect(url_for('adminPage')) else: session['loginStatus'] = 'Normal' return redirect(url_for('home')) else: session['loggedUser'] = '******' session['loggedUserID'] = ' ' session['loginStatus'] = 'WrongPass' return redirect(url_for('login')) if(request.form["action"] == "Register"): result = dbmanager.addUser(request.form['firstname'],request.form['lastname'], request.form['age'],request.form['gender'],request.form['email'],request.form['username'],request.form['password'],connection) session['loggedUser'] = '******' session['loggedUserID'] = ' ' session['loginStatus'] = result if(result != 'OK'): return redirect(url_for('login')) return redirect(url_for('home'))