mirror of
https://github.com/python-telegram-bot/python-telegram-bot.git
synced 2024-12-13 02:10:40 +01:00
107 lines
4.1 KiB
Python
107 lines
4.1 KiB
Python
#!/usr/bin/env python
|
|
#
|
|
# A library that provides a Python interface to the Telegram Bot API
|
|
# Copyright (C) 2015-2020
|
|
# Leandro Toledo de Souza <devs@python-telegram-bot.org>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Lesser Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# 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 Lesser Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser Public License
|
|
# along with this program. If not, see [http://www.gnu.org/licenses/].
|
|
"""This module contains the Promise class."""
|
|
|
|
import logging
|
|
from threading import Event
|
|
from telegram.utils.types import JSONDict, HandlerArg
|
|
from typing import Callable, List, Tuple, Optional, Union, TypeVar
|
|
RT = TypeVar('RT')
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Promise:
|
|
"""A simple Promise implementation for use with the run_async decorator, DelayQueue etc.
|
|
|
|
Args:
|
|
pooled_function (:obj:`callable`): The callable that will be called concurrently.
|
|
args (:obj:`list` | :obj:`tuple`): Positional arguments for :attr:`pooled_function`.
|
|
kwargs (:obj:`dict`): Keyword arguments for :attr:`pooled_function`.
|
|
update (:class:`telegram.Update`, optional): The update this promise is associated with.
|
|
error_handling (:obj:`bool`, optional): Whether exceptions raised by :attr:`func`
|
|
may be handled by error handlers. Defaults to :obj:`True`.
|
|
|
|
Attributes:
|
|
pooled_function (:obj:`callable`): The callable that will be called concurrently.
|
|
args (:obj:`list` | :obj:`tuple`): Positional arguments for :attr:`pooled_function`.
|
|
kwargs (:obj:`dict`): Keyword arguments for :attr:`pooled_function`.
|
|
done (:obj:`threading.Event`): Is set when the result is available.
|
|
update (:class:`telegram.Update`): Optional. The update this promise is associated with.
|
|
error_handling (:obj:`bool`): Optional. Whether exceptions raised by :attr:`func`
|
|
may be handled by error handlers. Defaults to :obj:`True`.
|
|
|
|
"""
|
|
|
|
# TODO: Remove error_handling parameter once we drop the @run_async decorator
|
|
def __init__(self,
|
|
pooled_function: Callable[..., RT],
|
|
args: Union[List, Tuple],
|
|
kwargs: JSONDict,
|
|
update: HandlerArg = None,
|
|
error_handling: bool = True):
|
|
self.pooled_function = pooled_function
|
|
self.args = args
|
|
self.kwargs = kwargs
|
|
self.update = update
|
|
self.error_handling = error_handling
|
|
self.done = Event()
|
|
self._result: Optional[RT] = None
|
|
self._exception: Optional[Exception] = None
|
|
|
|
def run(self) -> None:
|
|
"""Calls the :attr:`pooled_function` callable."""
|
|
|
|
try:
|
|
self._result = self.pooled_function(*self.args, **self.kwargs)
|
|
|
|
except Exception as exc:
|
|
self._exception = exc
|
|
|
|
finally:
|
|
self.done.set()
|
|
|
|
def __call__(self) -> None:
|
|
self.run()
|
|
|
|
def result(self, timeout: float = None) -> Optional[RT]:
|
|
"""Return the result of the ``Promise``.
|
|
|
|
Args:
|
|
timeout (:obj:`float`, optional): Maximum time in seconds to wait for the result to be
|
|
calculated. ``None`` means indefinite. Default is ``None``.
|
|
|
|
Returns:
|
|
Returns the return value of :attr:`pooled_function` or ``None`` if the ``timeout``
|
|
expires.
|
|
|
|
Raises:
|
|
Any exception raised by :attr:`pooled_function`.
|
|
"""
|
|
self.done.wait(timeout=timeout)
|
|
if self._exception is not None:
|
|
raise self._exception # pylint: disable=raising-bad-type
|
|
return self._result
|
|
|
|
@property
|
|
def exception(self) -> Optional[Exception]:
|
|
"""The exception raised by :attr:`pooled_function` or ``None`` if no exception has been
|
|
raised (yet)."""
|
|
return self._exception
|