application.py

from flask import Flask, render_template, request, flash, redirect, url_for, session
from database import DBhandler
import hashlib
import sys
import math
from flask import jsonify

application = Flask(__name__)
application.config["SECRET_KEY"] = "helloosp"

DB = DBhandler()

@application.route("/")
def hello():
    return render_template("index.html")
    #return redirect(url_for('view_list'))

@application.route("/list")
def view_list():
    page = request.args.get("page", 0, type=int)
    category = request.args.get("category", "all")
    per_page=6
    per_row=3
    row_count=int(per_page/per_row)
    start_idx=per_page*page
    end_idx=per_page*(page+1)
    
    if category == "all":
        data = DB.get_items()
    else:
        data = DB.get_items_bycategory(category)
    data = dict(sorted(data.items(), key=lambda x: x[0], reverse=False))
    item_counts = len(data)
    if item_counts <= per_page:
        data = dict(list(data.items())[:item_counts])
    else:
        data = dict(list(data.items())[start_idx:end_idx])
    tot_count = len(data)

    for i in range(row_count):
        if (i == row_count-1) and (tot_count%per_row != 0):
            locals()['data_{}'.format(i)] = dict(list(data.items())[i*per_row:])
        else:
            locals()['data_{}'.format(i)] = dict(list(data.items())[i*per_row:(i+1)*per_row])
    return render_template(
        "list.html",
        datas=data.items(),
        row1=locals()['data_0'].items(),
        row2=locals()['data_1'].items(),
        limit=per_page,
        page=page,
        page_count=int(math.ceil(item_counts/per_page)),
        total=item_counts,
        category=category
    )

@application.route("/reg_review_init/<name>/")
def reg_review_init(name):
    if 'id' not in session:
        return redirect(url_for('login'))
    return render_template("reg_reviews.html", name=name)

@application.route("/reg_review", methods=['POST'])
def reg_review():
    if 'id' not in session:
        return redirect(url_for('login'))
    data = request.form
    image_file = request.files["file"]
    img_path = "static/images/{}".format(image_file.filename)
    image_file.save(img_path)
    DB.reg_review(data, image_file.filename)
    return redirect(url_for('view_review'))

@application.route("/review")
def view_review():
    if 'id' not in session:
        return redirect(url_for('login'))
    
    page = request.args.get("page", 0, type = int)
    per_page = 6
    per_row = 3
    row_count = int(per_page/per_row)
    start_idx = per_page*page
    end_idx = per_page*(page+1)
    
    data = DB.get_reviews()
    item_counts = len(data)
    data = dict(list(data.items())[start_idx:end_idx])
    tot_count = len(data)
    
    for i in range(row_count):
        if(i==row_count-1) and (tot_count%per_row!=0):
            locals()['data_{}'.format(i)] = dict(list(data.items())[i*per_row:])
        else: 
            locals()['data_{}'.format(i)] = dict(list(data.items())[i*per_row:(i+1)*per_row])
    return render_template(
        "review.html",
        datas = data.items(),
        row1 = locals()['data_0'].items(),
        row2 = locals()['data_1'].items(),
        limit = per_page,
        page = page,
        page_count = int(item_counts/per_page+1),
        total = item_counts
    )

@application.route("/view_review_detail/<name>/")
def view_review_detail(name):
    if 'id' not in session:
        return redirect(url_for('login'))
    
    review_data = DB.get_review_byname(name)
    if review_data:
        return render_template("review_detail.html", data=review_data)
    else:
        return redirect(url_for('view_review'))

@application.route("/reg_items")
def reg_item():
    if 'id' not in session:
        return redirect(url_for('login'))
    
    return render_template("reg_items.html")

@application.route("/submit_item_post", methods=['POST'])
def reg_item_submit_post():
    if 'id' not in session:
        return redirect(url_for('login'))
    
    image_file = request.files["file"]
    image_file.save("static/images/{}".format(image_file.filename))
    data = request.form
    DB.insert_item(data['name'], data, image_file.filename)

    return render_template("submit_item_result.html", data=data, img_path=
                           "static/images/{}".format(image_file.filename))

@application.route("/submit_item")
def reg_item_submit():
    if 'id' not in session:
        return redirect(url_for('login'))
    
    name = request.args.get("name")
    seller = request.args.get("seller")
    addr = request.args.get("addr")
    email = request.args.get("email")
    category = request.args.get("category")
    card = request.args.get("card")
    status = request.args.get("status")
    phone = request.args.get("phone")

@application.route("/login")
def login():
    return render_template("login.html")

@application.route("/login_confirm", methods=['POST'])
def login_user():
    id_ = request.form['id']
    pw = request.form['pw']
    pw_hash = hashlib.sha256(pw.encode('utf-8')).hexdigest()
    
    if DB.find_user(id_, pw_hash):
        session['id'] = id_
        return redirect(url_for('hello'))
    else:
        flash("Wrong ID or PW!")
        return render_template("login.html")

@application.route("/signup")
def signup():
    return render_template("signup.html")

@application.route("/signup_post", methods=['POST'])
def register_user():
    data = request.form
    pw = data['pw']
    pw_hash = hashlib.sha256(pw.encode('utf-8')).hexdigest()

    if DB.insert_user(data, pw_hash):
        return render_template("login.html")
    else:
        flash("User ID already exist")
        return render_template("signup.html")
      

    #print(name,addr,phone,category,status)
    #return render_template("reg_item.html")
    
    
@application.route("/logout")
def logout_user():
    session.clear()
    return redirect(url_for('hello'))

@application.route("/view_detail/<name>/")
def view_item_detail(name):
    print("###name:",name)
    data = DB.get_item_byname(str(name))
    print("####data:",data)
    return render_template("detail.html", name=name, data=data)

@application.route('/show_heart/<name>/', methods=['GET'])
def show_heart(name):
    my_heart = DB.get_heart_byname(session['id'],name)
    return jsonify({'my_heart': my_heart})

@application.route('/like/<name>/', methods=['POST'])
def like(name):
    my_heart = DB.update_heart(session['id'],'Y',name)
    return jsonify({'msg': '좋아요 완료!'})

@application.route('/unlike/<name>/', methods=['POST'])
def unlike(name):
    my_heart = DB.update_heart(session['id'],'N',name)
    return jsonify({'msg': '안좋아요 완료!'})

@application.route("/search")
def search():
    query = request.args.get("query")
    all_items = DB.get_items()
    
    # Filter items based on item name or seller's ID
    filtered_items = {name: details for name, details in all_items.items() 
                      if query.lower() in name.lower() or query.lower() in details.get('seller', '').lower()}
    
    return render_template("search_result.html", items=filtered_items)



if __name__ == "__main__":
    application.run(host='0.0.0.0', debug=True)