Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,9 @@ These changes are available on the `master` branch, but have not yet been releas

### Added

- Added `RawMessageUpdateEvent.new_message` - message update events now contain full
message objects ([#2780](https://github.com/Pycord-Development/pycord/pull/2780))

### Changed

### Fixed
Expand Down
20 changes: 0 additions & 20 deletions discord/message.py
Original file line number Diff line number Diff line change
Expand Up @@ -1208,26 +1208,6 @@ def _clear_emoji(self, emoji) -> Reaction | None:
del self.reactions[index]
return reaction

def _update(self, data):
# In an update scheme, 'author' key has to be handled before 'member'
# otherwise they overwrite each other which is undesirable.
# Since there's no good way to do this we have to iterate over every
# handler rather than iterating over the keys which is a little slower
for key, handler in self._HANDLERS:
try:
value = data[key]
except KeyError:
continue
else:
handler(self, value)

# clear the cached properties
for attr in self._CACHED_SLOTS:
try:
delattr(self, attr)
except AttributeError:
pass

def _handle_edited_timestamp(self, value: str) -> None:
self._edited_timestamp = utils.parse_time(value)

Expand Down
8 changes: 7 additions & 1 deletion discord/raw_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -185,15 +185,21 @@ class RawMessageUpdateEvent(_RawReprMixin):
cached_message: Optional[:class:`Message`]
The cached message, if found in the internal message cache. Represents the message before
it is modified by the data in :attr:`RawMessageUpdateEvent.data`.
new_message: :class:`Message`
The new message object. Represents the message after it is modified by the data in
:attr:`RawMessageUpdateEvent.data`.

.. versionadded:: 2.7
"""

__slots__ = ("message_id", "channel_id", "guild_id", "data", "cached_message")

def __init__(self, data: MessageUpdateEvent) -> None:
def __init__(self, data: MessageUpdateEvent, new_message: Message) -> None:
self.message_id: int = int(data["id"])
self.channel_id: int = int(data["channel_id"])
self.data: MessageUpdateEvent = data
self.cached_message: Message | None = None
self.new_message: Message = new_message

try:
self.guild_id: int | None = int(data["guild_id"])
Expand Down
22 changes: 10 additions & 12 deletions discord/state.py
Original file line number Diff line number Diff line change
Expand Up @@ -775,21 +775,19 @@ def parse_message_delete_bulk(self, data) -> None:
self._messages.remove(msg) # type: ignore

def parse_message_update(self, data) -> None:
raw = RawMessageUpdateEvent(data)
message = self._get_message(raw.message_id)
if message is not None:
older_message = copy.copy(message)
raw.cached_message = older_message
self.dispatch("raw_message_edit", raw)
message._update(data)
# Coerce the `after` parameter to take the new updated Member
# ref: #5999
older_message.author = message.author
self.dispatch("message_edit", older_message, message)
old_message = self._get_message(raw.message_id)
if old_message is not None:
self._messages.remove(old_message)
channel, _ = self._get_guild_channel(data)
message = Message(channel=channel, data=data, state=self)
self._messages.append(message)
raw = RawMessageUpdateEvent(data, message)
self.dispatch("raw_message_edit", raw)
if old_message is not None:
self.dispatch("message_edit", old_message, message)
else:
if poll_data := data.get("poll"):
self.store_raw_poll(poll_data, raw)
self.dispatch("raw_message_edit", raw)

if "components" in data and self._view_store.is_message_tracked(raw.message_id):
self._view_store.update_from_message(raw.message_id, data["components"])
Expand Down
Loading