def game_factor_4():
if 'username' not in session:
return redirect('/login')
user_model = UserModel(db.get_connection())
user_model.add_factor_4(session["user_id"])
session['user_factor'] += 4
session['user_game'] -= 400
return redirect('/game')
def game_plus():
if 'username' not in session:
return redirect('/login')
user_model = UserModel(db.get_connection())
user_model.add_game(session["user_id"])
#print(user_model.get_all())
session['user_game'] += session['user_factor']
return redirect('/game')
def registration():
form = RegistrationForm()
user_name = form.user_name
password = form.password
self = render_template('registration.html', title='Регистрация', form=form)
if request.method == 'GET':
return self
elif request.method == 'POST':
if not (request.form.get('user_name') and request.form.get('password')):
return self
# make error введите данные
else:
user_model = UserModel(db.get_connection())
exists = user_model.exists(user_name, password)
if exists[0]:
return self # make error пользователь уже есть
else:
user_model.insert(user_name, password)
exists = user_model.exists(user_name, password)
session['user_name'] = user_name
session['user_id'] = exists[1]
return redirect("/news")
def switch():
if 'user_name' not in session:
return redirect('/login')
sw = UserModel(db.get_connection())
flag = sw.get(session['user_id'])[3]
print(flag)
who = session['user_id']
if flag == '0':
# sw.switch(who, how, id)
return redirect("/{}".format(id))
@app.route('/logout')
def logout():
session.pop('user_name', 0)
session.pop('user_id', 0)
return redirect('/login')
if __name__ == '__main__':
app.run(port=8080, host='127.0.0.1')
def login():
form = LoginForm()
user_name = form.user_name
password = form.password
self = render_template('login.html', title='Авторизация', form=form)
if request.method == 'GET':
return self
elif request.method == 'POST':
if not (request.form.get('user_name') and request.form.get('password')):
return self
# make error введите данные
else:
user_model = UserModel(db.get_connection())
exists = user_model.exists(user_name, password)
if exists[0]:
session['user_name'] = user_name
session['user_id'] = exists[1]
return redirect("/news")
else:
return self
def login():
form = LoginForm()
user_name = form.username.data
password = form.password.data
user_model = UserModel(db.get_connection())
exists = user_model.exists(user_name, password)
if exists[0]:
session['username'] = user_name
session['user_id'] = exists[1]
session['user_game'] = exists[2]
session['user_factor'] = exists[3]
print(user_model.get_all())
return redirect("/game")
else:
user_model.insert(user_name, password)
return render_template('login.html', title='Авторизация', form=form)
def game():
if 'username' not in session:
return redirect('/login')
user_model = UserModel(db.get_connection())
#print(user_model.get_all())
return render_template('game.html')