75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
from . import main
|
|
from flask import Flask, request, render_template, redirect, url_for, flash, make_response, session
|
|
from flask_login import login_required, login_user, current_user, logout_user
|
|
from app.models import User, Post, Category, Feedback, db
|
|
from .forms import ContactForm, LoginForm
|
|
from werkzeug.exceptions import abort
|
|
|
|
|
|
@main.route('/admin/')
|
|
@login_required
|
|
def admin():
|
|
return render_template('admin.html')
|
|
|
|
|
|
@main.route('/')
|
|
def index():
|
|
name = None
|
|
if current_user.is_authenticated:
|
|
name = current_user.name
|
|
return render_template('index.html', name=name)
|
|
|
|
|
|
@main.route('/login/', methods=['post', 'get'])
|
|
def login():
|
|
if current_user.is_authenticated:
|
|
return redirect(url_for('.admin'))
|
|
form = LoginForm()
|
|
if form.validate_on_submit():
|
|
user = db.session.query(User).filter(User.username == form.username.data).first()
|
|
if user and user.check_password(form.password.data):
|
|
login_user(user, remember=form.remember.data)
|
|
nextpage = request.args.get('next', url_for('.admin'))
|
|
return redirect(nextpage)
|
|
else:
|
|
flash("Invalid username/password", 'error')
|
|
return render_template('login.html', form=form)
|
|
|
|
|
|
@main.route('/user/<int:user_id>/')
|
|
def user_profile(user_id):
|
|
user = db.session.query(User).filter(User.id == user_id).first()
|
|
if not user:
|
|
abort(404)
|
|
return "Profile for user: {}".format(user.name)
|
|
|
|
|
|
@main.route('/logout/')
|
|
@login_required
|
|
def logout():
|
|
logout_user()
|
|
flash("You have been logged out.")
|
|
return redirect(url_for('.login'))
|
|
|
|
|
|
@main.route('/contact/', methods=['get', 'post'])
|
|
def contact():
|
|
form = ContactForm()
|
|
if form.validate_on_submit():
|
|
name = form.name.data
|
|
email = form.email.data
|
|
message = form.message.data
|
|
print(name)
|
|
print(email)
|
|
print(message)
|
|
# здесь логика базы данных
|
|
feedback = Feedback(name=name, email=email, message=message)
|
|
db.session.add(feedback)
|
|
db.session.commit()
|
|
|
|
print("\nData received. Now redirecting ...")
|
|
flash("Message Received", "success")
|
|
return redirect(url_for('.contact'))
|
|
|
|
return render_template('contact.html', form=form)
|