From c292435ade89aefb51bb77af2ea153f9a191d77d Mon Sep 17 00:00:00 2001 From: jsl12 <32917998+jsl12@users.noreply.github.com> Date: Thu, 19 May 2022 21:04:06 -0500 Subject: [PATCH] added around kwarg to message_gen --- src/kwaylon/msg.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/kwaylon/msg.py b/src/kwaylon/msg.py index 13c033b..2ed7e78 100644 --- a/src/kwaylon/msg.py +++ b/src/kwaylon/msg.py @@ -14,15 +14,20 @@ async def message_gen(client: Client, before: datetime = None, days: int = None, after: datetime = None, + around: datetime = None, oldest_first: bool = True) -> AsyncIterator[Message]: if days is not None: after = (datetime.today() - timedelta(days=days)).astimezone() + if around is not None and limit > 100: + limit = 100 + kwargs = { 'limit': limit, 'before': before, 'after': after, - 'oldest_first': oldest_first + 'oldest_first': oldest_first, + 'around': around } LOGGER.info(kwargs)