app.py

from pymongo import MongoClient
import jwt
import datetime
import hashlib
from flask import Flask, render_template, jsonify, request, redirect, url_for
from werkzeug.utils import secure_filename
from datetime import datetime, timedelta

import requests
from bs4 import BeautifulSoup

app = Flask(__name__)
app.config["TEMPLATES_AUTO_RELOAD"] = True
app.config['UPLOAD_FOLDER'] = "./static/profile_pics"

SECRET_KEY = 'STUDYWITH'

#client = MongoClient('mongodb://test:test@localhost', 27017)
client = MongoClient('localhost', 27017)
db = client.DO_STUDY

# autoIncrement 를  위해 서버가 꺼져도 변하지 않는 디비에 리뷰의 number 값을 저장
if (db.number.find_one({"id": 'autoIncrement'}) is None):
    db.number.insert_one({"number": 1, "id": 'autoIncrement'})




@app.route('/sign_up/check_dup', methods=['POST'])
def check_dup():
    # 유저 닉네임 중복확인
    username_receive = request.form['username_give']
    exists = bool(db.users.find_one({"username": username_receive}))
    return jsonify({'result': 'success', 'exists': exists})


@app.route('/')
def home():
    # /로 들어왔을 때 조건에 따라 구현
    review_list = list(db.reviews.find({}, {'_id': False}))
    return render_template('reviews.html', review_list=review_list)



@app.route('/guest')
def guest():
    # 게스트로 입장한다면 -> 기능 구현 필요 def home()에 추가해야함
    review_list = list(db.reviews.find({}))
    return render_template("reviews.html", review_list=review_list)


@app.route('/login')
def login():
    # 로그인 페이지로 들어왔을때 렌더링
    # 쿼리에 올라가 있는 msg 값
    msg = request.args.get("msg")
    return render_template('login.html', msg=msg)


@app.route('/sign_up/save', methods=['POST'])
def sign_up():
    # 유저객체를 생성했으면 더 쉽게 구현가능
    # 회원 가입 기능 구현 ajax로 아이디 패스워드를 받아서 패스워드를 해시화한 후 db에 저장
    username_receive = request.form['username_give']
    password_receive = request.form['password_give']
    password_hash = hashlib.sha256(password_receive.encode('utf-8')).hexdigest()
    doc = {
        "username": username_receive,  # 아이디
        "password": password_hash,  # 비밀번호
        "like_list": [],  # 내가 좋아요누른 리스트
        "review_list": [],  # 내가 리뷰한 리스트
    }
    db.users.insert_one(doc)
    return jsonify({'result': 'success'})


@app.route('/sign_in', methods=['POST'])
def sign_in():
    # 로그인 기능 구현
    username_receive = request.form['username_give']
    password_receive = request.form['password_give']

    pw_hash = hashlib.sha256(password_receive.encode('utf-8')).hexdigest()
    result = db.users.find_one({'username': username_receive, 'password': pw_hash})
    # 로그인과 동시에 토큰 발급 캐시로 포함되어 넘어감

    # db에 아이디와 패스워드가 맞아서 result 값이 있다면
    if result is not None:
        payload = {
            'id': username_receive,
            'exp': datetime.utcnow() + timedelta(seconds=60 * 60 * 24)  # 로그인 24시간 유지
        }
        token = jwt.encode(payload, SECRET_KEY, algorithm='HS256').decode('utf-8')

        return jsonify({'result': 'success', 'token': token})
    # 찾지 못하면
    else:
        return jsonify({'result': 'fail', 'msg': '아이디/비밀번호가 일치하지 않습니다.'})


@app.route('/', methods=['POST'])
def saving():
    # 리뷰글을 누가 작성했는지 유저 디비에 저장
    special_number = db.number.find_one({'id': 'autoIncrement'})['number']
    token_receive = request.cookies.get('mytoken')
    payload = jwt.decode(token_receive, SECRET_KEY, algorithms=['HS256'])
    review_list = list(db.users.find_one({'username': payload['id']})['review_list'])
    review_list.append(special_number);
    # set을 통해 중복 제거
    my_set = set(review_list)
    review_list = list(my_set)
    db.users.update_one({'username': payload['id']}, {'$set': {'review_list': review_list}})

    # 크롤링 한 데이터를 리뷰 디비에 저장
    # 예지님 구현
    url_receive = request.form['url_give']
    comment_receive = request.form['comment_give']
    studyOption_receive = request.form['studyOption_give']
    headers = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64)AppleWebKit/537.36 (KHTML, like Gecko) Chrome/73.0.3683.86 Safari/537.36'}

    # 리뷰라는 객체를 만들었으면 db에 쌓이는게 훨씬 간단하게 들어갔을 것임
    data = requests.get(url_receive, headers=headers)
    soup = BeautifulSoup(data.text, 'html.parser')
    title = soup.select_one('meta[property="og:title"]')['content']
    url = soup.select_one('meta[property="og:url"]')['content']
    image = soup.select_one('meta[property="og:image"]')['content']
    desc = soup.select_one('meta[property="og:description"]')['content']
    doc = {'title': title, 'desc': desc, 'image': image, 'url': url, 'comment': comment_receive,
           'studyOption': studyOption_receive, 'like': 0, 'special_number': special_number}
    db.reviews.insert_one(doc)
    # Auto Increment
    special_number += 1
    db.number.update_one({'id': 'autoIncrement'}, {'$set': {'number': special_number}})
    return redirect('/')

