From 8e6ee4636efb67e766cfcdfbb905d766d461e1fd Mon Sep 17 00:00:00 2001 From: Mads Marquart Date: Sun, 10 Mar 2019 20:25:02 +0100 Subject: [PATCH] Move gender dict into _user.py --- fbchat/_user.py | 38 ++++++++++++++++++++++++++++++++++---- fbchat/_util.py | 30 ------------------------------ fbchat/utils.py | 1 - 3 files changed, 34 insertions(+), 35 deletions(-) diff --git a/fbchat/_user.py b/fbchat/_user.py index 17ba014..b58a946 100644 --- a/fbchat/_user.py +++ b/fbchat/_user.py @@ -3,10 +3,40 @@ from __future__ import unicode_literals import attr from ._core import Enum -from . import _util, _plan +from . import _plan 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): """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"), last_name=data.get("last_name"), is_friend=data.get("is_viewer_friend"), - gender=_util.GENDERS.get(data.get("gender")), + gender=GENDERS.get(data.get("gender")), affinity=data.get("affinity"), nickname=c_info.get("nickname"), color=c_info.get("color"), @@ -127,7 +157,7 @@ class User(Thread): first_name=first_name, last_name=last_name, is_friend=user.get("is_viewer_friend"), - gender=_util.GENDERS.get(user.get("gender")), + gender=GENDERS.get(user.get("gender")), affinity=user.get("affinity"), nickname=c_info.get("nickname"), color=c_info.get("color"), @@ -148,7 +178,7 @@ class User(Thread): photo=data.get("thumbSrc"), name=data.get("name"), is_friend=data.get("is_friend"), - gender=_util.GENDERS.get(data.get("gender")), + gender=GENDERS.get(data.get("gender")), ) diff --git a/fbchat/_util.py b/fbchat/_util.py index 9bfe09c..6639b3e 100644 --- a/fbchat/_util.py +++ b/fbchat/_util.py @@ -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): """A class containing all urls used by `fbchat`""" diff --git a/fbchat/utils.py b/fbchat/utils.py index 66c5b9a..dc3385e 100644 --- a/fbchat/utils.py +++ b/fbchat/utils.py @@ -7,7 +7,6 @@ from ._util import ( log, handler, USER_AGENTS, - GENDERS, ReqUrl, facebookEncoding, now,