Move gender dict into _user.py

This commit is contained in:
Mads Marquart
2019-03-10 20:25:02 +01:00
parent 71f19dd3c7
commit 8e6ee4636e
3 changed files with 34 additions and 35 deletions

View File

@@ -3,10 +3,40 @@ from __future__ import unicode_literals
import attr import attr
from ._core import Enum from ._core import Enum
from . import _util, _plan from . import _plan
from ._thread import ThreadType, Thread from ._thread import ThreadType, Thread
GENDERS = {
# For standard requests
0: "unknown",
1: "female_singular",
2: "male_singular",
3: "female_singular_guess",
4: "male_singular_guess",
5: "mixed",
6: "neuter_singular",
7: "unknown_singular",
8: "female_plural",
9: "male_plural",
10: "neuter_plural",
11: "unknown_plural",
# For graphql requests
"UNKNOWN": "unknown",
"FEMALE": "female_singular",
"MALE": "male_singular",
# '': 'female_singular_guess',
# '': 'male_singular_guess',
# '': 'mixed',
"NEUTER": "neuter_singular",
# '': 'unknown_singular',
# '': 'female_plural',
# '': 'male_plural',
# '': 'neuter_plural',
# '': 'unknown_plural',
}
class TypingStatus(Enum): class TypingStatus(Enum):
"""Used to specify whether the user is typing or has stopped typing""" """Used to specify whether the user is typing or has stopped typing"""
@@ -81,7 +111,7 @@ class User(Thread):
first_name=data.get("first_name"), first_name=data.get("first_name"),
last_name=data.get("last_name"), last_name=data.get("last_name"),
is_friend=data.get("is_viewer_friend"), is_friend=data.get("is_viewer_friend"),
gender=_util.GENDERS.get(data.get("gender")), gender=GENDERS.get(data.get("gender")),
affinity=data.get("affinity"), affinity=data.get("affinity"),
nickname=c_info.get("nickname"), nickname=c_info.get("nickname"),
color=c_info.get("color"), color=c_info.get("color"),
@@ -127,7 +157,7 @@ class User(Thread):
first_name=first_name, first_name=first_name,
last_name=last_name, last_name=last_name,
is_friend=user.get("is_viewer_friend"), is_friend=user.get("is_viewer_friend"),
gender=_util.GENDERS.get(user.get("gender")), gender=GENDERS.get(user.get("gender")),
affinity=user.get("affinity"), affinity=user.get("affinity"),
nickname=c_info.get("nickname"), nickname=c_info.get("nickname"),
color=c_info.get("color"), color=c_info.get("color"),
@@ -148,7 +178,7 @@ class User(Thread):
photo=data.get("thumbSrc"), photo=data.get("thumbSrc"),
name=data.get("name"), name=data.get("name"),
is_friend=data.get("is_friend"), is_friend=data.get("is_friend"),
gender=_util.GENDERS.get(data.get("gender")), gender=GENDERS.get(data.get("gender")),
) )

View File

@@ -47,36 +47,6 @@ USER_AGENTS = [
] ]
GENDERS = {
# For standard requests
0: "unknown",
1: "female_singular",
2: "male_singular",
3: "female_singular_guess",
4: "male_singular_guess",
5: "mixed",
6: "neuter_singular",
7: "unknown_singular",
8: "female_plural",
9: "male_plural",
10: "neuter_plural",
11: "unknown_plural",
# For graphql requests
"UNKNOWN": "unknown",
"FEMALE": "female_singular",
"MALE": "male_singular",
# '': 'female_singular_guess',
# '': 'male_singular_guess',
# '': 'mixed',
"NEUTER": "neuter_singular",
# '': 'unknown_singular',
# '': 'female_plural',
# '': 'male_plural',
# '': 'neuter_plural',
# '': 'unknown_plural',
}
class ReqUrl(object): class ReqUrl(object):
"""A class containing all urls used by `fbchat`""" """A class containing all urls used by `fbchat`"""

View File

@@ -7,7 +7,6 @@ from ._util import (
log, log,
handler, handler,
USER_AGENTS, USER_AGENTS,
GENDERS,
ReqUrl, ReqUrl,
facebookEncoding, facebookEncoding,
now, now,