SignalHandler.py

class SignalHandler:
    async def handle_exit(self):
        """
        Handle the exit of the server

        Args:
            None
        Returns:
            None
        """
        print("Signal intercepted")
        create_or_update_flag(False)  # Set the flag to False
        while len(queue) > 0:
            # Wait for the entire queue to be consumed
            print("Waiting for the queue to be consumed")
            await asyncio.sleep(5)
        print("Queue is empty. Exiting.")
        await manager.close_all_connections()  # Close all the websocket connections
        await manager.wait_for_connections_to_close()
        await server.shutdown()
        await server.lifespan.shutdown()
        for task in asyncio.all_tasks():
            try:
                task.cancel()
            except asyncio.CancelledError:
                pass
        try:
            await asyncio.gather(*asyncio.all_tasks() - {asyncio.current_task()})
            # Wait for all tasks to finish
        except asyncio.CancelledError:
            pass
    
    def signal_handler(self):
        """
        Signal handler function
          Args:
            None
        Returns:
            None
        """
        asyncio.create_task(self.handle_exit())

    def register_signal_handler(self) -> None:
        """
        Register signal handlers for the server
        
        Args:
            None
        Returns:
            None
        """
        loop = asyncio.get_running_loop()
        for sig in HANDLED_SIGNALS:
            loop.add_signal_handler(sig, self.signal_handler)