database.py

import json
import pyrebase

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()

    # User
    
    def insert_user(self, data, password):
        phone = data.get('phone', None)

        user_info = {
            "id": data['id'],
            "password": password,
            "email": data['email'],
            "phone": phone  # "phone" 키가 없으면 None으로 설정
        }
        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": # first registration
            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() 
        target_value=[]
        for res in users.each():
            value = res.val()
            
            if value['id'] == id_ and value['password'] == pw_:
                return True 
            
        return False

    
    def get_user_info(self, id_):
        users = self.db.child("user").get()
        for res in users.each():
            value = res.val()
            if value['id'] == id_:
                user_info = {
                    "id": value['id'],
                    "email": value['email'],
                    "phone": value['phone']
                }
                print(user_info)
                return user_info
        
        return False

    # Item
    
    def insert_item(self, name, data):
        item_info ={
            "sellerId": data['seller-id'],
            "productName": data['product-name'],
            "productPrice": data['product-price'],
            "product-status": data['product-status'],
            "description": data['product-description'],
            "product-image": data['img_path'],
            "location": data['location'],
            "timestamp": datetime.now().timestamp()  # 등록 시간을 timestamp로 추가
        }
        self.db.child("item").child(name).set(item_info)
        print(data)
        return True


    def get_all_items(self):
        items = self.db.child("item").get()
        result = []

        if items.val():
            for item in items.each():
                if 'timestamp' in item.val():
                    timestamp = item.val()['timestamp']
                    dt = datetime.fromtimestamp(timestamp)
                    item.val()['datetime'] = dt.strftime("%Y-%m-%d %H:%M:%S")
                    result.append(item.val())

        sorted_result = sorted(result, key=lambda x: x['datetime'], reverse=True)
        return sorted_result


    def get_item_byname(self, name): 
        items = self.db.child("item").get() 
        target_value="" 
        print("###########",name)
        for res in items.each(): 
            key_value = res.key()
            if key_value == name: 
                target_value=res.val()
        return target_value
    
    def get_heart_byname(self, uid, name):
        hearts = self.db.child("heart").child(uid).get()
        target_value=""
        if hearts.val() == None:
            return target_value

        for res in hearts.each():
            key_value = res.key()

        if key_value == name:
            target_value=res.val()
        return target_value
    
    def update_heart(self, user_id, isHeart, item):
        heart_info ={
             "interested": isHeart
        }
        self.db.child("heart").child(user_id).child(item).set(heart_info)
        return True

    def buy_item(self, user_id, item):
        purchase_info = {
            "sold": "Y"
        }
        self.db.child("purchase").child(user_id).child(item).set(purchase_info)
    
    # Review
    
    def reg_review(self, data):
        review_info = {
            "productName": data['productName'],
            "title": data['title'],
            "point": data['point'],
            "content": data['content'],
            "img_path": data['img_path'],
            "authorId": data['authorId']
        }
        self.db.child("review").child(data['productName']).set(review_info)
        return True
      
    def get_all_reviews(self ):
        reviews = self.db.child("review").get().val()
        return reviews
    
    def get_review_byname(self, name): 
        reviews = self.db.child("review").get() 
        target_value="" 
        print("###########",name)
        for res in reviews.each():
            key_value = res.key()
            if key_value == name: 
                target_value=res.val()
                break

        return target_value
    
    # 등록내역 조회
    def get_users_registered_item(self, seller):
        items = self.db.child("item").get()
        result = []
        length = 0
        print("###########", seller)
        for res in items.each(): 
            item = res.val()
            if item['sellerId'] == seller: 
                result.append(item)
                length += 1
            if length >= 2:
                break
        return result
    
    # 회원탈퇴
    def withdraw_user(self, id_):
        users = self.db.child("user").get()
        for res in users.each():
            key = res.key()
            value = res.val()
            if value['id'] == id_:
                #print("user:", value)
                self.db.child("user").child(key).remove()
        return True
    
    # heart
    def get_top_2_hearts_byname(self, uid):
        hearts = self.db.child("heart").child(uid).get()
        target_product = []

        if hearts.val() is None:
            return target_product
        # Create a list of tuples containing key-value pairs
        heart_list = [(res.key(), res.val()) for res in hearts.each()]
        # Sort the list based on the values in descending order
        
        top_2_hearts = heart_list[:2]
        
        print("########### target_heart : ", top_2_hearts)
        for name, _ in top_2_hearts:
            # Assuming that the name is the key for the product in "products" node
            product_info = self.db.child("item").child(name).get()
            if product_info.val() is not None:
                target_product.append(product_info.val())
        print("####### target product : ", target_product)
        return target_product
    
    # 구매 내역
    def get_users_purchase(self, buyer):
        result = []
        purchases = self.db.child("purchase").child(buyer).get()
        purchase_list = [res.key() for res in purchases.each()]
        top_2_purchases = purchase_list[:2]
        
        for name in top_2_purchases:
            item = self.db.child("item").child(name).get()
            if item.val() is not None:
                result.append(item.val())
        return result