database.py

import pyrebase
import json
from datetime import datetime

class DBhandler:
    def __init__(self) :
        with open('./authentication/firebase_auth.json') as f:
            config=json.load(f)

        firebase = pyrebase.initialize_app(config)
        self.db = firebase.database()

    def restaurant_duplicate_check(self, name):
        restaurants = self.db.child("restaurant").get()
        for res in restaurants.each():
            value = res.val()
            if value['맛집이름'] == name:
                return False
            return True 

    def menu_duplicate_check(self, rest, name):
        menus = self.db.child("menu").get()
        for men in menus.each():
            value = men.val()
            if value['맛집이름'] == rest:
                if value['메뉴이름'] == name:
                    return False
            return True 

    #맛집등록
    def add_restaurant(self, name, data, img_path):
        restaurant_info ={
            "맛집이름":name,
            "음식종류":data.getlist('음식종류'),
            "vegan":data['vegan'],
            "pricerange":data['pricerange'],
            "location":data['location'],
            "address":data['address'],
            "phonenum":data['phonenum'],
            "othersite":data['othersite'],
            "parking":data['parking'],
            "mondaytime":data['mondaytime'],
            "tuesdaytime":data['tuesdaytime'],
            "wednesdaytime":data['wednesdaytime'],
            "thursdaytime":data['thursdaytime'],
            "fridaytime":data['fridaytime'],
            "saturdaytime":data['saturdaytime'],
            "sundaytime":data['sundaytime'],
            "isBreaktime":data['isBreaktime'],
            "breakday":data.getlist('breakday'),

            "평점":0,
            "likes":0,
            "theme":0,
            "timestamp":datetime.now().strftime('%Y-%m-%d %H:%M:%S'),

            "img_path":img_path
        } 
        print(restaurant_info)
        if self.restaurant_duplicate_check(name):
            self.db.child("restaurant").push(restaurant_info)
            print(data,img_path)
            return True
        else : 
            return False
    
    #댓글등록
    def add_comment(self, restaurant, data, id, nickname):
        comment_info ={
            "맛집이름":restaurant,
            "댓글내용":data['comment'],
            "writer":id,
            "작성자":nickname,
        } 
        print(comment_info)
        if self.db.child("comment").push(comment_info):
            return True
        else : 
            return False

    #리뷰등록
    def add_review(self, name, data, id, nickname, img_path):
        review_info ={
            "맛집이름":name,
            "메뉴이름":data['메뉴이름'],
            "평점":int(data['평점']),
            "timePeriod":data['timePeriod'],
            "waiting":data['waiting'],
            "리뷰작성내용":data['리뷰작성내용'],
            "img_path":img_path,
            "writer":id,
            "작성자":nickname,
            "timestamp":datetime.now().strftime('%Y-%m-%d'),
        }
        print(review_info)
        
        restaurants = self.db.child("restaurant").get()
        for res in restaurants.each():
            value = res.val()
            if value['맛집이름'] == name:
                key = res.key()
                reviews=DBhandler.get_reviews(self, name)
                total=len(reviews)
                num=float((int(data['평점'])+total*float(value['평점']))/(total+1))
                print(num)
                self.db.child("restaurant").child(key).update({"평점": float((int(data['평점'])+total*float(value['평점']))/(total+1))})

        self.db.child("review").push(review_info)
        print(data,img_path)
        return True

    #대표메뉴등록
    def add_menu(self, name, data, img_path):
        print(name)
        menu_info ={
            "맛집이름":name,
            "메뉴이름":data['메뉴이름'],
            "img_path":img_path,
            "가격":data['가격'],
            "allergyinfo":data['allergyinfo'],
            "vegan":data['vegan'],
            "etcinfo":data['etcinfo'],
            "한줄소개":data['한줄소개'],
        }
        print(menu_info)
        if self.menu_duplicate_check(name, data['메뉴이름']):
            self.db.child("menu").push(menu_info)
            print(data,img_path)
            return True
        else : 
            return False

    #레스토랑 테이블 가져오기
    def get_restaurants(self ):
        restaurants = self.db.child("restaurant").get().val()
        return restaurants

    #맛집이름으로 맛집 정보 가져오기
    def get_restaurant_byname(self, name):
        restaurants = self.db.child("restaurant").get()
        target_value=""
        for res in restaurants.each():
            value = res.val()
            if value['맛집이름'] == name:
                target_value=value
        return target_value

    #댓글 테이블 가져오기
    def get_comments(self, name):
        comments = self.db.child("comment").get()
        target_value={}
        for com in comments.each():
            value = com.val()
            if value['맛집이름'] == name:
                writer=value['작성자']
                target_value[writer]=dict((list(value.items())))
        print(target_value)
        return target_value

    #조건별 맛집 가져오기
    def get_restaurants_bycondition(self, location, foodtype, search, theme):
        if search : #검색 기능
            restaurants = self.db.child("restaurant").get()
            target_value=[]
            for res in restaurants.each():
                value = res.val()
                if value['맛집이름'] == search:
                    target_value.append(value)
                print("######target_value",target_value)
                new_dict={}
                for k,v in enumerate(target_value):
                    new_dict[k]=v
            return new_dict
        
        elif theme>0:
            restaurants = self.db.child("restaurant").get()
            target_value=[]
            for res in restaurants.each():
                value = res.val()
                if value['theme'] == theme:
                    if value['location'] == location or location=="all" :
                        if foodtype=="all":
                            target_value.append(value)
                        else :
                            for food in value['음식종류']:
                                if food == foodtype :
                                    target_value.append(value)
                print("######target_value",target_value)
                new_dict={}
                for k,v in enumerate(target_value):
                    new_dict[k]=v
            return new_dict

        elif location=="all" and foodtype=="all" : #전체 조회
            restaurants = self.db.child("restaurant").get().val()
            return restaurants
        
        elif location or foodtype :
            restaurants = self.db.child("restaurant").get()
            target_value=[]
            for res in restaurants.each():
                value = res.val()
                if value['location'] == location or location=="all" :
                    if foodtype=="all":
                        target_value.append(value)
                    else :
                        for food in value['음식종류']:
                            if food == foodtype :
                                target_value.append(value)
                new_dict={}
                for k,v in enumerate(target_value):
                    new_dict[k]=v
            return new_dict

        else :
            restaurants = self.db.child("restaurant").get().val()
            return restaurants

    #리뷰 테이블 가져오기
    def get_reviews(self, name):
        reviews = self.db.child("review").get()
        target_value={}
        for rev in reviews.each():
            value = rev.val()
            if value['맛집이름'] == name:
                writer=value['img_path']
                target_value[writer]=dict((list(value.items())))
        print(target_value)
        return target_value

    #메뉴 테이블 가져오기
    def get_menus(self, name):
        menus = self.db.child("menu").get()
        target_value={}
        for men in menus.each():
            value = men.val()
            if value['맛집이름'] == name:
                menu=value['메뉴이름']
                target_value[menu]=dict((list(value.items())))
        print(target_value)
        return target_value

    #유저 테이블 가져오기
    def get_users(self, id):
        users = self.db.child("user").get()
        target_value={}
        for use in users.each():
            value = use.val()
            if value['id'] == id:
                user=value['nickname']
                target_value[user]=dict((list(value.items())))
        print(target_value)
        return target_value

    #찜 기능
    def my_fav_list(self, name, id):
        users = self.db.child("user").get()
        for use in users.each():
            value=use.val()
            if value['id'] == id:
                key = use.key()
                if "isFavorite" in value:
                    for favs in value['isFavorite']:
                        fav=value['isFavorite'][favs]
                        key = use.key()
                        if name==fav:
                            self.db.child("user").child(key).child("isFavorite").child(favs).remove()
                            return True
                    key = use.key()
                    self.db.child("user").child(key).child("isFavorite").push(name)
                    return True
                else :
                    self.db.child("user").child(key).update({"isFavorite":{"FirstZzim":name}})
                    return True

    #찜 여부 확인하기
    def check_my_fav_list(self, name, id):
        users = self.db.child("user").get()
        for use in users.each():
            user = use.val()
            if user['id']==id:
                if "isFavorite" in user:
                    for favs in user['isFavorite']:
                        fav=user['isFavorite'][favs]
                        if name == fav:
                            return True
        return False

    def check_my_fav_lists(self, names, id):
        users = self.db.child("user").get()
        print(names)
        n=len(names)
        print(n)
        target_value={}
        for use in users.each():
            user = use.val()
            if user['id']==id:
                if "isFavorite" in user:
                    for favs in user['isFavorite']:
                        fav=user['isFavorite'][favs]
                        for key, val in names.items():
                            restaurant=names[key]['맛집이름']
                            if restaurant == fav:
                                print(restaurant)
                                target_value.update({fav: True})
                    for key, val in names.items():
                        if (names[key]['맛집이름'] in target_value)==False:
                            target_value.update({names[key]['맛집이름']: False})
                    print(target_value)
                    return target_value
        for key, val in names.items():
            target_value.update({names[key]['맛집이름']: False})
        print(target_value)
        return target_value

    #찜목록 가져오기
    def get_my_fav_list(self, id, location, foodtype):
        users = self.db.child("user").get()
        restaurants = self.db.child("restaurant").get()
        target_value=[]
        for use in users.each():
            user = use.val()
            if user['id']==id:
                if "isFavorite" in user:
                    for favs in user['isFavorite']:
                        fav=user['isFavorite'][favs]
                        for res in restaurants.each():
                            value = res.val()
                            if value['맛집이름'] == fav:
                                if value['location'] == location or location=="all" :
                                    if foodtype=="all":
                                        target_value.append(value)
                                    else :
                                        for food in value['음식종류']:
                                            if food == foodtype :
                                                target_value.append(value)
                else:
                    return False
        new_dict={}
        for k,v in enumerate(target_value):
            new_dict[k]=v
        return new_dict

    #좋아요 수
    def like_num(self, name):
        restaurant = self.db.child("restaurant").get()
        for res in restaurant.each():
            value=res.val()
            if value['맛집이름'] == name:
                key = res.key()
                num=int(value['likes'])+1
                print(num)
                self.db.child("restaurant").child(key).update({"likes": int(value['likes'])+1})
                return True

    #내가 쓴 리뷰 가져오기
    def get_myreviews(self, id):
        reviews = self.db.child("review").get()
        target_value={}
        for rev in reviews.each():
            value = rev.val()
            if value['writer'] == id:
                key=value['리뷰작성내용']
                target_value[key]=dict((list(value.items())))
        print(target_value)
        return target_value

    #회원가입
    def insert_user(self, data, pw):
        user_info={
            "id":data['id'],
            "pw":pw,
            "nickname":data['nickname'],
        }
        if self.user_duplicate_check(str(data['id'])):
            self.db.child("user").push(user_info)
            print(data)
            return True
        else:
            return False

    def user_duplicate_check(self, id_string):
        users=self.db.child("user").get()
        print("users###", users.val())
        if str(users.val()) == "None":
            return True
        else:
            for res in users.each():
                value=res.val()
                if value['id']==id_string:
                    return False
            return True

    #로그인
    def find_user(self, id, pw):
        users = self.db.child("user").get()
        for res in users.each():
            value = res.val()
            if value['id'] == id and value['pw'] == pw:
                return True
        return False

    def get_nickname(self, id, pw):
        users = self.db.child("user").get()
        for res in users.each():
            value = res.val()
            if value['id'] == id and value['pw'] == pw:
                return value['nickname']

    #탈퇴하기
    def withdrawl(self, id, pw):
        users = self.db.child("user").get()
        for res in users.each():
            value = res.val()
            key = res.key()
            if value['id'] == id and value['pw'] == pw:
                self.db.child("user").child(key).remove()
                return True
        return False