Move enum_extend_if_invalid -> Enum._extend_if_invalid

This commit is contained in:
Mads Marquart
2019-03-07 18:42:58 +01:00
parent 403870e39e
commit d940b64517
4 changed files with 16 additions and 17 deletions

View File

@@ -34,7 +34,7 @@ def graphql_color_to_enum(color):
return ThreadColor.MESSENGER_BLUE
color = color[2:] # Strip the alpha value
color_value = "#{}".format(color.lower())
return enum_extend_if_invalid(ThreadColor, color_value)
return ThreadColor._extend_if_invalid(color_value)
def get_customization_info(thread):
@@ -379,7 +379,7 @@ def graphql_to_message(message):
if message.get("unread") is not None:
rtn.is_read = not message["unread"]
rtn.reactions = {
str(r["user"]["id"]): enum_extend_if_invalid(MessageReaction, r["reaction"])
str(r["user"]["id"]): MessageReaction._extend_if_invalid(r["reaction"])
for r in message.get("message_reactions")
}
if message.get("blob_attachments") is not None: