Fill out the prune_event method.

This commit is contained in:
Erik Johnston 2014-09-24 11:37:06 +01:00
parent 3d2cca6762
commit 4e79b09dd9

View File

@ -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