Compare commits
50 Commits
Author | SHA1 | Date | |
---|---|---|---|
|
0dac7b7b81 | ||
|
b750e753d6 | ||
|
ee33e92bed | ||
|
7413a643f6 | ||
|
cd4a18cb5a | ||
|
c00b3df8b2 | ||
|
f0c6e8612f | ||
|
1cebbf92e6 | ||
|
a64982583b | ||
|
cb8b0915de | ||
|
1d2576b06d | ||
|
ead9a3c0e9 | ||
|
59ba418faa | ||
|
c51a332560 | ||
|
a73d2feed6 | ||
|
6929193e9d | ||
|
fea4ad9e89 | ||
|
68099049d4 | ||
|
44cf08bdfd | ||
|
9e32cf17a4 | ||
|
0661367ebb | ||
|
3c07e42ba2 | ||
|
2cd6376818 | ||
|
5e7f7750de | ||
|
2a223ec6db | ||
|
a99108fff6 | ||
|
8de4698cc4 | ||
|
637319ec2c | ||
|
f9398564cd | ||
|
b57f423eb4 | ||
|
3093f1f2b6 | ||
|
961777e0c1 | ||
|
d7139701f7 | ||
|
c6bac17d48 | ||
|
3638fc5356 | ||
|
aca9176f7f | ||
|
0d5e4f6d3f | ||
|
92a5ffdef8 | ||
|
b3359fccdb | ||
|
d8f7366d1f | ||
|
ff94dc20af | ||
|
a8df0a548f | ||
|
13d0dc4ba4 | ||
|
64125a1aca | ||
|
4feae03092 | ||
|
5f993c2bf8 | ||
|
35bbcbffba | ||
|
5faca54d67 | ||
|
82496b8e04 | ||
|
2d74ec7823 |
@@ -17,7 +17,7 @@ from .client import *
|
||||
|
||||
|
||||
__copyright__ = 'Copyright 2015 - {} by Taehoon Kim'.format(datetime.now().year)
|
||||
__version__ = '1.0.4'
|
||||
__version__ = '1.0.21'
|
||||
__license__ = 'BSD'
|
||||
__author__ = 'Taehoon Kim; Moreels Pieter-Jan; Mads Marquart'
|
||||
__email__ = 'carpedm20@gmail.com'
|
||||
|
351
fbchat/client.py
351
fbchat/client.py
@@ -3,7 +3,6 @@
|
||||
from __future__ import unicode_literals
|
||||
import requests
|
||||
import urllib
|
||||
import traceback
|
||||
from uuid import uuid1
|
||||
from random import choice
|
||||
from datetime import datetime
|
||||
@@ -44,7 +43,7 @@ class Client(object):
|
||||
:type max_tries: int
|
||||
:type session_cookies: dict
|
||||
:type logging_level: int
|
||||
:raises: Exception on failed login
|
||||
:raises: FBchatException on failed login
|
||||
"""
|
||||
|
||||
self.sticky, self.pool = (None, None)
|
||||
@@ -55,14 +54,15 @@ class Client(object):
|
||||
self.client = 'mercury'
|
||||
self.default_thread_id = None
|
||||
self.default_thread_type = None
|
||||
self.req_url = ReqUrl()
|
||||
|
||||
if not user_agent:
|
||||
user_agent = choice(USER_AGENTS)
|
||||
|
||||
self._header = {
|
||||
'Content-Type' : 'application/x-www-form-urlencoded',
|
||||
'Referer' : ReqUrl.BASE,
|
||||
'Origin' : ReqUrl.BASE,
|
||||
'Referer' : self.req_url.BASE,
|
||||
'Origin' : self.req_url.BASE,
|
||||
'User-Agent' : user_agent,
|
||||
'Connection' : 'keep-alive',
|
||||
}
|
||||
@@ -72,6 +72,9 @@ class Client(object):
|
||||
# If session cookies aren't set, not properly loaded or gives us an invalid session, then do the login
|
||||
if not session_cookies or not self.setSession(session_cookies) or not self.isLoggedIn():
|
||||
self.login(email, password, max_tries)
|
||||
else:
|
||||
self.email = email
|
||||
self.password = password
|
||||
|
||||
"""
|
||||
INTERNAL REQUEST METHODS
|
||||
@@ -89,13 +92,50 @@ class Client(object):
|
||||
self.req_counter += 1
|
||||
return payload
|
||||
|
||||
def _get(self, url, query=None, timeout=30):
|
||||
payload = self._generatePayload(query)
|
||||
return self._session.get(url, headers=self._header, params=payload, timeout=timeout)
|
||||
def _fix_fb_errors(self, error_code):
|
||||
"""
|
||||
This fixes "Please try closing and re-opening your browser window" errors (1357004)
|
||||
This error usually happens after 1-2 days of inactivity
|
||||
It may be a bad idea to do this in an exception handler, if you have a better method, please suggest it!
|
||||
"""
|
||||
if error_code == '1357004':
|
||||
log.warning('Got error #1357004. Doing a _postLogin, and resending request')
|
||||
self._postLogin()
|
||||
return True
|
||||
return False
|
||||
|
||||
def _post(self, url, query=None, timeout=30):
|
||||
def _get(self, url, query=None, timeout=30, fix_request=False, as_json=False, error_retries=3):
|
||||
payload = self._generatePayload(query)
|
||||
return self._session.post(url, headers=self._header, data=payload, timeout=timeout)
|
||||
r = self._session.get(url, headers=self._header, params=payload, timeout=timeout)
|
||||
if not fix_request:
|
||||
return r
|
||||
try:
|
||||
return check_request(r, as_json=as_json)
|
||||
except FBchatFacebookError as e:
|
||||
if error_retries > 0 and self._fix_fb_errors(e.fb_error_code):
|
||||
return self._get(url, query=query, timeout=timeout, fix_request=fix_request, as_json=as_json, error_retries=error_retries-1)
|
||||
raise e
|
||||
|
||||
def _post(self, url, query=None, timeout=30, fix_request=False, as_json=False, error_retries=3):
|
||||
payload = self._generatePayload(query)
|
||||
r = self._session.post(url, headers=self._header, data=payload, timeout=timeout)
|
||||
if not fix_request:
|
||||
return r
|
||||
try:
|
||||
return check_request(r, as_json=as_json)
|
||||
except FBchatFacebookError as e:
|
||||
if error_retries > 0 and self._fix_fb_errors(e.fb_error_code):
|
||||
return self._post(url, query=query, timeout=timeout, fix_request=fix_request, as_json=as_json, error_retries=error_retries-1)
|
||||
raise e
|
||||
|
||||
def _graphql(self, payload, error_retries=3):
|
||||
content = self._post(self.req_url.GRAPHQL, payload, fix_request=True, as_json=False)
|
||||
try:
|
||||
return graphql_response_to_json(content)
|
||||
except FBchatFacebookError as e:
|
||||
if error_retries > 0 and self._fix_fb_errors(e.fb_error_code):
|
||||
return self._graphql(payload, error_retries=error_retries-1)
|
||||
raise e
|
||||
|
||||
def _cleanGet(self, url, query=None, timeout=30):
|
||||
return self._session.get(url, headers=self._header, params=query, timeout=timeout)
|
||||
@@ -104,38 +144,42 @@ class Client(object):
|
||||
self.req_counter += 1
|
||||
return self._session.post(url, headers=self._header, data=query, timeout=timeout)
|
||||
|
||||
def _postFile(self, url, files=None, query=None, timeout=30):
|
||||
def _postFile(self, url, files=None, query=None, timeout=30, fix_request=False, as_json=False, error_retries=3):
|
||||
payload=self._generatePayload(query)
|
||||
# Removes 'Content-Type' from the header
|
||||
headers = dict((i, self._header[i]) for i in self._header if i != 'Content-Type')
|
||||
return self._session.post(url, headers=headers, data=payload, timeout=timeout, files=files)
|
||||
r = self._session.post(url, headers=headers, data=payload, timeout=timeout, files=files)
|
||||
if not fix_request:
|
||||
return r
|
||||
try:
|
||||
return check_request(r, as_json=as_json)
|
||||
except FBchatFacebookError as e:
|
||||
if error_retries > 0 and self._fix_fb_errors(e.fb_error_code):
|
||||
return self._postFile(url, files=files, query=query, timeout=timeout, fix_request=fix_request, as_json=as_json, error_retries=error_retries-1)
|
||||
raise e
|
||||
|
||||
def graphql_requests(self, *queries):
|
||||
"""
|
||||
.. todo::
|
||||
Documenting this
|
||||
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
payload = {
|
||||
|
||||
return tuple(self._graphql({
|
||||
'method': 'GET',
|
||||
'response_format': 'json',
|
||||
'queries': graphql_queries_to_json(*queries)
|
||||
}
|
||||
|
||||
j = graphql_response_to_json(checkRequest(self._post(ReqUrl.GRAPHQL, payload), do_json_check=False))
|
||||
|
||||
return tuple(j)
|
||||
}))
|
||||
|
||||
def graphql_request(self, query):
|
||||
"""
|
||||
Shorthand for `graphql_requests(query)[0]`
|
||||
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
return self.graphql_requests(query)[0]
|
||||
|
||||
|
||||
"""
|
||||
END INTERNAL REQUEST METHODS
|
||||
"""
|
||||
@@ -155,11 +199,14 @@ class Client(object):
|
||||
self.payloadDefault = {}
|
||||
self.client_id = hex(int(random()*2147483648))[2:]
|
||||
self.start_time = now()
|
||||
self.uid = str(self._session.cookies['c_user'])
|
||||
self.uid = self._session.cookies.get_dict().get('c_user')
|
||||
if self.uid is None:
|
||||
raise FBchatException('Could not find c_user cookie')
|
||||
self.uid = str(self.uid)
|
||||
self.user_channel = "p_" + self.uid
|
||||
self.ttstamp = ''
|
||||
|
||||
r = self._get(ReqUrl.BASE)
|
||||
r = self._get(self.req_url.BASE)
|
||||
soup = bs(r.text, "lxml")
|
||||
self.fb_dtsg = soup.find("input", {'name':'fb_dtsg'})['value']
|
||||
self.fb_h = soup.find("input", {'name':'h'})['value']
|
||||
@@ -191,23 +238,25 @@ class Client(object):
|
||||
|
||||
def _login(self):
|
||||
if not (self.email and self.password):
|
||||
raise Exception("Email and password not found.")
|
||||
raise FBchatUserError("Email and password not found.")
|
||||
|
||||
soup = bs(self._get(ReqUrl.MOBILE).text, "lxml")
|
||||
soup = bs(self._get(self.req_url.MOBILE).text, "lxml")
|
||||
data = dict((elem['name'], elem['value']) for elem in soup.findAll("input") if elem.has_attr('value') and elem.has_attr('name'))
|
||||
data['email'] = self.email
|
||||
data['pass'] = self.password
|
||||
data['login'] = 'Log In'
|
||||
|
||||
r = self._cleanPost(ReqUrl.LOGIN, data)
|
||||
r = self._cleanPost(self.req_url.LOGIN, data)
|
||||
|
||||
# Usually, 'Checkpoint' will refer to 2FA
|
||||
if 'checkpoint' in r.url and 'Enter Security Code to Continue' in r.text:
|
||||
if ('checkpoint' in r.url
|
||||
and ('enter security code to continue' in r.text.lower()
|
||||
or 'enter login code to continue' in r.text.lower())):
|
||||
r = self._2FA(r)
|
||||
|
||||
# Sometimes Facebook tries to show the user a "Save Device" dialog
|
||||
if 'save-device' in r.url:
|
||||
r = self._cleanGet(ReqUrl.SAVE_DEVICE)
|
||||
r = self._cleanGet(self.req_url.SAVE_DEVICE)
|
||||
|
||||
if 'home' in r.url:
|
||||
self._postLogin()
|
||||
@@ -228,7 +277,7 @@ class Client(object):
|
||||
data['codes_submitted'] = 0
|
||||
log.info('Submitting 2FA code.')
|
||||
|
||||
r = self._cleanPost(ReqUrl.CHECKPOINT, data)
|
||||
r = self._cleanPost(self.req_url.CHECKPOINT, data)
|
||||
|
||||
if 'home' in r.url:
|
||||
return r
|
||||
@@ -240,14 +289,14 @@ class Client(object):
|
||||
data['name_action_selected'] = 'save_device'
|
||||
data['submit[Continue]'] = 'Continue'
|
||||
log.info('Saving browser.') # At this stage, we have dtsg, nh, name_action_selected, submit[Continue]
|
||||
r = self._cleanPost(ReqUrl.CHECKPOINT, data)
|
||||
r = self._cleanPost(self.req_url.CHECKPOINT, data)
|
||||
|
||||
if 'home' in r.url:
|
||||
return r
|
||||
|
||||
del(data['name_action_selected'])
|
||||
log.info('Starting Facebook checkup flow.') # At this stage, we have dtsg, nh, submit[Continue]
|
||||
r = self._cleanPost(ReqUrl.CHECKPOINT, data)
|
||||
r = self._cleanPost(self.req_url.CHECKPOINT, data)
|
||||
|
||||
if 'home' in r.url:
|
||||
return r
|
||||
@@ -255,7 +304,7 @@ class Client(object):
|
||||
del(data['submit[Continue]'])
|
||||
data['submit[This was me]'] = 'This Was Me'
|
||||
log.info('Verifying login attempt.') # At this stage, we have dtsg, nh, submit[This was me]
|
||||
r = self._cleanPost(ReqUrl.CHECKPOINT, data)
|
||||
r = self._cleanPost(self.req_url.CHECKPOINT, data)
|
||||
|
||||
if 'home' in r.url:
|
||||
return r
|
||||
@@ -264,7 +313,7 @@ class Client(object):
|
||||
data['submit[Continue]'] = 'Continue'
|
||||
data['name_action_selected'] = 'save_device'
|
||||
log.info('Saving device again.') # At this stage, we have dtsg, nh, submit[Continue], name_action_selected
|
||||
r = self._cleanPost(ReqUrl.CHECKPOINT, data)
|
||||
r = self._cleanPost(self.req_url.CHECKPOINT, data)
|
||||
return r
|
||||
|
||||
def isLoggedIn(self):
|
||||
@@ -275,7 +324,7 @@ class Client(object):
|
||||
:rtype: bool
|
||||
"""
|
||||
# Send a request to the login url, to see if we're directed to the home page
|
||||
r = self._cleanGet(ReqUrl.LOGIN)
|
||||
r = self._cleanGet(self.req_url.LOGIN)
|
||||
return 'home' in r.url
|
||||
|
||||
def getSession(self):
|
||||
@@ -303,7 +352,8 @@ class Client(object):
|
||||
# Load cookies into current session
|
||||
self._session.cookies = requests.cookies.merge_cookies(self._session.cookies, session_cookies)
|
||||
self._postLogin()
|
||||
except Exception:
|
||||
except Exception as e:
|
||||
log.exception('Failed loading session')
|
||||
self._resetValues()
|
||||
return False
|
||||
return True
|
||||
@@ -316,15 +366,15 @@ class Client(object):
|
||||
:param password: Facebook account password
|
||||
:param max_tries: Maximum number of times to try logging in
|
||||
:type max_tries: int
|
||||
:raises: Exception on failed login
|
||||
:raises: FBchatException on failed login
|
||||
"""
|
||||
self.onLoggingIn(email=email)
|
||||
|
||||
if max_tries < 1:
|
||||
raise Exception('Cannot login: max_tries should be at least one')
|
||||
raise FBchatUserError('Cannot login: max_tries should be at least one')
|
||||
|
||||
if not (email and password):
|
||||
raise Exception('Email and password not set')
|
||||
raise FBchatUserError('Email and password not set')
|
||||
|
||||
self.email = email
|
||||
self.password = password
|
||||
@@ -339,7 +389,7 @@ class Client(object):
|
||||
self.onLoggedIn(email=email)
|
||||
break
|
||||
else:
|
||||
raise Exception('Login failed. Check email/password. (Failed on url: {})'.format(login_url))
|
||||
raise FBchatUserError('Login failed. Check email/password. (Failed on url: {})'.format(login_url))
|
||||
|
||||
def logout(self):
|
||||
"""
|
||||
@@ -354,7 +404,7 @@ class Client(object):
|
||||
'h': self.fb_h
|
||||
}
|
||||
|
||||
r = self._get(ReqUrl.LOGOUT, data)
|
||||
r = self._get(self.req_url.LOGOUT, data)
|
||||
|
||||
self._resetValues()
|
||||
|
||||
@@ -412,21 +462,24 @@ class Client(object):
|
||||
|
||||
:return: :class:`models.User` objects
|
||||
:rtype: list
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
|
||||
data = {
|
||||
'viewer': self.uid,
|
||||
}
|
||||
j = checkRequest(self._post(ReqUrl.ALL_USERS, query=data))
|
||||
if not j['payload']:
|
||||
raise Exception('Missing payload')
|
||||
j = self._post(self.req_url.ALL_USERS, query=data, fix_request=True, as_json=True)
|
||||
if j.get('payload') is None:
|
||||
raise FBchatException('Missing payload while fetching users: {}'.format(j))
|
||||
|
||||
users = []
|
||||
|
||||
for key in j['payload']:
|
||||
k = j['payload'][key]
|
||||
if k['type'] in ['user', 'friend']:
|
||||
if k['id'] in ['0', 0]:
|
||||
# Skip invalid users
|
||||
pass
|
||||
users.append(User(k['id'], first_name=k.get('firstName'), url=k.get('uri'), photo=k.get('thumbSrc'), name=k.get('name'), is_friend=k.get('is_friend'), gender=GENDERS[k.get('gender')]))
|
||||
|
||||
return users
|
||||
@@ -439,7 +492,7 @@ class Client(object):
|
||||
:param limit: The max. amount of users to fetch
|
||||
:return: :class:`models.User` objects, ordered by relevance
|
||||
:rtype: list
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
|
||||
j = self.graphql_request(GraphQL(query=GraphQL.SEARCH_USER, params={'search': name, 'limit': limit}))
|
||||
@@ -453,7 +506,7 @@ class Client(object):
|
||||
:param name: Name of the page
|
||||
:return: :class:`models.Page` objects, ordered by relevance
|
||||
:rtype: list
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
|
||||
j = self.graphql_request(GraphQL(query=GraphQL.SEARCH_PAGE, params={'search': name, 'limit': limit}))
|
||||
@@ -468,7 +521,7 @@ class Client(object):
|
||||
:param limit: The max. amount of groups to fetch
|
||||
:return: :class:`models.Group` objects, ordered by relevance
|
||||
:rtype: list
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
|
||||
j = self.graphql_request(GraphQL(query=GraphQL.SEARCH_GROUP, params={'search': name, 'limit': limit}))
|
||||
@@ -483,7 +536,7 @@ class Client(object):
|
||||
:param limit: The max. amount of groups to fetch
|
||||
:return: :class:`models.User`, :class:`models.Group` and :class:`models.Page` objects, ordered by relevance
|
||||
:rtype: list
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
|
||||
j = self.graphql_request(GraphQL(query=GraphQL.SEARCH_THREAD, params={'search': name, 'limit': limit}))
|
||||
@@ -509,10 +562,10 @@ class Client(object):
|
||||
data = {
|
||||
"ids[{}]".format(i): _id for i, _id in enumerate(ids)
|
||||
}
|
||||
j = checkRequest(self._post(ReqUrl.INFO, data))
|
||||
j = self._post(self.req_url.INFO, data, fix_request=True, as_json=True)
|
||||
|
||||
if not j['payload']['profiles']:
|
||||
raise Exception('No users/pages returned')
|
||||
if j.get('payload') is None or j['payload'].get('profiles') is None:
|
||||
raise FBchatException('No users/pages returned: {}'.format(j))
|
||||
|
||||
entries = {}
|
||||
for _id in j['payload']['profiles']:
|
||||
@@ -537,7 +590,7 @@ class Client(object):
|
||||
'name': k.get('name')
|
||||
}
|
||||
else:
|
||||
raise Exception('{} had an unknown thread type: {}'.format(_id, k))
|
||||
raise FBchatException('{} had an unknown thread type: {}'.format(_id, k))
|
||||
|
||||
log.debug(entries)
|
||||
return entries
|
||||
@@ -552,7 +605,7 @@ class Client(object):
|
||||
:param user_ids: One or more user ID(s) to query
|
||||
:return: :class:`models.User` objects, labeled by their ID
|
||||
:rtype: dict
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
|
||||
threads = self.fetchThreadInfo(*user_ids)
|
||||
@@ -561,7 +614,7 @@ class Client(object):
|
||||
if threads[k].type == ThreadType.USER:
|
||||
users[k] = threads[k]
|
||||
else:
|
||||
raise Exception('Thread {} was not a user'.format(threads[k]))
|
||||
raise FBchatUserError('Thread {} was not a user'.format(threads[k]))
|
||||
|
||||
return users
|
||||
|
||||
@@ -575,7 +628,7 @@ class Client(object):
|
||||
:param page_ids: One or more page ID(s) to query
|
||||
:return: :class:`models.Page` objects, labeled by their ID
|
||||
:rtype: dict
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
|
||||
threads = self.fetchThreadInfo(*page_ids)
|
||||
@@ -584,7 +637,7 @@ class Client(object):
|
||||
if threads[k].type == ThreadType.PAGE:
|
||||
pages[k] = threads[k]
|
||||
else:
|
||||
raise Exception('Thread {} was not a page'.format(threads[k]))
|
||||
raise FBchatUserError('Thread {} was not a page'.format(threads[k]))
|
||||
|
||||
return pages
|
||||
|
||||
@@ -595,7 +648,7 @@ class Client(object):
|
||||
:param group_ids: One or more group ID(s) to query
|
||||
:return: :class:`models.Group` objects, labeled by their ID
|
||||
:rtype: dict
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
|
||||
threads = self.fetchThreadInfo(*group_ids)
|
||||
@@ -604,7 +657,7 @@ class Client(object):
|
||||
if threads[k].type == ThreadType.GROUP:
|
||||
groups[k] = threads[k]
|
||||
else:
|
||||
raise Exception('Thread {} was not a group'.format(threads[k]))
|
||||
raise FBchatUserError('Thread {} was not a group'.format(threads[k]))
|
||||
|
||||
return groups
|
||||
|
||||
@@ -618,7 +671,7 @@ class Client(object):
|
||||
:param thread_ids: One or more thread ID(s) to query
|
||||
:return: :class:`models.Thread` objects, labeled by their ID
|
||||
:rtype: dict
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
|
||||
queries = []
|
||||
@@ -657,14 +710,14 @@ class Client(object):
|
||||
elif entry.get('thread_type') == 'ONE_TO_ONE':
|
||||
_id = entry['thread_key']['other_user_id']
|
||||
if pages_and_users.get(_id) is None:
|
||||
raise Exception('Could not fetch thread {}'.format(_id))
|
||||
raise FBchatException('Could not fetch thread {}'.format(_id))
|
||||
entry.update(pages_and_users[_id])
|
||||
if entry['type'] == ThreadType.USER:
|
||||
rtn[_id] = graphql_to_user(entry)
|
||||
else:
|
||||
rtn[_id] = graphql_to_page(entry)
|
||||
else:
|
||||
raise Exception('{} had an unknown thread type: {}'.format(thread_ids[i], entry))
|
||||
raise FBchatException('{} had an unknown thread type: {}'.format(thread_ids[i], entry))
|
||||
|
||||
return rtn
|
||||
|
||||
@@ -679,7 +732,7 @@ class Client(object):
|
||||
:type before: int
|
||||
:return: :class:`models.Message` objects
|
||||
:rtype: list
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
|
||||
j = self.graphql_request(GraphQL(doc_id='1386147188135407', params={
|
||||
@@ -690,8 +743,8 @@ class Client(object):
|
||||
'before': before
|
||||
}))
|
||||
|
||||
if j['message_thread'] is None:
|
||||
raise Exception('Could not fetch thread {}'.format(thread_id))
|
||||
if j.get('message_thread') is None:
|
||||
raise FBchatException('Could not fetch thread {}: {}'.format(thread_id, j))
|
||||
|
||||
return list(reversed([graphql_to_message(message) for message in j['message_thread']['messages']['nodes']]))
|
||||
|
||||
@@ -704,11 +757,11 @@ class Client(object):
|
||||
:type limit: int
|
||||
:return: :class:`models.Thread` objects
|
||||
:rtype: list
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
|
||||
if limit > 20 or limit < 1:
|
||||
raise Exception('`limit` should be between 1 and 20')
|
||||
raise FBchatUserError('`limit` should be between 1 and 20')
|
||||
|
||||
data = {
|
||||
'client' : self.client,
|
||||
@@ -716,9 +769,9 @@ class Client(object):
|
||||
'inbox[limit]' : limit,
|
||||
}
|
||||
|
||||
j = checkRequest(self._post(ReqUrl.THREADS, data))
|
||||
j = self._post(self.req_url.THREADS, data, fix_request=True, as_json=True)
|
||||
if j.get('payload') is None:
|
||||
raise Exception('Missing payload: {}, with data: {}'.format(j, data))
|
||||
raise FBchatException('Missing payload: {}, with data: {}'.format(j, data))
|
||||
|
||||
participants = {}
|
||||
for p in j['payload']['participants']:
|
||||
@@ -727,18 +780,19 @@ class Client(object):
|
||||
elif p['type'] == 'user':
|
||||
participants[p['fbid']] = User(p['fbid'], url=p['href'], first_name=p['short_name'], is_friend=p['is_friend'], gender=GENDERS[p['gender']], photo=p['image_src'], name=p['name'])
|
||||
else:
|
||||
raise Exception('A participant had an unknown type {}: {}'.format(p['type'], p))
|
||||
raise FBchatException('A participant had an unknown type {}: {}'.format(p['type'], p))
|
||||
|
||||
entries = []
|
||||
for k in j['payload']['threads']:
|
||||
if k['thread_type'] == 1:
|
||||
if k['other_user_fbid'] not in participants:
|
||||
raise Exception('A thread was not in participants: {}'.format(j['payload']))
|
||||
raise FBchatException('The thread {} was not in participants: {}'.format(k, j['payload']))
|
||||
participants[k['other_user_fbid']].message_count = k['message_count']
|
||||
entries.append(participants[k['other_user_fbid']])
|
||||
elif k['thread_type'] == 2:
|
||||
entries.append(Group(k['thread_fbid'], participants=[p.strip('fbid:') for p in k['participants']], photo=k['image_src'], name=k['name']))
|
||||
entries.append(Group(k['thread_fbid'], participants=set([p.strip('fbid:') for p in k['participants']]), photo=k['image_src'], name=k['name'], message_count=k['message_count']))
|
||||
else:
|
||||
raise Exception('A thread had an unknown thread type: {}'.format(k))
|
||||
raise FBchatException('A thread had an unknown thread type: {}'.format(k))
|
||||
|
||||
return entries
|
||||
|
||||
@@ -747,7 +801,7 @@ class Client(object):
|
||||
.. todo::
|
||||
Documenting this
|
||||
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
form = {
|
||||
'client': 'mercury_sync',
|
||||
@@ -756,7 +810,7 @@ class Client(object):
|
||||
# 'last_action_timestamp': 0
|
||||
}
|
||||
|
||||
j = checkRequest(self._post(ReqUrl.THREAD_SYNC, form))
|
||||
j = self._post(self.req_url.THREAD_SYNC, form, fix_request=True, as_json=True)
|
||||
|
||||
return {
|
||||
"message_counts": j['payload']['message_counts'],
|
||||
@@ -815,7 +869,7 @@ class Client(object):
|
||||
|
||||
def _doSendRequest(self, data):
|
||||
"""Sends the data to `SendURL`, and returns the message ID or None on failure"""
|
||||
j = checkRequest(self._post(ReqUrl.SEND, data))
|
||||
j = self._post(self.req_url.SEND, data, fix_request=True, as_json=True)
|
||||
|
||||
try:
|
||||
message_ids = [action['message_id'] for action in j['payload']['actions'] if 'message_id' in action]
|
||||
@@ -823,7 +877,14 @@ class Client(object):
|
||||
log.warning("Got multiple message ids' back: {}".format(message_ids))
|
||||
message_id = message_ids[0]
|
||||
except (KeyError, IndexError) as e:
|
||||
raise Exception('Error when sending message: No message IDs could be found: {}'.format(j))
|
||||
raise FBchatException('Error when sending message: No message IDs could be found: {}'.format(j))
|
||||
|
||||
# update JS token if received in response
|
||||
if j.get('jsmods') is not None and j['jsmods'].get('require') is not None:
|
||||
try:
|
||||
self.payloadDefault['fb_dtsg'] = j['jsmods']['require'][0][3][0]
|
||||
except (KeyError, IndexError) as e:
|
||||
log.warning('Error when updating fb_dtsg. Facebook might have changed protocol!')
|
||||
|
||||
return message_id
|
||||
|
||||
@@ -836,7 +897,7 @@ class Client(object):
|
||||
:param thread_type: See :ref:`intro_threads`
|
||||
:type thread_type: models.ThreadType
|
||||
:return: :ref:`Message ID <intro_message_ids>` of the sent message
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
thread_id, thread_type = self._getThread(thread_id, thread_type)
|
||||
data = self._getSendData(thread_id, thread_type)
|
||||
@@ -860,7 +921,7 @@ class Client(object):
|
||||
:type size: models.EmojiSize
|
||||
:type thread_type: models.ThreadType
|
||||
:return: :ref:`Message ID <intro_message_ids>` of the sent emoji
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
thread_id, thread_type = self._getThread(thread_id, thread_type)
|
||||
data = self._getSendData(thread_id, thread_type)
|
||||
@@ -880,13 +941,13 @@ class Client(object):
|
||||
def _uploadImage(self, image_path, data, mimetype):
|
||||
"""Upload an image and get the image_id for sending in a message"""
|
||||
|
||||
j = checkRequest(self._postFile(ReqUrl.UPLOAD, {
|
||||
j = self._postFile(self.req_url.UPLOAD, {
|
||||
'file': (
|
||||
image_path,
|
||||
data,
|
||||
mimetype
|
||||
)
|
||||
}))
|
||||
}, fix_request=True, as_json=True)
|
||||
# Return the image_id
|
||||
return j['payload']['metadata'][0]['image_id']
|
||||
|
||||
@@ -900,7 +961,7 @@ class Client(object):
|
||||
:param thread_type: See :ref:`intro_threads`
|
||||
:type thread_type: models.ThreadType
|
||||
:return: :ref:`Message ID <intro_message_ids>` of the sent image
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
thread_id, thread_type = self._getThread(thread_id, thread_type)
|
||||
data = self._getSendData(thread_id, thread_type)
|
||||
@@ -925,7 +986,7 @@ class Client(object):
|
||||
:param thread_type: See :ref:`intro_threads`
|
||||
:type thread_type: models.ThreadType
|
||||
:return: :ref:`Message ID <intro_message_ids>` of the sent image
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
thread_id, thread_type = self._getThread(thread_id, thread_type)
|
||||
mimetype = guess_type(image_url)[0]
|
||||
@@ -937,13 +998,13 @@ class Client(object):
|
||||
"""
|
||||
Sends a local image to a thread
|
||||
|
||||
:param image_path: URL of an image to upload and send
|
||||
:param image_path: Path of an image to upload and send
|
||||
:param message: Additional message
|
||||
:param thread_id: User/Group ID to send to. See :ref:`intro_threads`
|
||||
:param thread_type: See :ref:`intro_threads`
|
||||
:type thread_type: models.ThreadType
|
||||
:return: :ref:`Message ID <intro_message_ids>` of the sent image
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
thread_id, thread_type = self._getThread(thread_id, thread_type)
|
||||
mimetype = guess_type(image_path)[0]
|
||||
@@ -958,7 +1019,7 @@ class Client(object):
|
||||
:param thread_id: Group ID to add people to. See :ref:`intro_threads`
|
||||
:type user_ids: list
|
||||
:return: :ref:`Message ID <intro_message_ids>` of the executed action
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
thread_id, thread_type = self._getThread(thread_id, None)
|
||||
data = self._getSendData(thread_id, ThreadType.GROUP)
|
||||
@@ -974,7 +1035,7 @@ class Client(object):
|
||||
|
||||
for i, user_id in enumerate(user_ids):
|
||||
if user_id == self.uid:
|
||||
raise Exception('Error when adding users: Cannot add self to group thread')
|
||||
raise FBchatUserError('Error when adding users: Cannot add self to group thread')
|
||||
else:
|
||||
data['log_message_data[added_participants][' + str(i) + ']'] = "fbid:" + str(user_id)
|
||||
|
||||
@@ -986,7 +1047,7 @@ class Client(object):
|
||||
|
||||
:param user_id: User ID to remove
|
||||
:param thread_id: Group ID to remove people from. See :ref:`intro_threads`
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
|
||||
thread_id, thread_type = self._getThread(thread_id, None)
|
||||
@@ -996,7 +1057,7 @@ class Client(object):
|
||||
"tid": thread_id
|
||||
}
|
||||
|
||||
j = checkRequest(self._post(ReqUrl.REMOVE_USER, data))
|
||||
j = self._post(self.req_url.REMOVE_USER, data, fix_request=True, as_json=True)
|
||||
|
||||
def changeThreadTitle(self, title, thread_id=None, thread_type=ThreadType.USER):
|
||||
"""
|
||||
@@ -1007,8 +1068,9 @@ class Client(object):
|
||||
:param thread_id: Group ID to change title of. See :ref:`intro_threads`
|
||||
:param thread_type: See :ref:`intro_threads`
|
||||
:type thread_type: models.ThreadType
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
|
||||
thread_id, thread_type = self._getThread(thread_id, thread_type)
|
||||
|
||||
if thread_type == ThreadType.USER:
|
||||
@@ -1021,6 +1083,8 @@ class Client(object):
|
||||
data['log_message_data[name]'] = title
|
||||
data['log_message_type'] = 'log:thread-name'
|
||||
|
||||
return self._doSendRequest(data)
|
||||
|
||||
def changeNickname(self, nickname, user_id, thread_id=None, thread_type=ThreadType.USER):
|
||||
"""
|
||||
Changes the nickname of a user in a thread
|
||||
@@ -1030,7 +1094,7 @@ class Client(object):
|
||||
:param thread_id: User/Group ID to change color of. See :ref:`intro_threads`
|
||||
:param thread_type: See :ref:`intro_threads`
|
||||
:type thread_type: models.ThreadType
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
thread_id, thread_type = self._getThread(thread_id, thread_type)
|
||||
|
||||
@@ -1040,7 +1104,7 @@ class Client(object):
|
||||
'thread_or_other_fbid': thread_id
|
||||
}
|
||||
|
||||
j = checkRequest(self._post(ReqUrl.THREAD_NICKNAME, data))
|
||||
j = self._post(self.req_url.THREAD_NICKNAME, data, fix_request=True, as_json=True)
|
||||
|
||||
def changeThreadColor(self, color, thread_id=None):
|
||||
"""
|
||||
@@ -1049,7 +1113,7 @@ class Client(object):
|
||||
:param color: New thread color
|
||||
:param thread_id: User/Group ID to change color of. See :ref:`intro_threads`
|
||||
:type color: models.ThreadColor
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
thread_id, thread_type = self._getThread(thread_id, None)
|
||||
|
||||
@@ -1058,7 +1122,7 @@ class Client(object):
|
||||
'thread_or_other_fbid': thread_id
|
||||
}
|
||||
|
||||
j = checkRequest(self._post(ReqUrl.THREAD_COLOR, data))
|
||||
j = self._post(self.req_url.THREAD_COLOR, data, fix_request=True, as_json=True)
|
||||
|
||||
def changeThreadEmoji(self, emoji, thread_id=None):
|
||||
"""
|
||||
@@ -1068,7 +1132,7 @@ class Client(object):
|
||||
|
||||
:param color: New thread emoji
|
||||
:param thread_id: User/Group ID to change emoji of. See :ref:`intro_threads`
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
thread_id, thread_type = self._getThread(thread_id, None)
|
||||
|
||||
@@ -1077,7 +1141,7 @@ class Client(object):
|
||||
'thread_or_other_fbid': thread_id
|
||||
}
|
||||
|
||||
j = checkRequest(self._post(ReqUrl.THREAD_EMOJI, data))
|
||||
j = self._post(self.req_url.THREAD_EMOJI, data, fix_request=True, as_json=True)
|
||||
|
||||
def reactToMessage(self, message_id, reaction):
|
||||
"""
|
||||
@@ -1086,7 +1150,7 @@ class Client(object):
|
||||
:param message_id: :ref:`Message ID <intro_message_ids>` to react to
|
||||
:param reaction: Reaction emoji to use
|
||||
:type reaction: models.MessageReaction
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
full_data = {
|
||||
"doc_id": 1491398900900362,
|
||||
@@ -1109,7 +1173,7 @@ class Client(object):
|
||||
.replace('u%27', '%27')\
|
||||
.replace('%5CU{}'.format(MessageReactionFix[reaction.value][0]), MessageReactionFix[reaction.value][1])
|
||||
|
||||
j = checkRequest(self._post('{}/?{}'.format(ReqUrl.MESSAGE_REACTION, url_part)))
|
||||
j = self._post('{}/?{}'.format(self.req_url.MESSAGE_REACTION, url_part), fix_request=True, as_json=True)
|
||||
|
||||
def setTypingStatus(self, status, thread_id=None, thread_type=None):
|
||||
"""
|
||||
@@ -1120,7 +1184,7 @@ class Client(object):
|
||||
:param thread_type: See :ref:`intro_threads`
|
||||
:type status: models.TypingStatus
|
||||
:type thread_type: models.ThreadType
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
thread_id, thread_type = self._getThread(thread_id, None)
|
||||
|
||||
@@ -1131,7 +1195,7 @@ class Client(object):
|
||||
"source": "mercury-chat"
|
||||
}
|
||||
|
||||
j = checkRequest(self._post(ReqUrl.TYPING, data))
|
||||
j = self._post(self.req_url.TYPING, data, fix_request=True, as_json=True)
|
||||
|
||||
"""
|
||||
END SEND METHODS
|
||||
@@ -1147,7 +1211,7 @@ class Client(object):
|
||||
"thread_ids[%s][0]" % userID: threadID
|
||||
}
|
||||
|
||||
r = self._post(ReqUrl.DELIVERED, data)
|
||||
r = self._post(self.req_url.DELIVERED, data)
|
||||
return r.ok
|
||||
|
||||
def markAsRead(self, userID):
|
||||
@@ -1161,7 +1225,7 @@ class Client(object):
|
||||
"ids[%s]" % userID: True
|
||||
}
|
||||
|
||||
r = self._post(ReqUrl.READ_STATUS, data)
|
||||
r = self._post(self.req_url.READ_STATUS, data)
|
||||
return r.ok
|
||||
|
||||
def markAsSeen(self):
|
||||
@@ -1169,7 +1233,7 @@ class Client(object):
|
||||
.. todo::
|
||||
Documenting this
|
||||
"""
|
||||
r = self._post(ReqUrl.MARK_SEEN, {"seen_timestamp": 0})
|
||||
r = self._post(self.req_url.MARK_SEEN, {"seen_timestamp": 0})
|
||||
return r.ok
|
||||
|
||||
def friendConnect(self, friend_id):
|
||||
@@ -1182,7 +1246,7 @@ class Client(object):
|
||||
"action": "confirm"
|
||||
}
|
||||
|
||||
r = self._post(ReqUrl.CONNECT, data)
|
||||
r = self._post(self.req_url.CONNECT, data)
|
||||
return r.ok
|
||||
|
||||
|
||||
@@ -1202,7 +1266,7 @@ class Client(object):
|
||||
'viewer_uid': self.uid,
|
||||
'state': 'active'
|
||||
}
|
||||
checkRequest(self._get(ReqUrl.PING, data), do_json_check=False)
|
||||
self._get(self.req_url.PING, data, fix_request=True, as_json=False)
|
||||
|
||||
def _fetchSticky(self):
|
||||
"""Call pull api to get sticky and pool parameter, newer api needs these parameters to work"""
|
||||
@@ -1213,10 +1277,10 @@ class Client(object):
|
||||
"clientid": self.client_id
|
||||
}
|
||||
|
||||
j = checkRequest(self._get(ReqUrl.STICKY, data))
|
||||
j = self._get(self.req_url.STICKY, data, fix_request=True, as_json=True)
|
||||
|
||||
if 'lb_info' not in j:
|
||||
raise Exception('Missing lb_info')
|
||||
if j.get('lb_info') is None:
|
||||
raise FBchatException('Missing lb_info: {}'.format(j))
|
||||
|
||||
return j['lb_info']['sticky'], j['lb_info']['pool']
|
||||
|
||||
@@ -1230,7 +1294,7 @@ class Client(object):
|
||||
"clientid": self.client_id,
|
||||
}
|
||||
|
||||
j = checkRequest(self._get(ReqUrl.STICKY, data))
|
||||
j = self._get(ReqUrl.STICKY, data, fix_request=True, as_json=True)
|
||||
|
||||
self.seq = j.get('seq', '0')
|
||||
return j
|
||||
@@ -1273,7 +1337,6 @@ class Client(object):
|
||||
thread_id = str(metadata['threadKey']['threadFbId'])
|
||||
self.onPeopleAdded(mid=mid, added_ids=added_ids, author_id=author_id, thread_id=thread_id,
|
||||
ts=ts, msg=m)
|
||||
continue
|
||||
|
||||
# Left/removed participants
|
||||
elif 'leftParticipantFbId' in delta:
|
||||
@@ -1281,15 +1344,13 @@ class Client(object):
|
||||
thread_id = str(metadata['threadKey']['threadFbId'])
|
||||
self.onPersonRemoved(mid=mid, removed_id=removed_id, author_id=author_id, thread_id=thread_id,
|
||||
ts=ts, msg=m)
|
||||
continue
|
||||
|
||||
# Color change
|
||||
elif delta_type == "change_thread_theme":
|
||||
new_color = ThreadColor(delta["untypedData"]["theme_color"])
|
||||
new_color = graphql_color_to_enum(delta["untypedData"]["theme_color"])
|
||||
thread_id, thread_type = getThreadIdAndThreadType(metadata)
|
||||
self.onColorChange(mid=mid, author_id=author_id, new_color=new_color, thread_id=thread_id,
|
||||
thread_type=thread_type, ts=ts, metadata=metadata, msg=m)
|
||||
continue
|
||||
|
||||
# Emoji change
|
||||
elif delta_type == "change_thread_icon":
|
||||
@@ -1297,7 +1358,6 @@ class Client(object):
|
||||
thread_id, thread_type = getThreadIdAndThreadType(metadata)
|
||||
self.onEmojiChange(mid=mid, author_id=author_id, new_emoji=new_emoji, thread_id=thread_id,
|
||||
thread_type=thread_type, ts=ts, metadata=metadata, msg=m)
|
||||
continue
|
||||
|
||||
# Thread title change
|
||||
elif delta.get("class") == "ThreadName":
|
||||
@@ -1305,7 +1365,6 @@ class Client(object):
|
||||
thread_id, thread_type = getThreadIdAndThreadType(metadata)
|
||||
self.onTitleChange(mid=mid, author_id=author_id, new_title=new_title, thread_id=thread_id,
|
||||
thread_type=thread_type, ts=ts, metadata=metadata, msg=m)
|
||||
continue
|
||||
|
||||
# Nickname change
|
||||
elif delta_type == "change_thread_nickname":
|
||||
@@ -1315,7 +1374,6 @@ class Client(object):
|
||||
self.onNicknameChange(mid=mid, author_id=author_id, changed_for=changed_for,
|
||||
new_nickname=new_nickname,
|
||||
thread_id=thread_id, thread_type=thread_type, ts=ts, metadata=metadata, msg=m)
|
||||
continue
|
||||
|
||||
# Message delivered
|
||||
elif delta.get("class") == "DeliveryReceipt":
|
||||
@@ -1325,7 +1383,6 @@ class Client(object):
|
||||
thread_id, thread_type = getThreadIdAndThreadType(delta)
|
||||
self.onMessageDelivered(msg_ids=message_ids, delivered_for=delivered_for,
|
||||
thread_id=thread_id, thread_type=thread_type, ts=ts, metadata=metadata, msg=m)
|
||||
continue
|
||||
|
||||
# Message seen
|
||||
elif delta.get("class") == "ReadReceipt":
|
||||
@@ -1335,7 +1392,6 @@ class Client(object):
|
||||
thread_id, thread_type = getThreadIdAndThreadType(delta)
|
||||
self.onMessageSeen(seen_by=seen_by, thread_id=thread_id, thread_type=thread_type,
|
||||
seen_ts=seen_ts, ts=delivered_ts, metadata=metadata, msg=m)
|
||||
continue
|
||||
|
||||
# Messages marked as seen
|
||||
elif delta.get("class") == "MarkRead":
|
||||
@@ -1347,19 +1403,21 @@ class Client(object):
|
||||
threads = [getThreadIdAndThreadType({"threadKey": thr}) for thr in delta.get("threadKeys")]
|
||||
|
||||
# thread_id, thread_type = getThreadIdAndThreadType(delta)
|
||||
self.onMarkedSeen(threads=threads, seen_ts=seen_ts, delivered_ts=delivered_ts, metadata=delta, msg=m)
|
||||
continue
|
||||
self.onMarkedSeen(threads=threads, seen_ts=seen_ts, ts=delivered_ts, metadata=delta, msg=m)
|
||||
|
||||
# New message
|
||||
elif delta.get("class") == "NewMessage":
|
||||
message = delta.get('body', '')
|
||||
thread_id, thread_type = getThreadIdAndThreadType(metadata)
|
||||
self.onMessage(mid=mid, author_id=author_id, message=message,
|
||||
thread_id=thread_id, thread_type=thread_type, ts=ts, metadata=m, msg=m)
|
||||
continue
|
||||
thread_id=thread_id, thread_type=thread_type, ts=ts, metadata=metadata, msg=m)
|
||||
|
||||
# Unknown message type
|
||||
else:
|
||||
self.onUnknownMesssageType(msg=m)
|
||||
|
||||
# Inbox
|
||||
if mtype == "inbox":
|
||||
elif mtype == "inbox":
|
||||
self.onInbox(unseen=m["unseen"], unread=m["unread"], recent_unread=m["recent_unread"], msg=m)
|
||||
|
||||
# Typing
|
||||
@@ -1387,6 +1445,14 @@ class Client(object):
|
||||
elif mtype == "deltaflow":
|
||||
pass
|
||||
|
||||
# Chat timestamp
|
||||
elif mtype == "chatproxy-presence":
|
||||
buddylist = {}
|
||||
for _id in m.get('buddyList', {}):
|
||||
payload = m['buddyList'][_id]
|
||||
buddylist[_id] = payload.get('lat')
|
||||
self.onChatTimestamp(buddylist=buddylist, msg=m)
|
||||
|
||||
# Unknown message type
|
||||
else:
|
||||
self.onUnknownMesssageType(msg=m)
|
||||
@@ -1398,7 +1464,7 @@ class Client(object):
|
||||
"""
|
||||
Start listening from an external event loop
|
||||
|
||||
:raises: Exception if request failed
|
||||
:raises: FBchatException if request failed
|
||||
"""
|
||||
self.listening = True
|
||||
self.sticky, self.pool = self._fetchSticky()
|
||||
@@ -1414,18 +1480,27 @@ class Client(object):
|
||||
:rtype: bool
|
||||
"""
|
||||
try:
|
||||
if markAlive: self._ping(self.sticky, self.pool)
|
||||
try:
|
||||
content = self._pullMessage(self.sticky, self.pool)
|
||||
if content: self._parseMessage(content)
|
||||
except requests.exceptions.RequestException:
|
||||
pass
|
||||
except Exception as e:
|
||||
return self.onListenError(exception=e)
|
||||
if markAlive:
|
||||
self._ping(self.sticky, self.pool)
|
||||
content = self._pullMessage(self.sticky, self.pool)
|
||||
if content:
|
||||
self._parseMessage(content)
|
||||
except KeyboardInterrupt:
|
||||
return False
|
||||
except requests.exceptions.Timeout:
|
||||
except requests.Timeout:
|
||||
pass
|
||||
except requests.ConnectionError:
|
||||
# If the client has lost their internet connection, keep trying every 30 seconds
|
||||
time.sleep(30)
|
||||
except FBchatFacebookError as e:
|
||||
# Fix 502 and 503 pull errors
|
||||
if e.request_status_code in [502, 503]:
|
||||
self.req_url.change_pull_channel()
|
||||
self.startListening()
|
||||
else:
|
||||
raise e
|
||||
except Exception as e:
|
||||
return self.onListenError(exception=e)
|
||||
|
||||
return True
|
||||
|
||||
@@ -1467,7 +1542,7 @@ class Client(object):
|
||||
|
||||
def on2FACode(self):
|
||||
"""Called when a 2FA code is needed to progress"""
|
||||
input('Please enter your 2FA code --> ')
|
||||
return input('Please enter your 2FA code --> ')
|
||||
|
||||
def onLoggedIn(self, email=None):
|
||||
"""
|
||||
@@ -1486,8 +1561,10 @@ class Client(object):
|
||||
Called when an error was encountered while listening
|
||||
|
||||
:param exception: The exception that was encountered
|
||||
:return: Whether the loop should keep running
|
||||
"""
|
||||
traceback.print_exc()
|
||||
log.exception('Got exception while listening')
|
||||
return True
|
||||
|
||||
|
||||
def onMessage(self, mid=None, author_id=None, message=None, thread_id=None, thread_type=ThreadType.USER, ts=None, metadata=None, msg={}):
|
||||
@@ -1674,6 +1751,14 @@ class Client(object):
|
||||
"""
|
||||
pass
|
||||
|
||||
def onChatTimestamp(self, buddylist={}, msg={}):
|
||||
"""
|
||||
Called when the client receives chat online presence update
|
||||
|
||||
:param buddylist: A list of dicts with friend id and last seen timestamp
|
||||
:param msg: A full set of the data recieved
|
||||
"""
|
||||
log.debug('Chat Timestamps received: {}'.format(buddylist))
|
||||
|
||||
def onUnknownMesssageType(self, msg={}):
|
||||
"""
|
||||
|
@@ -26,10 +26,12 @@ class ConcatJSONDecoder(json.JSONDecoder):
|
||||
def graphql_color_to_enum(color):
|
||||
if color is None:
|
||||
return None
|
||||
if len(color) == 0:
|
||||
return ThreadColor.MESSENGER_BLUE
|
||||
try:
|
||||
return ThreadColor('#{}'.format(color[2:].lower()))
|
||||
except KeyError:
|
||||
raise Exception('Could not get ThreadColor from color: {}'.format(color))
|
||||
except ValueError:
|
||||
raise FBchatException('Could not get ThreadColor from color: {}'.format(color))
|
||||
|
||||
def get_customization_info(thread):
|
||||
if thread is None or thread.get('customization_info') is None:
|
||||
@@ -42,18 +44,21 @@ def get_customization_info(thread):
|
||||
}
|
||||
if thread.get('thread_type') == 'GROUP' or thread.get('is_group_thread') or thread.get('thread_key', {}).get('thread_fbid'):
|
||||
rtn['nicknames'] = {}
|
||||
for k in info['participant_customizations']:
|
||||
for k in info.get('participant_customizations', []):
|
||||
rtn['nicknames'][k['participant_id']] = k.get('nickname')
|
||||
else:
|
||||
_id = thread.get('thread_key', {}).get('other_user_id') or thread.get('id')
|
||||
if info['participant_customizations'][0]['participant_id'] == _id:
|
||||
rtn['nickname'] = info['participant_customizations'][0]
|
||||
rtn['own_nickname'] = info['participant_customizations'][1]
|
||||
elif info['participant_customizations'][1]['participant_id'] == _id:
|
||||
rtn['nickname'] = info['participant_customizations'][1]
|
||||
rtn['own_nickname'] = info['participant_customizations'][0]
|
||||
else:
|
||||
raise Exception('No participant matching the user {} found: {}'.format(_id, info['participant_customizations']))
|
||||
elif info.get('participant_customizations'):
|
||||
uid = thread.get('thread_key', {}).get('other_user_id') or thread.get('id')
|
||||
pc = info['participant_customizations']
|
||||
if len(pc) > 0:
|
||||
if pc[0].get('participant_id') == uid:
|
||||
rtn['nickname'] = pc[0].get('nickname')
|
||||
else:
|
||||
rtn['own_nickname'] = pc[0].get('nickname')
|
||||
if len(pc) > 1:
|
||||
if pc[1].get('participant_id') == uid:
|
||||
rtn['nickname'] = pc[1].get('nickname')
|
||||
else:
|
||||
rtn['own_nickname'] = pc[1].get('nickname')
|
||||
return rtn
|
||||
|
||||
def graphql_to_message(message):
|
||||
@@ -73,7 +78,8 @@ def graphql_to_message(message):
|
||||
text=message.get('message').get('text'),
|
||||
mentions=[Mention(m.get('entity', {}).get('id'), offset=m.get('offset'), length=m.get('length')) for m in message.get('message').get('ranges', [])],
|
||||
sticker=message.get('sticker'),
|
||||
attachments=message.get('blob_attachments')
|
||||
attachments=message.get('blob_attachments'),
|
||||
extensible_attachment=message.get('extensible_attachment')
|
||||
)
|
||||
|
||||
def graphql_to_user(user):
|
||||
@@ -93,7 +99,8 @@ def graphql_to_user(user):
|
||||
emoji=c_info.get('emoji'),
|
||||
own_nickname=c_info.get('own_nickname'),
|
||||
photo=user['profile_picture'].get('uri'),
|
||||
name=user.get('name')
|
||||
name=user.get('name'),
|
||||
message_count=user.get('messages_count')
|
||||
)
|
||||
|
||||
def graphql_to_group(group):
|
||||
@@ -102,12 +109,13 @@ def graphql_to_group(group):
|
||||
c_info = get_customization_info(group)
|
||||
return Group(
|
||||
group['thread_key']['thread_fbid'],
|
||||
participants=[node['messaging_actor']['id'] for node in group['all_participants']['nodes']],
|
||||
participants=set([node['messaging_actor']['id'] for node in group['all_participants']['nodes']]),
|
||||
nicknames=c_info.get('nicknames'),
|
||||
color=c_info.get('color'),
|
||||
emoji=c_info.get('emoji'),
|
||||
photo=group['image'].get('uri'),
|
||||
name=group.get('name')
|
||||
name=group.get('name'),
|
||||
message_count=group.get('messages_count')
|
||||
)
|
||||
|
||||
def graphql_to_page(page):
|
||||
@@ -121,7 +129,8 @@ def graphql_to_page(page):
|
||||
city=page.get('city').get('name'),
|
||||
category=page.get('category_type'),
|
||||
photo=page['profile_picture'].get('uri'),
|
||||
name=page.get('name')
|
||||
name=page.get('name'),
|
||||
message_count=page.get('messages_count')
|
||||
)
|
||||
|
||||
def graphql_queries_to_json(*queries):
|
||||
@@ -134,7 +143,11 @@ def graphql_queries_to_json(*queries):
|
||||
return json.dumps(rtn)
|
||||
|
||||
def graphql_response_to_json(content):
|
||||
j = json.loads(content, cls=ConcatJSONDecoder)
|
||||
content = strip_to_json(content) # Usually only needed in some error cases
|
||||
try:
|
||||
j = json.loads(content, cls=ConcatJSONDecoder)
|
||||
except Exception:
|
||||
raise FBchatException('Error while parsing JSON: {}'.format(repr(content)))
|
||||
|
||||
rtn = [None]*(len(j))
|
||||
for x in j:
|
||||
@@ -167,7 +180,7 @@ class GraphQL(object):
|
||||
'query_params': params
|
||||
}
|
||||
else:
|
||||
raise Exception('A query or doc_id must be specified')
|
||||
raise FBchatUserError('A query or doc_id must be specified')
|
||||
|
||||
|
||||
FRAGMENT_USER = """
|
||||
|
@@ -4,6 +4,26 @@ from __future__ import unicode_literals
|
||||
import enum
|
||||
|
||||
|
||||
class FBchatException(Exception):
|
||||
"""Custom exception thrown by fbchat. All exceptions in the fbchat module inherits this"""
|
||||
|
||||
class FBchatFacebookError(FBchatException):
|
||||
#: The error code that Facebook returned
|
||||
fb_error_code = str
|
||||
#: The error message that Facebook returned (In the user's own language)
|
||||
fb_error_message = str
|
||||
#: The status code that was sent in the http response (eg. 404) (Usually only set if not successful, aka. not 200)
|
||||
request_status_code = int
|
||||
def __init__(self, message, fb_error_code=None, fb_error_message=None, request_status_code=None):
|
||||
super(FBchatFacebookError, self).__init__(message)
|
||||
"""Thrown by fbchat when Facebook returns an error"""
|
||||
self.fb_error_code = str(fb_error_code)
|
||||
self.fb_error_message = fb_error_message
|
||||
self.request_status_code = request_status_code
|
||||
|
||||
class FBchatUserError(FBchatException):
|
||||
"""Thrown by fbchat when wrong values are entered"""
|
||||
|
||||
class Thread(object):
|
||||
#: The unique identifier of the thread. Can be used a `thread_id`. See :ref:`intro_threads` for more info
|
||||
uid = str
|
||||
@@ -13,13 +33,18 @@ class Thread(object):
|
||||
photo = str
|
||||
#: The name of the thread
|
||||
name = str
|
||||
|
||||
def __init__(self, _type, uid, photo=None, name=None):
|
||||
#: Timestamp of last message
|
||||
last_message_timestamp = str
|
||||
#: Number of messages in the thread
|
||||
message_count = int
|
||||
def __init__(self, _type, uid, photo=None, name=None, last_message_timestamp=None, message_count=None):
|
||||
"""Represents a Facebook thread"""
|
||||
self.uid = str(uid)
|
||||
self.type = _type
|
||||
self.photo = photo
|
||||
self.name = name
|
||||
self.last_message_timestamp = last_message_timestamp
|
||||
self.message_count = message_count
|
||||
|
||||
def __repr__(self):
|
||||
return self.__unicode__()
|
||||
@@ -66,8 +91,8 @@ class User(Thread):
|
||||
|
||||
|
||||
class Group(Thread):
|
||||
#: List of the group thread's participant user IDs
|
||||
participants = list
|
||||
#: Unique list (set) of the group thread's participant user IDs
|
||||
participants = set
|
||||
#: Dict, containing user nicknames mapped to their IDs
|
||||
nicknames = dict
|
||||
#: A :class:`ThreadColor`. The groups's message color
|
||||
@@ -75,7 +100,7 @@ class Group(Thread):
|
||||
#: The groups's default emoji
|
||||
emoji = str
|
||||
|
||||
def __init__(self, uid, participants=[], nicknames=[], color=None, emoji=None, **kwargs):
|
||||
def __init__(self, uid, participants=set(), nicknames=[], color=None, emoji=None, **kwargs):
|
||||
"""Represents a Facebook group. Inherits `Thread`"""
|
||||
super(Group, self).__init__(ThreadType.GROUP, uid, **kwargs)
|
||||
self.participants = participants
|
||||
@@ -125,8 +150,10 @@ class Message(object):
|
||||
sticker = str
|
||||
#: A list of attachments
|
||||
attachments = list
|
||||
#: An extensible attachment, e.g. share object
|
||||
extensible_attachment = dict
|
||||
|
||||
def __init__(self, uid, author=None, timestamp=None, is_read=None, reactions=[], text=None, mentions=[], sticker=None, attachments=[]):
|
||||
def __init__(self, uid, author=None, timestamp=None, is_read=None, reactions=[], text=None, mentions=[], sticker=None, attachments=[], extensible_attachment={}):
|
||||
"""Represents a Facebook message"""
|
||||
self.uid = uid
|
||||
self.author = author
|
||||
@@ -137,6 +164,7 @@ class Message(object):
|
||||
self.mentions = mentions
|
||||
self.sticker = sticker
|
||||
self.attachments = attachments
|
||||
self.extensible_attachment = extensible_attachment
|
||||
|
||||
|
||||
class Mention(object):
|
||||
|
@@ -48,7 +48,7 @@ GENDERS = {
|
||||
11: 'unknown_plural',
|
||||
|
||||
# For graphql requests
|
||||
#'': 'unknown',
|
||||
'UNKNOWN': 'unknown',
|
||||
'FEMALE': 'female_singular',
|
||||
'MALE': 'male_singular',
|
||||
#'': 'female_singular_guess',
|
||||
@@ -93,6 +93,16 @@ class ReqUrl(object):
|
||||
TYPING = "https://www.facebook.com/ajax/messaging/typ.php"
|
||||
GRAPHQL = "https://www.facebook.com/api/graphqlbatch/"
|
||||
|
||||
pull_channel = 0
|
||||
|
||||
def change_pull_channel(self, channel=None):
|
||||
if channel is None:
|
||||
self.pull_channel = (self.pull_channel + 1) % 5 # Pull channel will be 0-4
|
||||
else:
|
||||
self.pull_channel = channel
|
||||
self.STICKY = "https://{}-edge-chat.facebook.com/pull".format(self.pull_channel)
|
||||
self.PING = "https://{}-edge-chat.facebook.com/active_ping".format(self.pull_channel)
|
||||
|
||||
|
||||
facebookEncoding = 'UTF-8'
|
||||
|
||||
@@ -103,7 +113,7 @@ def strip_to_json(text):
|
||||
try:
|
||||
return text[text.index('{'):]
|
||||
except ValueError:
|
||||
raise Exception('No JSON object found: {}, {}'.format(repr(text), text.index('{')))
|
||||
raise FBchatException('No JSON object found: {}, {}'.format(repr(text), text.index('{')))
|
||||
|
||||
def get_decoded_r(r):
|
||||
return get_decoded(r._content)
|
||||
@@ -143,30 +153,31 @@ def generateOfflineThreadingID():
|
||||
return str(int(msgs, 2))
|
||||
|
||||
def check_json(j):
|
||||
if 'error' in j and j['error'] is not None:
|
||||
if 'errorDescription' in j:
|
||||
# 'errorDescription' is in the users own language!
|
||||
raise Exception('Error #{} when sending request: {}'.format(j['error'], j['errorDescription']))
|
||||
elif 'debug_info' in j['error']:
|
||||
raise Exception('Error #{} when sending request: {}'.format(j['error']['code'], repr(j['error']['debug_info'])))
|
||||
else:
|
||||
raise Exception('Error {} when sending request'.format(j['error']))
|
||||
if j.get('error') is None:
|
||||
return
|
||||
if 'errorDescription' in j:
|
||||
# 'errorDescription' is in the users own language!
|
||||
raise FBchatFacebookError('Error #{} when sending request: {}'.format(j['error'], j['errorDescription']), fb_error_code=j['error'], fb_error_message=j['errorDescription'])
|
||||
elif 'debug_info' in j['error'] and 'code' in j['error']:
|
||||
raise FBchatFacebookError('Error #{} when sending request: {}'.format(j['error']['code'], repr(j['error']['debug_info'])), fb_error_code=j['error']['code'], fb_error_message=j['error']['debug_info'])
|
||||
else:
|
||||
raise FBchatFacebookError('Error {} when sending request'.format(j['error']), fb_error_code=j['error'])
|
||||
|
||||
def checkRequest(r, do_json_check=True):
|
||||
def check_request(r, as_json=True):
|
||||
if not r.ok:
|
||||
raise Exception('Error when sending request: Got {} response'.format(r.status_code))
|
||||
raise FBchatFacebookError('Error when sending request: Got {} response'.format(r.status_code), request_status_code=r.status_code)
|
||||
|
||||
content = get_decoded_r(r)
|
||||
|
||||
if content is None or len(content) == 0:
|
||||
raise Exception('Error when sending request: Got empty response')
|
||||
raise FBchatFacebookError('Error when sending request: Got empty response')
|
||||
|
||||
if do_json_check:
|
||||
if as_json:
|
||||
content = strip_to_json(content)
|
||||
try:
|
||||
j = json.loads(content)
|
||||
except Exception as e:
|
||||
raise Exception('Error while parsing JSON: {}'.format(repr(content)), e)
|
||||
except ValueError:
|
||||
raise FBchatFacebookError('Error while parsing JSON: {}'.format(repr(content)))
|
||||
check_json(j)
|
||||
return j
|
||||
else:
|
||||
|
@@ -1,3 +1,4 @@
|
||||
requests
|
||||
lxml
|
||||
beautifulsoup4
|
||||
enum34; python_version == '2.7'
|
||||
|
2
setup.py
2
setup.py
@@ -18,7 +18,7 @@ with open('README.rst') as f:
|
||||
|
||||
try:
|
||||
requirements = [line.rstrip('\n') for line in open(os.path.join('fbchat.egg-info', 'requires.txt'))]
|
||||
except FileNotFoundError:
|
||||
except IOError:
|
||||
requirements = [line.rstrip('\n') for line in open('requirements.txt')]
|
||||
|
||||
version = None
|
||||
|
Reference in New Issue
Block a user