mirror of https://gitlab.com/litecord/litecord.git
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
"""
|
|
|
|
Litecord
|
|
Copyright (C) 2018-2021 Luna Mendes and Litecord Contributors
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, version 3 of the License.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""
|
|
|
|
from typing import Callable, List
|
|
|
|
from quart import current_app as app
|
|
from .dispatcher import GatewayEvent
|
|
from .utils import send_event_to_states
|
|
|
|
|
|
async def dispatch_user_filter(
|
|
user_id: int, filter_func: Callable[[str], bool], event_data: GatewayEvent
|
|
) -> List[str]:
|
|
"""Dispatch to a given user's states, but only for states
|
|
where filter_func returns true."""
|
|
states = list(
|
|
filter(
|
|
lambda state: filter_func(state.session_id),
|
|
app.state_manager.user_states(user_id),
|
|
)
|
|
)
|
|
|
|
return await send_event_to_states(states, event_data)
|
|
|
|
|
|
async def dispatch_user(user_id: int, event_data: GatewayEvent) -> List[str]:
|
|
return await dispatch_user_filter(user_id, lambda sess_id: True, event_data)
|