added one more route app route for loading all the chats for hoeme page

Author: offiicialkamal

API/DATABASE/__pycache__/get_data.cpython-312.pyc

@@ -72,8 +72,10 @@ def all_users():
             if connection:
                 connection.close()
 
+    @staticmethod
     def stored_otp(RAW_UID, COOKIE):
         UID = int(RAW_UID)
+        connection = None
         try:
             connection = database.connect_users_db()
             cursor = connection.cursor()
@@ -95,7 +97,9 @@ def stored_otp(RAW_UID, COOKIE):
             if connection:
                 connection.close()
 
+    @staticmethod
     def password(EMAIL):
+        connection = None
         try:
             connection = database.connect_users_db()
             if connection:
@@ -114,7 +118,9 @@ def password(EMAIL):
             if connection:
                 connection.close()
 
+    @staticmethod
     def email(COOKIE):
+        connection = None
         try:
             connection = database.connect_users_db()
             if connection:
@@ -133,7 +139,9 @@ def email(COOKIE):
             if connection:
                 connection.close()
 
+    @staticmethod
     def cookie(EMAIL):
+        connection = None
         try:
             connection = database.connect_users_db()
             if connection:
@@ -152,7 +160,9 @@ def cookie(EMAIL):
             if connection:
                 connection.close()
 
+    @staticmethod
     def uid_by_email(EMAIL):
+        connection = None
         try:
             connection = database.connect_users_db()
             if connection:
@@ -173,9 +183,12 @@ def uid_by_email(EMAIL):
             if connection:
                 connection.close()
 
+    @staticmethod
     def uid_by_cookie(COOKIE):
+        connection = None
+        uid = None
         try:
-            uid = None
+            
             connection = database.connect_users_db()
             if connection:
                 cursor = connection.cursor()
@@ -196,7 +209,9 @@ def uid_by_cookie(COOKIE):
             if connection:
                 connection.close()
 
+    @staticmethod
     def uid_by_token(TOKEN):
+        connection = None
         try:
             connection = database.connect_users_db()
             if connection:
@@ -217,7 +232,9 @@ def uid_by_token(TOKEN):
             if connection:
                 connection.close()
 
+    @staticmethod
     def token(COOKIE):
+        connection = None
         try:
             connection = database.connect_users_db()
             if connection:
@@ -259,8 +276,9 @@ def profile_pic(*, UID=None, GUID=None):
         finally:
             if connection:
                 connection.close()
-
-    def first_name(RAW_UID):
+                
+    @staticmethod
+    def first_name(RAW_UID: str):
         UID = int(RAW_UID)
         try:
             connection = database.connect_users_db()

API/DATABASE/__pycache__/write_data.cpython-312.pyc

@@ -119,11 +119,10 @@ def store_this_message(GROUP_ID, SENDER_ID, SENDER_NAME, MESSAGE, PROFILE_PIC):
 
             result = cursor.fetchone()
             print(result)
-            cursor.close()
-            return MESSAGE_ID, result[0], result [1] #.isoformat()
+            return MESSAGE_ID, result[0], result [1]
         except Exception as e:
             print("error on store_this_message(GROUP_ID, SENDER_ID, SENDER_NAME, MESSAGE):", e)
-            return False, False
+            return False, False, False
         finally:
             if connection:
                 connection.close()

API/DATABASE/get_data.py

@@ -0,0 +1,49 @@
+from types import CodeType
+from fastapi import WebSocket
+
+
+class SocketManager:
+    """ IT MANAGES THE MESSAGE SENDING PROCESS LIKE BROADCASTING """
+
+    def __init__(self, all_online_users):
+        self.all_online_users = all_online_users
+        self.__rooms = dict()  #for storing all rooms and its an nested dict
+
+    def emit(self, event: str, data: dict, ws: WebSocket, room: int | str):
+        """ Sends the message to specific chat group"""
+        if (room in self.__rooms) and (ws in self.__rooms[room].values()):
+            for uid, wss in self.__rooms[room].items():
+                if uid == data["UID"]:
+                    continue
+                wss.send_jsomn({"event": event, "data": data})
+
+    def join_room(self, room_id: str, user_id: str, ws: str):
+        """ adds user to an group for listenong messages """
+        self.__rooms.setdefault(str(room_id), {})[str(user_id)] = str(ws)
+        return True
+
+    def leave_room(self, user_id: str, room_id: str):
+        """ MAINLY DESIGNED FOR WHEN USER LEAVES AN GROUP CHAT """
+        self.__rooms[room_id].remove(user_id) if self.__rooms.get(
+            room_id) else print(
+                'user not avilable in rooms skipped message sending')
+        return True
+
+    def leave_rooms(self, user_id: str):
+        """ REMOVES USER FROM ALL GROUPS MAINLY DESIGNED FOR OFFLINE, WHEN USER GOES OFFLINE """
+        for room in self.__rooms:
+            for uid in self.__rooms[room]:
+                if uid == user_id:
+                    self.__rooms[room].remove(uid)
+        return True
+
+    def broadCastMessage(self, message, type: CodeType | str):
+        """ Designed for any important notice by application admin to broadcast in all connected users """
+        for ws in self.all_online_users:
+            ws.send({
+                "event": "application_admin",
+                "type": type,
+                "data": {
+                    "message": message
+                }
+            })