mirror of
https://git.anonymousland.org/anonymousland/synapse-product.git
synced 2024-09-09 15:42:10 +00:00
Simplify intern_dict
This commit is contained in:
parent
2f0180b09e
commit
8122ad7bab
@ -65,27 +65,20 @@ def intern_string(string):
|
||||
def intern_dict(dictionary):
|
||||
"""Takes a dictionary and interns well known keys and their values
|
||||
"""
|
||||
return _intern_known_values({
|
||||
_intern_key(key): value for key, value in dictionary.items()
|
||||
})
|
||||
return {
|
||||
KNOWN_KEYS.get(key, key): _intern_known_values(key, value)
|
||||
for key, value in dictionary.items()
|
||||
}
|
||||
|
||||
|
||||
def _intern_known_values(dictionary):
|
||||
def _intern_known_values(key, value):
|
||||
intern_str_keys = ("event_id", "room_id")
|
||||
intern_unicode_keys = ("sender", "user_id", "type", "state_key")
|
||||
|
||||
for key in intern_str_keys:
|
||||
val = dictionary.get(key, None)
|
||||
if val is not None:
|
||||
dictionary[key] = intern(val.encode('ascii'))
|
||||
if key in intern_str_keys:
|
||||
return intern(value.encode('ascii'))
|
||||
|
||||
for key in intern_unicode_keys:
|
||||
val = dictionary.get(key, None)
|
||||
if val is not None:
|
||||
dictionary[key] = intern_string(val)
|
||||
if key in intern_unicode_keys:
|
||||
return intern_string(value)
|
||||
|
||||
return dictionary
|
||||
|
||||
|
||||
def _intern_key(key):
|
||||
return KNOWN_KEYS.get(key, key)
|
||||
return value
|
||||
|
Loading…
Reference in New Issue
Block a user