app.py

from flask import Flask, render_template, redirect
from flask.helpers import url_for
from forms import *
from flask_sqlalchemy import SQLAlchemy
from flask_bootstrap import Bootstrap
from flask_moment import Moment
from flask_login import LoginManager, current_user, login_required
from werkzeug.utils import secure_filename
import os
import uuid
from models import db,user,posts,items
from sqlalchemy.exc import SQLAlchemyError



IMAGE_UPLOAD_FOLDER = '/static/img'
EXTENSIONS = set(['jpg', 'jpeg'])

app = Flask(__name__)
application = app
app.config['SECRET_KEY'] = 'hard to guess string'
app.config['SQLALCHEMY_DATABASE_URI'] ='sqlite:///posts.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db.init_app(app)


from auth import auth as auth_blueprint
app.register_blueprint(auth_blueprint)

bootstrap = Bootstrap(app)
moment = Moment(app)
login_manager = LoginManager()
login_manager.login_view = 'auth.login'
login_manager.init_app(app)
with app.app_context():
    db.create_all()

@login_manager.user_loader
def load_user(user_id):
    return user.query.get(int(user_id))

@app.route('/update/<int:id>', methods=['GET', 'POST'])
def update(id):
    old_post = posts.query.get(id)
    update_form = UpdateForm(title=old_post.title, body=old_post.body,price=old_post.price,pic=old_post.pic)
        
    if update_form.validate_on_submit():

        old = posts.query.get_or_404(id)
        old.title = update_form.title.data
        old.body = update_form.body.data
        old.price = update_form.price.data
        pic = update_form.pic.data
        uuid_filename = str(uuid.uuid4())
        pic.save(os.path.join(
            app.root_path, 'static/img', uuid_filename
        ))
        old.pic = uuid_filename
        print(update_form.title.data)
        print(update_form.body.data)
        print(update_form.price.data)
        print(uuid_filename)

        try:
            db.session.commit()
            return redirect(url_for('index'))
        except:
            return "Update error"
    return render_template('update.html', update_form=update_form )

@app.route('/add_to_cart/<int:id>', methods=['GET'])
def add_to_cart(id):
    already_in = items.query.filter_by(post_id=id).all()
    if already_in:
        return redirect(url_for('index'))
    db_entry = items(post_id=id,email=current_user.email)
    print(db_entry)
    try:
        db.session.add(db_entry)
        db.session.commit()
        return redirect(url_for('index'))

    except SQLAlchemyError as e:
        error = str(e.__dict__['orig'])
        return error


@app.route('/delete_from_cart/<int:id>', methods=['GET'])
def delete_from_cart(id):
    deleted_post = items.query.filter_by(post_id=id).first()
    try:
        db.session.delete(deleted_post)
        db.session.commit()
        return redirect(url_for('index'))
    except SQLAlchemyError as e:
        error = str(e.__dict__['orig'])
        return error

def cart_list():
    return items.query.filter_by(email=current_user.email).all()


@app.route('/delete/<int:id>', methods=['GET', 'POST'])
def delete(id):

    deleted_post = posts.query.get_or_404(id)
    in_cart = items.query.filter_by(post_id=id).first()
    if in_cart:
        try:
            db.session.delete(in_cart)
        except:
            return "Delete error"
    try:
        db.session.delete(deleted_post)
        db.session.commit()
        return redirect(url_for('index'))
    except:
        return "Delete error"

@app.route('/', methods=['GET', 'POST'])
def index():
    if not current_user.is_authenticated:
         return render_template("home.html")

    title = body = price = pic = None
    post_form = PostForm()
    if post_form.validate_on_submit():
        title = post_form.title.data
        body = post_form.body.data
        price = post_form.price.data
        pic = post_form.pic.data
        uuid_filename = str(uuid.uuid4())
        pic.save(os.path.join(
            app.root_path, 'static/img', uuid_filename
        ))
        
        db_entry = posts(title = title, body = body, price = price, pic = uuid_filename, author_username=current_user.username, author_email=current_user.email)
        try:
            db.session.add(db_entry)
            db.session.commit()
        except:
            return "Database error"
    else :
        print(post_form.errors)
    items_list = [item.post_id for item in cart_list()]
    return render_template('index.html', post_form=post_form , posts_table=posts.query.order_by(posts.id.desc()).all(), current_user=current_user,cart_list =items_list, len_cart_list = len(items_list) )