karaolidis/fbchat
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Extend ThreadColor when encountering unknown values

Browse Source
This commit is contained in:
Mads Marquart
2018-12-12 22:44:19 +01:00
parent 381227af66
commit e41d981449
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
fbchat/graphql.py
View File

@@ -3,6 +3,7 @@
from __future__ import unicode_literals from __future__ import unicode_literals
import json import json
import re import re
import aenum
from .models import * from .models import *
from .utils import * from .utils import *
@@ -26,12 +27,16 @@ class ConcatJSONDecoder(json.JSONDecoder):
def graphql_color_to_enum(color): def graphql_color_to_enum(color):
if color is None: if color is None:
return None return None
if len(color) == 0: if not color:
return ThreadColor.MESSENGER_BLUE return ThreadColor.MESSENGER_BLUE
color = color[2:] # Strip the alpha value
color_value = '#{}'.format(color.lower())
try: try:
return ThreadColor('#{}'.format(color[2:].lower())) return ThreadColor(color_value)
except ValueError: except ValueError:
raise FBchatException('Could not get ThreadColor from color: {}'.format(color)) log.warning("Failed parsing color {}. Extending enum.".format(color))
aenum.extend_enum(ThreadColor, "UNKNOWN_{}".format(color), color_value)
return ThreadColor(color_value)
def get_customization_info(thread): def get_customization_info(thread):
if thread is None or thread.get('customization_info') is None: if thread is None or thread.get('customization_info') is None: