221 lines
6.8 KiB
Python
221 lines
6.8 KiB
Python
import attr
|
|
from ._core import log, attrs_default, Image
|
|
from . import _util, _session, _plan, _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',
|
|
}
|
|
|
|
|
|
@attrs_default
|
|
class User(_thread.ThreadABC):
|
|
"""Represents a Facebook user. Implements `ThreadABC`."""
|
|
|
|
#: The session to use when making requests.
|
|
session = attr.ib(type=_session.Session)
|
|
#: The user's unique identifier.
|
|
id = attr.ib(converter=str)
|
|
|
|
def _to_send_data(self):
|
|
return {
|
|
"other_user_fbid": self.id,
|
|
# The entry below is to support .wave
|
|
"specific_to_list[0]": "fbid:{}".format(self.id),
|
|
}
|
|
|
|
def confirm_friend_request(self):
|
|
"""Confirm a friend request, adding the user to your friend list."""
|
|
data = {"to_friend": self.id, "action": "confirm"}
|
|
j = self.session._payload_post("/ajax/add_friend/action.php?dpr=1", data)
|
|
|
|
def remove_friend(self):
|
|
"""Remove the user from the client's friend list."""
|
|
data = {"uid": self.id}
|
|
j = self.session._payload_post("/ajax/profile/removefriendconfirm.php", data)
|
|
|
|
def block(self):
|
|
"""Block messages from the user."""
|
|
data = {"fbid": self.id}
|
|
j = self.session._payload_post("/messaging/block_messages/?dpr=1", data)
|
|
|
|
def unblock(self):
|
|
"""Unblock a previously blocked user."""
|
|
data = {"fbid": self.id}
|
|
j = self.session._payload_post("/messaging/unblock_messages/?dpr=1", data)
|
|
|
|
|
|
@attrs_default
|
|
class UserData(User):
|
|
"""Represents data about a Facebook user.
|
|
|
|
Inherits `User`, and implements `ThreadABC`.
|
|
"""
|
|
|
|
#: The user's picture
|
|
photo = attr.ib()
|
|
#: The name of the user
|
|
name = attr.ib()
|
|
#: Whether the user and the client are friends
|
|
is_friend = attr.ib()
|
|
#: The users first name
|
|
first_name = attr.ib()
|
|
#: The users last name
|
|
last_name = attr.ib(None)
|
|
#: Datetime when the thread was last active / when the last message was sent
|
|
last_active = attr.ib(None)
|
|
#: Number of messages in the thread
|
|
message_count = attr.ib(None)
|
|
#: Set `Plan`
|
|
plan = attr.ib(None)
|
|
#: The profile URL. ``None`` for Messenger-only users
|
|
url = attr.ib(None)
|
|
#: The user's gender
|
|
gender = attr.ib(None)
|
|
#: From 0 to 1. How close the client is to the user
|
|
affinity = attr.ib(None)
|
|
#: The user's nickname
|
|
nickname = attr.ib(None)
|
|
#: The clients nickname, as seen by the user
|
|
own_nickname = attr.ib(None)
|
|
#: The message color
|
|
color = attr.ib(None)
|
|
#: The default emoji
|
|
emoji = attr.ib(None)
|
|
|
|
@staticmethod
|
|
def _get_other_user(data):
|
|
(user,) = (
|
|
node["messaging_actor"]
|
|
for node in data["all_participants"]["nodes"]
|
|
if node["messaging_actor"]["id"] == data["thread_key"]["other_user_id"]
|
|
)
|
|
return user
|
|
|
|
@classmethod
|
|
def _from_graphql(cls, session, data):
|
|
c_info = cls._parse_customization_info(data)
|
|
|
|
plan = None
|
|
if data.get("event_reminders") and data["event_reminders"].get("nodes"):
|
|
plan = _plan.PlanData._from_graphql(
|
|
session, data["event_reminders"]["nodes"][0]
|
|
)
|
|
|
|
return cls(
|
|
session=session,
|
|
id=data["id"],
|
|
url=data["url"],
|
|
first_name=data["first_name"],
|
|
last_name=data.get("last_name"),
|
|
is_friend=data["is_viewer_friend"],
|
|
gender=GENDERS.get(data["gender"]),
|
|
affinity=data.get("viewer_affinity"),
|
|
nickname=c_info.get("nickname"),
|
|
color=c_info["color"],
|
|
emoji=c_info["emoji"],
|
|
own_nickname=c_info.get("own_nickname"),
|
|
photo=Image._from_uri(data["profile_picture"]),
|
|
name=data["name"],
|
|
message_count=data.get("messages_count"),
|
|
plan=plan,
|
|
)
|
|
|
|
@classmethod
|
|
def _from_thread_fetch(cls, session, data):
|
|
user = cls._get_other_user(data)
|
|
if user["__typename"] != "User":
|
|
# TODO: Add Page._from_thread_fetch, and parse it there
|
|
log.warning("Tried to parse %s as a user.", user["__typename"])
|
|
return None
|
|
|
|
c_info = cls._parse_customization_info(data)
|
|
|
|
plan = None
|
|
if data["event_reminders"]["nodes"]:
|
|
plan = _plan.PlanData._from_graphql(
|
|
session, data["event_reminders"]["nodes"][0]
|
|
)
|
|
|
|
return cls(
|
|
session=session,
|
|
id=user["id"],
|
|
url=user["url"],
|
|
name=user["name"],
|
|
first_name=user["short_name"],
|
|
is_friend=user["is_viewer_friend"],
|
|
gender=GENDERS.get(user["gender"]),
|
|
nickname=c_info.get("nickname"),
|
|
color=c_info["color"],
|
|
emoji=c_info["emoji"],
|
|
own_nickname=c_info.get("own_nickname"),
|
|
photo=Image._from_uri(user["big_image_src"]),
|
|
message_count=data["messages_count"],
|
|
last_active=_util.millis_to_datetime(int(data["updated_time_precise"])),
|
|
plan=plan,
|
|
)
|
|
|
|
@classmethod
|
|
def _from_all_fetch(cls, session, data):
|
|
return cls(
|
|
session=session,
|
|
id=data["id"],
|
|
first_name=data["firstName"],
|
|
url=data["uri"],
|
|
photo=Image(url=data["thumbSrc"]),
|
|
name=data["name"],
|
|
is_friend=data["is_friend"],
|
|
gender=GENDERS.get(data["gender"]),
|
|
)
|
|
|
|
|
|
@attr.s
|
|
class ActiveStatus:
|
|
#: Whether the user is active now
|
|
active = attr.ib(None)
|
|
#: Datetime when the user was last active
|
|
last_active = attr.ib(None)
|
|
#: Whether the user is playing Messenger game now
|
|
in_game = attr.ib(None)
|
|
|
|
@classmethod
|
|
def _from_chatproxy_presence(cls, id_, data):
|
|
return cls(
|
|
active=data["p"] in [2, 3] if "p" in data else None,
|
|
last_active=_util.millis_to_datetime(data.get("lat")),
|
|
in_game=int(id_) in data.get("gamers", {}),
|
|
)
|
|
|
|
@classmethod
|
|
def _from_buddylist_overlay(cls, data, in_game=None):
|
|
return cls(
|
|
active=data["a"] in [2, 3] if "a" in data else None,
|
|
last_active=_util.millis_to_datetime(data.get("la")),
|
|
in_game=None,
|
|
)
|