mirror of
https://mau.dev/maunium/synapse.git
synced 2024-08-27 12:42:39 +00:00
Fill out the prune_event method.
This commit is contained in:
parent
3d2cca6762
commit
4e79b09dd9
@ -13,7 +13,11 @@
|
|||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
|
|
||||||
from .room import RoomMemberEvent
|
from .room import (
|
||||||
|
RoomMemberEvent, RoomJoinRulesEvent, RoomPowerLevelsEvent,
|
||||||
|
RoomAddStateLevelEvent, RoomSendEventLevelEvent, RoomOpsPowerLevelsEvent,
|
||||||
|
RoomAliasesEvent, RoomCreateEvent,
|
||||||
|
)
|
||||||
|
|
||||||
def prune_event(event):
|
def prune_event(event):
|
||||||
""" Prunes the given event of all keys we don't know about or think could
|
""" Prunes the given event of all keys we don't know about or think could
|
||||||
@ -27,12 +31,33 @@ def prune_event(event):
|
|||||||
# Remove all extraneous fields.
|
# Remove all extraneous fields.
|
||||||
event.unrecognized_keys = {}
|
event.unrecognized_keys = {}
|
||||||
|
|
||||||
|
new_content = {}
|
||||||
|
|
||||||
|
def add_fields(*fields):
|
||||||
|
for field in fields:
|
||||||
|
if field in event.content:
|
||||||
|
new_content[field] = event.content[field]
|
||||||
|
|
||||||
if event.type == RoomMemberEvent.TYPE:
|
if event.type == RoomMemberEvent.TYPE:
|
||||||
new_content = {
|
add_fields("membership")
|
||||||
"membership": event.content["membership"]
|
elif event.type == RoomCreateEvent.TYPE:
|
||||||
}
|
add_fields("creator")
|
||||||
else:
|
elif event.type == RoomJoinRulesEvent.TYPE:
|
||||||
new_content = {}
|
add_fields("join_rule")
|
||||||
|
elif event.type == RoomPowerLevelsEvent.TYPE:
|
||||||
|
# TODO: Actually check these are valid user_ids etc.
|
||||||
|
add_fields("default")
|
||||||
|
for k, v in event.content.items():
|
||||||
|
if k.startswith("@") and isinstance(v, (int, long)):
|
||||||
|
new_content[k] = v
|
||||||
|
elif event.type == RoomAddStateLevelEvent.TYPE:
|
||||||
|
add_fields("level")
|
||||||
|
elif event.type == RoomSendEventLevelEvent.TYPE:
|
||||||
|
add_fields("level")
|
||||||
|
elif event.type == RoomOpsPowerLevelsEvent.TYPE:
|
||||||
|
add_fields("kick_level", "ban_level", "delete_level")
|
||||||
|
elif event.type == RoomAliasesEvent.TYPE:
|
||||||
|
add_fields("aliases")
|
||||||
|
|
||||||
event.content = new_content
|
event.content = new_content
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user