TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-30 17:43:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix update_chat_notifications

Browse Source
This commit is contained in:
KurimuzonAkuma 2023-11-13 02:00:12 +03:00
parent 344e942d77
commit 3e73201f1d
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pyrogram/methods/chats/update_chat_notifications.py
View File

@ -29,7 +29,7 @@ class UpdateChatNotifications:
self: "pyrogram.Client", self: "pyrogram.Client",
chat_id: Union[int, str], chat_id: Union[int, str],
mute: bool = None, mute: bool = None,
mute_until: datetime = utils.zero_datetime(), mute_until: datetime = None,
stories_muted: bool = None, stories_muted: bool = None,
stories_hide_sender: bool = None, stories_hide_sender: bool = None,
show_previews: bool = None show_previews: bool = None
@ -44,7 +44,7 @@ class UpdateChatNotifications:
Pass True if you want to mute chat. Pass True if you want to mute chat.
until_date (:py:obj:`~datetime.datetime`, *optional*): until_date (:py:obj:`~datetime.datetime`, *optional*):
Date when the user will be unmuted. Defaults to epoch (mute forever). Date when the user will be unmuted. Works only if the mute parameter is set to True. Defaults to forever.
stories_muted (``bool``, *optional*): stories_muted (``bool``, *optional*):
N/A N/A
@ -74,6 +74,9 @@ class UpdateChatNotifications:
# Unmute a chat # Unmute a chat
app.update_chat_notifications(chat_id, mute=False) app.update_chat_notifications(chat_id, mute=False)
""" """
if not mute_until:
mute_until = utils.max_datetime() if mute else utils.zero_datetime()
if isinstance(mute_until, datetime.timedelta): if isinstance(mute_until, datetime.timedelta):
mute_until = datetime.datetime.now() + mute_until mute_until = datetime.datetime.now() + mute_until

4
pyrogram/utils.py
View File

@ -461,6 +461,10 @@ def zero_datetime() -> datetime:
return datetime.fromtimestamp(0, timezone.utc) return datetime.fromtimestamp(0, timezone.utc)
def max_datetime() -> datetime:
return datetime.fromtimestamp((1 << 31) - 1, timezone.utc)
def timestamp_to_datetime(ts: Optional[int]) -> Optional[datetime]: def timestamp_to_datetime(ts: Optional[int]) -> Optional[datetime]:
return datetime.fromtimestamp(ts) if ts else None return datetime.fromtimestamp(ts) if ts else None