Add exists() query

panther/cli/utils.py

@@ -142,7 +142,7 @@ def print_info(config: Config):
     if config.HAS_WS:
         try:
             import gunicorn
-            gunicorn_msg = f'{h} * You have WS so make sure to run gunicorn with --preload{h}\n'
+            gunicorn_msg = f'{h} * You have WS, so make sure to run gunicorn with --preload{h}\n'
         except ImportError:
             pass
 

panther/db/queries/queries.py

@@ -364,6 +364,26 @@ async def find_one_or_raise(cls, _filter: dict | None = None, /, **kwargs) -> Se
 
         raise NotFoundAPIError(detail=f'{cls.__name__} Does Not Exist')
 
+    @classmethod
+    async def exists(cls, _filter: dict | None = None, /, **kwargs) -> bool:
+        """
+        Check if document exists in collection or not
+
+        Example:
+        -------
+            >>> from app.models import User
+
+            >>> await User.exists(age=18, name='Ali')
+            or
+            >>> await User.exists({'age': 18, 'name': 'Ali'})
+            or
+            >>> await User.exists({'age': 18}, name='Ali')
+        """
+        if await cls.count(_filter, **kwargs) > 0:
+            return True
+        else:
+            return False
+
     async def save(self) -> None:
         """
         Save the document

panther/utils.py

@@ -50,6 +50,20 @@ def generate_secret_key() -> str:
 
 
 def round_datetime(dt: datetime, delta: timedelta):
+    """
+    Example:
+        >>> round_datetime(datetime(2024, 7, 15, 13, 22, 11, 562159), timedelta(days=2))
+        datetime.datetime(2024, 7, 16, 0, 0)
+
+        >>> round_datetime(datetime(2024, 7, 16, 13, 22, 11, 562159), timedelta(days=2))
+        datetime.datetime(2024, 7, 16, 0, 0)
+
+        >>> round_datetime(datetime(2024, 7, 17, 13, 22, 11, 562159), timedelta(days=2))
+        datetime.datetime(2024, 7, 18, 0, 0)
+
+        >>> round_datetime(datetime(2024, 7, 18, 13, 22, 11, 562159), timedelta(days=2))
+        datetime.datetime(2024, 7, 18, 0, 0)
+    """
     return datetime.min + round((dt - datetime.min) / delta) * delta
 
 

panther/websocket.py

@@ -27,6 +27,12 @@ async def send(self, data: any = None):
         """
         return await super().send(data=data)
 
+    async def close(self, code: int = status.WS_1000_NORMAL_CLOSURE, reason: str = ''):
+        """
+        Called whenever server or client, wants to close the connection
+        """
+        return await super().close(code=code, reason=reason)
+
 
 async def send_message_to_websocket(connection_id: str, data: any):
     await config.WEBSOCKET_CONNECTIONS.publish(connection_id=connection_id, action='send', data=data)