Main.py

from flask import Flask, render_template, request, make_response, session, redirect, abort
from flask_restful import reqparse, abort, Api, Resource
from data.product_resourses import ProductResource, ProductListResource
from data import db_session
from data.users import User
from data.RegisterForm import RegisterForm
from data.LoginForm import LoginForm
from data.ReviewsForm import ReviewsForm
from data.OrderForm import OrderForm
from data.products import Product
from data.songs import Song
from data.Musicians import Musician
from data.carts import Cart
from data.cart_product import Cart_Product
from data.reviews import Review
from data.orders import Order
from flask_login import LoginManager, login_user, login_required, logout_user, current_user
from sqlalchemy import func
import product_api
import datetime
from flask_ngrok import run_with_ngrok


app = Flask(__name__)
app.config['PERMANENT_SESSION_LIFETIME'] = datetime.timedelta(days=365)
app.config['SECRET_KEY'] = 'yandexlyceum_secret_key'
api = Api(app)
login_manager = LoginManager()
login_manager.init_app(app)
run_with_ngrok(app)

def main():
    db_session.global_init("db/shop.db")
    api.add_resource(ProductListResource, '/api/products')
    api.add_resource(ProductResource, '/api/product/<int:product_id>')
    app.register_blueprint(product_api.blueprint)
    session = db_session.create_session()

    @app.route("/")
    def index():
        session = db_session.create_session()
        special = session.query(Product).get(1)
        products = session.query(Product).order_by(Product.views.desc())
        return render_template("index.html", products=products, special=special)

    @app.route("/TopSingles")
    def singles():
        session = db_session.create_session()
        special = False
        products = session.query(Product).filter(Product.is_lp == False).order_by(Product.views.desc())
        return render_template("index.html", products=products, special=special)

    @app.route("/TopAlbums")
    def albums():
        session = db_session.create_session()
        special = False
        products = session.query(Product).filter(Product.is_lp == True).order_by(Product.views.desc())
        return render_template("index.html", products=products, special=special)

    @app.route("/year/<int:year>")
    def year(year):
        session = db_session.create_session()
        special = False
        products = session.query(Product).filter(Product.year == year).order_by(Product.views.desc())
        return render_template("index.html", products=products, special=special)

    @app.route("/musician/<name>")
    def musician(name):
        name = name.replace('%20', ' ')
        session = db_session.create_session()
        musician = session.query(Musician).filter(Musician.name == name).first()
        special = False
        products = session.query(Product).filter(Product.musician_id == musician.id).order_by(Product.views.desc())
        return render_template("index.html", products=products, special=special)

    @app.route('/register', methods=['GET', 'POST'])
    def reqister():
        form = RegisterForm()
        if form.validate_on_submit():
            if form.password.data != form.password_again.data:
                return render_template('register.html', title='Registration',
                                       form=form,
                                       message="Passwords don't match")
            session = db_session.create_session()
            if session.query(User).filter(User.email == form.email.data).first():
                return render_template('register.html', title='Registration',
                                       form=form,
                                       message="This user already exists")
            user = User(
                name=form.name.data,
                surname=form.surname.data,
                email=form.email.data,
            )
            user.set_password(form.password.data)

            session.add(user)
            session.commit()
            return redirect('/login')
        return render_template('register.html', title='Registration', form=form)

    @app.route('/add_to_cart/<int:id>')
    @login_required
    def add_to_cart(id):
        product = session.query(Product).get(id)
        cart = session.query(Cart).filter(Cart.user_id == current_user.id).first()
        old_cart_product = session.query(Cart_Product).filter(Cart_Product.cart_id == cart.id, Cart_Product.product_id == product.id).first()
        if not old_cart_product:
            cart_product = Cart_Product(
                product_id=product.id,
                cart_id=cart.id,
                count=1,
                one_price=product.price,
                full_price=product.price
            )
            session.add(cart_product)
            session.commit()
            address = '/product/' + str(id)

        else:
            address = '/cart'
        return redirect(address)

    @app.route('/delete/<int:id>', methods=['GET', 'POST'])
    @login_required
    def delete(id):
        session = db_session.create_session()
        cart_products = session.query(Cart_Product).filter(Cart_Product.id == id).first()
        if cart_products:
            session.delete(cart_products)
            session.commit()
        else:
            abort(404)
        return redirect('/cart')

    @app.route('/cart')
    @login_required
    def cart():
        empty_cart = False
        cart = session.query(Cart).get(current_user.id)
        cart_products = session.query(Cart_Product).filter(Cart_Product.cart_id == cart.id).order_by(Cart_Product.id.desc())
        product_count = session.query(func.sum(Cart_Product.count)).filter_by(cart_id=cart.id).scalar()
        sum_price = session.query(func.sum(Cart_Product.full_price)).filter_by(cart_id=cart.id).scalar()
        sum_price = str(sum_price)
        full_price = sum_price[:(sum_price.find('.') + 3)]
        current_user_id = current_user.id
        if cart_products.count() == 0:
            empty_cart = True
        return render_template('cart.html', title='Cart', cart_products=cart_products,
                               product_count=product_count, full_price=full_price, empty_cart=empty_cart, current_user_id=current_user_id)

    @login_manager.user_loader
    def load_user(user_id):
        session = db_session.create_session()
        return session.query(User).get(user_id)

    @app.route('/login', methods=['GET', 'POST'])
    def login():
        form = LoginForm()
        if form.validate_on_submit():
            session = db_session.create_session()
            user = session.query(User).filter(User.email == form.email.data).first()
            if user and user.check_password(form.password.data):
                login_user(user, remember=form.remember_me.data)
                return redirect("/create_cart")
            return render_template('login.html',
                                   message="Wrong email or password",
                                   form=form)
        return render_template('login.html', title='Authorization', form=form)

    @app.route('/create_cart')
    def create_cart():
        old_cart = session.query(Cart).filter(Cart.user_id == current_user.id).first()
        if not old_cart:
            user_cart = Cart()
            user_cart.user_id = current_user.id
            session.add(user_cart)
            session.commit()
        return redirect("/")

    @app.route('/logout')
    @login_required
    def logout():
        logout_user()
        return redirect("/")

    @app.route('/changetheme')
    @login_required
    def change_theme():
        session = db_session.create_session()
        if current_user.dark_theme:
            session.query(User).filter(User.id == current_user.id).update({User.dark_theme: 0},
                                                                          synchronize_session=False)
        else:
            session.query(User).filter(User.id == current_user.id).update({User.dark_theme: 1},
                                                                          synchronize_session=False)
        session.commit()
        return redirect("/")

    @app.route('/changeform')
    @login_required
    def change_form():
        session = db_session.create_session()
        if current_user.circle_theme:
            session.query(User).filter(User.id == current_user.id).update({User.circle_theme: 0},
                                                                          synchronize_session=False)
        else:
            session.query(User).filter(User.id == current_user.id).update({User.circle_theme: 1},
                                                                          synchronize_session=False)
        session.commit()
        return redirect("/")

    @app.route("/count+/<int:id>")
    @login_required
    def plus_count(id):
        cart = session.query(Cart).filter(Cart.user_id == current_user.id).first()
        session.query(Cart_Product).filter(Cart_Product.id == id, Cart_Product.cart_id == cart.id).\
            update({Cart_Product.count: Cart_Product.count + 1,
                    Cart_Product.full_price: Cart_Product.one_price * (Cart_Product.count + 1)})
        c_p = session.query(Cart_Product).filter(Cart_Product.id == id).first()
        full_price = round(c_p.full_price, 2)
        session.query(Cart_Product).filter(Cart_Product.id == id, Cart_Product.cart_id == cart.id). \
            update({Cart_Product.full_price: full_price})
        session.commit()
        return redirect("/cart")

    @app.route("/count-/<int:id>")
    @login_required
    def minus_count(id):
        cart = session.query(Cart).filter(Cart.user_id == current_user.id).first()
        count = session.query(Cart_Product).get(id)
        if count.count > 1:
            session.query(Cart_Product).filter(Cart_Product.id == id, Cart_Product.cart_id == cart.id). \
                update({Cart_Product.count: Cart_Product.count - 1,
                        Cart_Product.full_price: Cart_Product.one_price * (Cart_Product.count - 1)})
            c_p = session.query(Cart_Product).filter(Cart_Product.id == id).first()
            full_price = round(c_p.full_price, 2)
            session.query(Cart_Product).filter(Cart_Product.id == id, Cart_Product.cart_id == cart.id). \
                update({Cart_Product.full_price: full_price})
            session.commit()
        return redirect("/cart")

    @app.route('/product/<int:id>', methods=['GET', 'POST'])
    def product_page(id):
        session.query(Product).filter(Product.id == id).update({Product.views: Product.views + 1})
        session.commit()
        form = ReviewsForm()
        product = session.query(Product).filter(Product.id == id).first()
        if form.validate_on_submit():
            review_session = db_session.create_session()
            review = Review()
            review.content = form.content.data
            review.product_id = id
            current_user.review.append(review)
            review_session.merge(current_user)
            review_session.commit()
        reviews = session.query(Review).filter(Review.product_id == id)
        if product.is_lp == 0:
            return render_template("product.html", product=product, form=form, reviews=reviews)
        else:
            songs = session.query(Song).filter(Song.album_id == product.id)
            songs_count = songs.count()
            return render_template("product.html", product=product, form=form, reviews=reviews, songs=songs, songs_count=songs_count)

    @app.route('/order/<delivery>', methods=['GET', 'POST'])
    @login_required
    def order_page(delivery):
        cart = session.query(Cart).filter(Cart.id == current_user.id).first()
        id = cart.id
        if delivery == 'home':
            delivery_price = 9.99
        elif delivery == 'sdek':
            delivery_price = 5.99
        elif delivery == 'boxberry':
            delivery_price = 4.99
        full_price = session.query(func.sum(Cart_Product.full_price)).filter_by(cart_id=id).scalar()
        form = OrderForm()
        if form.validate_on_submit():
            order_session = db_session.create_session()
            order = Order()
            order.cart_id = id
            order.surname = form.surname.data
            order.name = form.name.data
            order.phone = form.phone.data
            order.delivery = delivery
            order.delivery_price = delivery_price
            order.country = form.country.data
            order.town = form.town.data
            order.street = form.street.data
            order.house = form.house.data
            order.flat = form.flat.data
            order.promo = form.promo.data
            order_session.add(order)
            order_session.commit()
            return redirect("/all_delete")
        return render_template("order.html", form=form, cart_id=id,
                               full_price=full_price, delivery_price=delivery_price, delivery=delivery)

    @app.route('/review_delete/<int:id>', methods=['GET', 'POST'])
    @login_required
    def review_delete(id):
        session = db_session.create_session()
        review = session.query(Review).filter(Review.id == id, Review.user == current_user).first()
        if review:
            session.delete(review)
            session.commit()
        else:
            abort(404)
        return redirect('/')

    @app.route('/all_delete')
    @login_required
    def all_delete():
        cart = session.query(Cart).filter(Cart.id == current_user.id).first()
        cart_products = session.query(Cart_Product).filter(Cart_Product.cart_id == cart.id)
        for product in cart_products:
            session.delete(product)
            session.commit()
        return redirect('/thanks')

    @app.route('/thanks')
    @login_required
    def thanks():
        return render_template("thanks.html")


    app.run()


if __name__ == '__main__':
    main()