Update database/filters_mdb.py

database/filters_mdb.py

@@ -2,37 +2,42 @@ import pymongo
 from pyrogram import enums 
 from info import DATABASE_URL, DATABASE_NAME
 import logging
+
 logger = logging.getLogger(__name__)
-logger.setLevel(logging.ERROR)
+logger.setLevel(logging.INFO)
 
 myclient = pymongo.MongoClient(DATABASE_URL)
 mydb = myclient["ManualFilters"]
 
-
-
 async def add_filter(grp_id, text, reply_text, btn, file, alert):
+    """
+    Add a filter to a specific group.
+    """
+    logger.info(f"Adding filter for group {grp_id} with text: {text}.")
     mycol = mydb[str(grp_id)]
-    # mycol.create_index([('text', 'text')])
 
     data = {
-        'text':str(text),
-        'reply':str(reply_text),
-        'btn':str(btn),
-        'file':str(file),
-        'alert':str(alert)
+        'text': str(text),
+        'reply': str(reply_text),
+        'btn': str(btn),
+        'file': str(file),
+        'alert': str(alert)
     }
 
     try:
-        mycol.update_one({'text': str(text)},  {"$set": data}, upsert=True)
-    except:
-        logger.exception('Some error occured!', exc_info=True)
-             
-     
+        mycol.update_one({'text': str(text)}, {"$set": data}, upsert=True)
+        logger.info(f"Filter added/updated for group {grp_id} with text: {text}.")
+    except Exception as e:
+        logger.exception(f"Error adding/updating filter for group {grp_id} with text: {text}: {e}")
+
 async def find_filter(group_id, name):
+    """
+    Find a filter in a specific group.
+    """
+    logger.info(f"Finding filter for group {group_id} with text: {name}.")
     mycol = mydb[str(group_id)]
     
-    query = mycol.find( {"text":name})
-    # query = mycol.find( { "$text": {"$search": name}})
+    query = mycol.find({"text": name})
     try:
         for file in query:
             reply_text = file['reply']
@@ -42,12 +47,19 @@ async def find_filter(group_id, name):
                 alert = file['alert']
             except:
                 alert = None
-        return reply_text, btn, alert, fileid
-    except:
+            logger.info(f"Filter found for group {group_id} with text: {name}.")
+            return reply_text, btn, alert, fileid
+        logger.info(f"No filter found for group {group_id} with text: {name}.")
+        return None, None, None, None
+    except Exception as e:
+        logger.exception(f"Error finding filter for group {group_id} with text: {name}: {e}")
         return None, None, None, None
-
 
 async def get_filters(group_id):
+    """
+    Get all filters in a specific group.
+    """
+    logger.info(f"Getting all filters for group {group_id}.")
     mycol = mydb[str(group_id)]
 
     texts = []
@@ -56,15 +68,20 @@ async def get_filters(group_id):
         for file in query:
             text = file['text']
             texts.append(text)
-    except:
-        pass
-    return texts
-
+        logger.info(f"Retrieved {len(texts)} filters for group {group_id}.")
+        return texts
+    except Exception as e:
+        logger.exception(f"Error getting filters for group {group_id}: {e}")
+        return []
 
 async def delete_filter(message, text, group_id):
+    """
+    Delete a specific filter from a group.
+    """
+    logger.info(f"Deleting filter for group {group_id} with text: {text}.")
     mycol = mydb[str(group_id)]
     
-    myquery = {'text':text }
+    myquery = {'text': text}
     query = mycol.count_documents(myquery)
     if query == 1:
         mycol.delete_one(myquery)
@@ -73,35 +90,50 @@ async def delete_filter(message, text, group_id):
             quote=True,
             parse_mode=enums.ParseMode.MARKDOWN
         )
+        logger.info(f"Filter deleted for group {group_id} with text: {text}.")
     else:
         await message.reply_text("Couldn't find that filter!", quote=True)
-
+        logger.info(f"No filter found to delete for group {group_id} with text: {text}.")
 
 async def del_all(message, group_id, title):
+    """
+    Delete all filters from a specific group.
+    """
+    logger.info(f"Deleting all filters for group {group_id} titled: {title}.")
     if str(group_id) not in mydb.list_collection_names():
         await message.edit_text(f"Nothing to remove in {title}!")
+        logger.info(f"No filters to remove in group {group_id} titled: {title}.")
         return
 
     mycol = mydb[str(group_id)]
     try:
         mycol.drop()
         await message.edit_text(f"All filters from {title} has been removed")
-    except:
+        logger.info(f"All filters removed from group {group_id} titled: {title}.")
+    except Exception as e:
+        logger.exception(f"Error removing all filters from group {group_id} titled: {title}: {e}")
         await message.edit_text("Couldn't remove all filters from group!")
-        return
-
 
 async def count_filters(group_id):
+    """
+    Count the number of filters in a specific group.
+    """
+    logger.info(f"Counting filters for group {group_id}.")
     mycol = mydb[str(group_id)]
 
-    count = mycol.count()
+    count = mycol.count_documents({})
     if count == 0:
+        logger.info(f"No filters found in group {group_id}.")
         return False
     else:
+        logger.info(f"Found {count} filters in group {group_id}.")
         return count
 
-
 async def filter_stats():
+    """
+    Get statistics about all filters.
+    """
+    logger.info("Getting filter statistics.")
     collections = mydb.list_collection_names()
 
     if "CONNECTION" in collections:
@@ -110,9 +142,10 @@ async def filter_stats():
     totalcount = 0
     for collection in collections:
         mycol = mydb[collection]
-        count = mycol.count()
+        count = mycol.count_documents({})
         totalcount += count
+        logger.info(f"Collection {collection} has {count} filters.")
 
     totalcollections = len(collections)
-
-    return totalcollections, totalcount
+    logger.info(f"Total collections: {totalcollections}, Total filters: {totalcount}.")
+    return totalcollections, totalcount