db.py

from datetime import datetime
import os
from bson.objectid import ObjectId
from user import User
from pymongo import DESCENDING, MongoClient
from werkzeug.security import generate_password_hash
from pytz import timezone
from dotenv import load_dotenv

load_dotenv()

client = MongoClient(os.getenv("MONGODB_URI"))

chat_db = client.get_database("chatoos")
users_collection = chat_db.get_collection("users")
rooms_collection = chat_db.get_collection("rooms")
room_members_collection = chat_db.get_collection("room_members")
messages_collection = chat_db.get_collection("messages")

#common room details------>
ROOM_ID = os.getenv("CHATOOS_ROOM_ID", "693fc58fcb016a1823ebb2e9")
ROOM_NAME = os.getenv("CHATOOS_ROOM_NAME", 'Chatoosites Room :)')
ADDED_BY = os.getenv("CHATOOS_ADDED_BY", 'aakarsh2504@gmail.com')

def get_users():
    return list(users_collection.find())
def save_user(username,name,dp_url):
    users_collection.insert_one({'_id':username,'name':name,'dp_url':dp_url,'notification-status':None,'notification-token':None})

def update_user_notification_status(username,status,token):
    return users_collection.update_one({'_id':username},{'$set':{'notification-status':status,'notification-token':token}})

def get_user(username):
    user_data = users_collection.find_one({'_id':username})
    return User(user_data['_id'],user_data['name'],user_data['dp_url'],user_data['notification-status'],user_data['notification-token']) if user_data else None

def save_room(room_name, created_by):
    room_id = rooms_collection.insert_one({'name':room_name, 'created_by': created_by, 'created_at':datetime.now()}).inserted_id
    add_room_member(room_id,room_name,created_by,created_by,is_room_admin=True)
    return room_id

def update_room(room_id,room_name):
    rooms_collection.update_one({'_id':ObjectId(room_id)},{'$set':{'name':room_name}})
    room_members_collection.update_many({'_id.room_id': ObjectId(room_id)},{'$set':{'room_name':room_name}})

def get_room(room_id):
    return rooms_collection.find_one({'_id':ObjectId(room_id)})

def add_room_member(room_id, room_name, username, added_by, is_room_admin=False):
    room_members_collection.insert_one({'_id':{'room_id':ObjectId(room_id), 'username':username}, 'room_name':room_name,'added_by':added_by,'added_at':datetime.now(), 'is_room_admin':is_room_admin})

def add_room_members(room_id,room_name,usernames,added_by):
    room_members_collection.insert_many([{'_id':{'room_id':ObjectId(room_id), 'username':username}, 'room_name':room_name,'added_by':added_by,'added_at':datetime.now(), 'is_room_admin':False} for username in usernames ])

def remove_room_members(room_id,usernames):
    room_members_collection.delete_many({'_id':{'$in':[{'room_id': ObjectId(room_id),'username':username} for username in usernames]}})

def get_room_members(room_id):
    return list(room_members_collection.find({'_id.room_id':ObjectId(room_id)}))

def get_rooms_for_user(username):
    return list(room_members_collection.find({'_id.username':username}))

def is_room_member(room_id,username):
    return room_members_collection.count_documents({'_id':{'room_id':ObjectId(room_id),'username':username}})

def is_room_admin(room_id,username):
    return room_members_collection.count_documents({'_id':{'room_id':ObjectId(room_id),'username':username},'is_room_admin':True})

def save_message(room_id,text,sender_name,sender,dp_url):
    messages_collection.insert_one({'room_id':room_id,'text':text,'sender_name':sender_name,'sender':sender, 'dp_url':dp_url,'created_at':datetime.now(timezone('Asia/Kolkata'))})

MESSAGE_FETCH_LIMIT = 3

def get_messages(room_id,page=0):
    offset = page * MESSAGE_FETCH_LIMIT
    messages = list(messages_collection.find({'room_id':room_id}).sort('_id',DESCENDING).limit(MESSAGE_FETCH_LIMIT).skip(offset))

    for message in messages:
        message['created_at'] = message['created_at'].strftime("%d %b, %H:%M")
    return messages[::-1]