application.py

from flask import Flask, render_template , url_for, request, redirect , flash, session, jsonify
import hashlib
from bs4 import BeautifulSoup
import requests
from database import DBhandler
import re



app = Flask(__name__)
app.config["SECRET_KEY"]="helloosp"
DB = DBhandler()

@app.route("/product-add") 
def productAdd():
    return render_template('product_add.html')

@app.route("/")
def index():
    page = request.args.get("page", 0, type=int)
    category = request.args.get("category", None)
    per_page = 20  # 페이지 당 표시할 아이템 수
    per_row = 4  # 행 당 표시할 아이템 수
    
    data = DB.get_items(category=category)  # 카테고리에 따라 아이템 가져오기
    print(data)
    
    # 페이지네이션 로직
    item_counts = len(data)
    data = dict(list(data.items())[page * per_page:(page + 1) * per_page])
    row_data = [list(data.items())[i * per_row:(i + 1) * per_row] for i in range((len(data) + per_row - 1) // per_row)]

    return render_template("index.html", row_data=row_data, limit=per_page, page=page, page_count=int((item_counts + per_page - 1) / per_page), total=item_counts)

@app.route("/header")
def headerBefore():
    return render_template('layout/header.html')
@app.route("/header-only")
def headerAfter():
    return render_template('layout/header_only.html')
@app.route("/footer")
def footerEnter():
    return render_template('layout/footer.html')

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

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

@app.route("/signup2", methods=["GET", "POST"])
def signup2():
    if request.method == "POST":
        return redirect(url_for('signup1'))
    return render_template('signup2.html')

@app.route("/signup1_post", methods=['POST'])
def register_user():
    data = request.form
    pw = request.form['pw']
    pw_hash = hashlib.sha256(pw.encode('utf-8')).hexdigest()
    
    if DB.insert_user(data, pw_hash):
        return 'signup3' 
    else:
        return jsonify({'error': '이미 존재하는 아이디입니다!'}), 400
    
@app.route('/login')
def login():
    return render_template('login.html')
    
@app.route("/login_confirm", methods=['POST'])
def login_user():
    print("실행됨")
    id_=request.form['id']
    pw=request.form['password']
    pw_hash = hashlib.sha256(pw.encode('utf-8')).hexdigest()
    if DB.find_user(id_,pw_hash):
        session['id']=id_
        return redirect(url_for('index'))
    else:
        flash("존재하지 않는 정보입니다! 다시 로그인을 시도해주세요.")
        return redirect(url_for('index'))
    
@app.route("/logout")
def logout_user():
    session.clear()
    return redirect(url_for('index'))



# 크롤링 코드
@app.route("/process-url", methods=['POST'])
def process_url():
    data = request.get_json()
    url = data['url']
    
    response = requests.get(url)
    response.raise_for_status()

    soup = BeautifulSoup(response.text, 'html.parser')

    title = soup.find('h3', class_='_22kNQuEXmb _copyable').text
    price = soup.find('span', class_='_1LY7DqCnwR').text
    delivery = soup.find('em', class_='_3tOxM3n343').text

    img_element = soup.find('img', class_='_2RYeHZAP_4')

    image_url = None
    
    if img_element:
    # src 속성 추출
        image_url = img_element['src']
        print("Extracted Image URL:", image_url)
    else:
        print("Image tag with specified class not found.")

    return {
        'title': title,
        'price': price,
        'delivery': delivery,
        'image_url': image_url,
        'url' : url
    }

@app.route("/add-product-post", methods=["POST"])
def registerproduct():

    hidden_title = request.form.get("hidden-title")
    hidden_price = request.form.get("hidden-price")
    hidden_delivery = request.form.get("hidden-delivery")
    hidden_url = request.form.get("hidden-url")
    hidden_image_url = request.form.get("hidden-image-url")
    user_id = session.get('id', 'default-user-id')
    data = {
        "product_description": request.form.get("product-description"),
        "product_number": request.form.get("product-number"),
        "product_category": request.form.get("product-category"),
        "start_date": request.form.get("start-date"),
        "end_date": request.form.get("end-date"),
        "people_number": request.form.get("people-number"),
        "title": hidden_title,
        "price": hidden_price,
        "delivery": hidden_delivery,
        "url": hidden_url,
        "image_url": hidden_image_url,
        "user_id": user_id
    }
    print(data)
    DB.insert_item(data)
    return redirect(url_for('mypage'))

@app.route("/products-list")
def view_list():
    page = request.args.get("page", 0, type=int)
    category = request.args.get("category", None)
    per_page = 6  # 페이지 당 표시할 아이템 수
    per_row = 3  # 행 당 표시할 아이템 수
    
    data = DB.get_items(category=category)  # 카테고리에 따라 아이템 가져오기
    print(data)
    
    # 페이지네이션 로직
    item_counts = len(data)
    data = dict(list(data.items())[page * per_page:(page + 1) * per_page])
    row_data = [list(data.items())[i * per_row:(i + 1) * per_row] for i in range((len(data) + per_row - 1) // per_row)]

    return render_template("product_list.html", row_data=row_data, limit=per_page, page=page, page_count=int((item_counts + per_page - 1) / per_page), total=item_counts)

@app.route('/mypage')
def mypage():
    page = request.args.get("page", 0, type=int)
    user_id = session['id']
    user_info = DB.get_user_info(user_id)
    data = DB.get_my_items(user_id=user_id)
    per_page = 6  # 페이지 당 표시할 아이템 수
    per_row = 3  # 행 당 표시할 아이템 수
    item_counts = len(data)
    data = dict(list(data.items())[page * per_page:(page + 1) * per_page])
    row_data = [list(data.items())[i * per_row:(i + 1) * per_row] for i in range((len(data) + per_row - 1) // per_row)]
    
    return render_template('mypage.html', user_info=user_info, row_data=row_data, limit=per_page, page=page, page_count=int((item_counts + per_page - 1) / per_page), total=item_counts)


@app.route("/parti-product")
def partiProduct():
    user_id = session['id']
    user_info = DB.get_user_info(user_id)
    data = DB.get_purchase_details(user_id=user_id)

    per_row = 5  # 행 당 표시할 아이템 수
    item_counts = len(data)
    row_data = [list(data.items())[i * per_row:(i + 1) * per_row] for i in range((len(data) + per_row - 1) // per_row)]

    return render_template("parti_product.html", user_info=user_info, row_data=row_data, total=item_counts)




@app.route("/all-reivew-page")
def all_review_check():
    return render_template('all_review_check.html')

@app.route("/view_detail/<name>/")
def view_item_detail(name):
        data = DB.get_item_byname(str(name))
        price = int(data['price'].replace(',', ''))
        people_number = int(data['people_number'])

        per_person_price = price / people_number

        return render_template("product_detail.html", name=name, data=data, per_person_price=per_person_price)

@app.route("/myparticipation", methods=["GET"])  
def my_participate(): 
    if 'id' in session:
        user_id = session['id']
        name = request.args.get('name')

        DB.insert_product_for_user(user_id, name)
        
        return redirect(url_for('partiProduct'))
    
@app.route("/reg_review_init/<name>/")
def reg_review_init(name):
    data = DB.get_item_byname(str(name))
    return render_template("review_add.html", name=name, data=data)

@app.route("/reg_review", methods=['POST'])
def reg_review():
    hidden_id = request.form.get("seller-id")
    key_name = request.form.get("key")
    writer_id = session['id']
    print(request.files)
    image_file = request.files.get("img_path")
    image_file.save("static/img/{}".format(image_file.filename))
    
    # 'reviewStar' 키가 없을 경우 기본값으로 '0' 사용
    rate = request.form.get('reviewStar', '0')

    data = {
        'key_name' : key_name,
        'name': request.form['name'],
        'title': request.form['title'],
        'rate': rate,
        'review': request.form['reviewContents'],
        "img_path": "static/img/" + image_file.filename,
        'seller_id' : hidden_id,
        'writer_id': writer_id
    }
    DB.reg_review(data, image_file.filename)
    return redirect(url_for('index'))

@app.route("/review")
def view_review():
    page = request.args.get("page", 0, type=int)
    per_page=5
    per_row=5
    row_count=int(per_page/per_row)
    start_idx=per_page*page
    end_idx=per_page*(page+1)
    data = DB.get_reviews() #read the table
    print(data)
    item_counts = len(data)
    data = dict(list(data.items())[start_idx:end_idx])
    tot_count = len(data)
    row_data = [list(data.items())[i * per_row:(i + 1) * per_row] for i in range(per_page // per_row)]
    return render_template(
        "all_review_check.html",
        row_data=row_data, limit=per_page,page=page, page_count=int((item_counts/per_page)+1),total=item_counts)

@app.route('/my-reviews')
def my_reviews():
    if 'id' not in session:
        flash("로그인 후 이용 가능합니다.")
        return redirect(url_for('index'))
    user_id = session['id']
    user_info = DB.get_user_info(user_id)
    page = request.args.get("page", 0, type=int)
    per_page=5
    per_row=5
    row_count=int(per_page/per_row)
    start_idx=per_page*page
    end_idx=per_page*(page+1)

    user_id = session['id']
    data = DB.get_user_reviews(user_id)
    item_counts = len(data)
    data = dict(list(data.items())[start_idx:end_idx])
    tot_count = len(data)
    row_data = [list(data.items())[i * per_row:(i + 1) * per_row] for i in range(per_page // per_row)]
    return render_template('my_reviews.html', user_info=user_info, row_data=row_data, limit=per_page,page=page, page_count=int((item_counts/per_page)+1),total=item_counts)

@app.route('/written-reviews')
def written_reviews():
    if 'id' not in session:
        flash("로그인 후 이용 가능합니다.")
        return redirect(url_for('index'))
    
    user_id = session['id']
    user_info = DB.get_user_info(user_id)
    page = request.args.get("page", 0, type=int)
    per_page=5
    per_row=5
    row_count=int(per_page/per_row)
    start_idx=per_page*page
    end_idx=per_page*(page+1)

    user_id = session['id']
    data = DB.get_written_reviews(user_id)
    item_counts = len(data)
    data = dict(list(data.items())[start_idx:end_idx])
    tot_count = len(data)
    row_data = [list(data.items())[i * per_row:(i + 1) * per_row] for i in range(per_page // per_row)]
    return render_template('written_reviews.html', user_info=user_info, row_data=row_data, limit=per_page,page=page, page_count=int((item_counts/per_page)+1),total=item_counts)


@app.route("/wishlist")
def wishlist():
    page = request.args.get("page", 0, type=int)
    per_page = 10
    per_row = 5

    data = DB.get_wish_product_list_byuser(session['id'])

    if not data:
        return render_template("wish_list.html", row_data=[], limit=per_page, page=page, page_count=0, total=0)

    data_list = list(data.items())
    products_count = len(data_list)

    start_idx = per_page * page
    end_idx = per_page * (page + 1)
    paginated_data = dict(data_list[start_idx:end_idx])

    row_data = [list(paginated_data.items())[i * per_row:(i + 1) * per_row] for i in range(per_page // per_row)]
    return render_template("wish_list.html", row_data=row_data, limit=per_page, page=page, page_count=int((products_count / per_page) + 1), total=products_count)

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


@app.route('/like/<name>/', methods=['POST'])
def like(name):
    my_heart = DB.update_heart(session['id'],'Y',name)
    return jsonify({'msg': '위시 상품에 등록되었습니다!'})

@app.route('/unlike/<name>/', methods=['POST'])
def unlike(name):
    my_heart = DB.update_heart(session['id'],'N',name)
    return jsonify({'msg': '위시 상품에서 제외되었습니다.'})



if __name__ == '__main__':
    app.run('0.0.0.0', port=5002, debug=True)