# @app.route('/reviews', methods=['POST'])
# def saving():
#     url_receive = request.form['url_give']
#     comment_receive = request.form['comment_give']
#     type_receive = request.form['type_give']
#
#     headers = {
#         'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64)AppleWebKit/537.36 (KHTML, like Gecko) Chrome/73.0.3683.86 Safari/537.36'}
#     data = requests.get(url_receive, headers=headers)
#
#     soup = BeautifulSoup(data.text, 'html.parser')
#
#     title = soup.select_one('meta[property="og:title"]')['content']
#     icon = soup.select_one('meta[property="og:image"]')['content']
#     desc = soup.select_one('meta[property="og:description"]')['content']
#
#     doc = {
#         'title': title,
#         'icon': icon,
#         'desc': desc,
#         'url':url_receive,
#         'comment':comment_receive,
#         'type':type_receive
#     }
#     db.study.insert_one(doc)
#
#     return jsonify({'msg':'저장'})


@app.route('/mypage')
def user():
    # 각 사용자의 리뷰글 추천글 회원 정보를 볼 수 있는 페이지
    # 강욱님 구현
    token_receive = request.cookies.get('mytoken')
    try:
        payload = jwt.decode(token_receive, SECRET_KEY, algorithms=['HS256'])
        user_info = db.users.find_one({"username": payload["id"]})
        like_num_set = user_info['like_list']
        review_num_set = user_info['review_list']
        like_list = []
        review_list = []
        for num in like_num_set:
            like_list.append(db.reviews.find_one({'special_number': num}))
        for num in review_num_set:
            review_list.append(db.reviews.find_one({'special_number' : num}))

        return render_template('mypage.html', user_info=user_info, like_list=like_list, review_list=review_list)
    except (jwt.ExpiredSignatureError, jwt.exceptions.DecodeError):
        return redirect(url_for("home"))


@app.route('/reviews/like', methods=['POST'])
def update_like():
    token_receive = request.cookies.get('mytoken')
    try:
        payload = jwt.decode(token_receive, SECRET_KEY, algorithms=['HS256'])
        number_receive = int(request.form['number_give'])
        # 좋아요 기능 구현
        review = db.reviews.find_one({'special_number': number_receive})
        like = review['like'] + 1
        db.reviews.update_one({'special_number': number_receive}, {'$set': {'like': like}})
        # 유저디비 좋아요 리스트에 리뷰 넣기
        like_list = list(db.users.find_one({'username': payload['id']})['like_list'])
        like_list.append(number_receive);
        my_set = set(like_list)
        like_list = list(my_set)
        db.users.update_one({'username': payload['id']}, {'$set': {'like_list': like_list}})
        return jsonify({"result": "success", 'msg': '좋아요!'})
    except (jwt.ExpiredSignatureError, jwt.exceptions.DecodeError):
        return redirect(url_for("home"))


@app.route('/reviews/section', methods=["GET"])
def review_filter():
    # 쿼리로 데이터 값 받아와서 옵션으로 리스트 분류화
    option = request.args.get('option')
    print(option)
    if (option == 'All'):
        selected_reviews = list(db.reviews.find({}))
    else:
        selected_reviews = list(db.reviews.find({"studyOption": option}))
    for selected_review in selected_reviews:
        print(selected_review)
    return render_template('reviews.html', review_list=selected_reviews)



@app.route('/reviews')
def search():
    # 쿼리로 데이터 값 받아와서 리스트 검색
    search = request.args.get('search')
    reviews = list(db.reviews.find({}, {'_id': False}))
    review_list = []
    for review in reviews:
        if (search in review['title'] or search in review['comment'] or search in review['desc']):
            review_list.append(review)
    return render_template('reviews.html', review_list=review_list)


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