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)