def test_client_send(self, mock_post):
mock_post.return_value.status_code = 200
mock_post.return_value.json.return_value = {
'success': True
}
messenger = MessengerClient(access_token='1234')
recipient = messages.Recipient(recipient_id='123')
message = messages.Message(text='Hello World')
request = messages.MessageRequest(recipient, message)
messenger.send(request)
mock_post.assert_called_with(
'https://graph.facebook.com/v2.6/me/messages',
json={
"message": {"text": "Hello World"},
"recipient": {"id": "123"}
},
params={'access_token': '1234'}
)
def test_client_send(self, mock_post):
mock_post.return_value.status_code = 200
mock_post.return_value.json.return_value = {
'success': True
}
messenger = MessengerClient(access_token='1234')
recipient = messages.Recipient(recipient_id='123')
message = messages.Message(text='Hello World')
request = messages.MessageRequest(recipient, message)
messenger.send(request)
mock_post.assert_called_with(
'https://graph.facebook.com/v2.6/me/messages',
json={
"message": {"text": "Hello World"},
"recipient": {"id": "123"}
},
params={'access_token': '1234'}
)
class MessengerBot(IntegrationBot):
"""
Facebook Messenger integration.
Permabots only uses Page Access Token but webhook is not set. It mus be set manually in Facebook dev platform using
UUID generated as id of the messenger bot after creation in Permabots.
This bot is used to Verify Token and generate url https://domain/processing/messengerbot/permabots_messenger_bot_id/
Read Messenger documentation `<https://developers.facebook.com/docs/messenger-platform/quickstart>` _.
"""
token = models.CharField(_('Messenger Token'),
max_length=512,
db_index=True)
class Meta:
verbose_name = _('Messenger Bot')
verbose_name_plural = _('Messenger Bots')
def __init__(self, *args, **kwargs):
super(MessengerBot, self).__init__(*args, **kwargs)
self._bot = None
self.webhook = False
if self.token:
self.init_bot()
def __str__(self):
return "%s" % self.token
def __repr__(self):
return "(%s, %s)" % (self.id, self.token)
def init_bot(self):
self._bot = MessengerClient(self.token)
def set_webhook(self, url):
# Url is set in facebook dashboard. Just subscribe
self._bot.subscribe_app()
@property
def hook_url(self):
return 'permabots:messengerbot'
@property
def hook_id(self):
return str(self.id)
@property
def null_url(self):
# not used
return "https://example.com"
@property
def identity(self):
return 'messenger'
def message_text(self, message):
return message.data
def get_chat_state(self, message):
try:
return MessengerChatState.objects.select_related('state').get(
chat=message.sender, state__bot=self.bot)
except MessengerChatState.DoesNotExist:
return None
def build_keyboard(self, keyboard):
def traverse(o, tree_types=(list, tuple)):
if isinstance(o, tree_types):
for value in o:
for subvalue in traverse(value, tree_types):
yield subvalue
else:
yield o
built_keyboard = None
if keyboard:
# same payload as title
built_keyboard = [
PostbackButton(element[0:20], element)
for element in traverse(ast.literal_eval(keyboard))
]
return built_keyboard
def create_chat_state(self, message, target_state, context):
MessengerChatState.objects.create(chat=message.sender,
state=target_state,
ctx=context)
def get_chat_id(self, message):
return message.sender
def send_message(self,
chat_id,
text,
keyboard,
reply_message=None,
user=None):
texts = text.strip().split('\\n')
msgs = []
for txt in texts:
for chunk in textwrap.wrap(txt, 320):
msgs.append(messages.Message(text=chunk))
if keyboard:
if len(msgs[-1].text) <= 45:
title = msgs.pop().text
else:
new_texts = textwrap.wrap(msgs[-1].text, 45)
msgs[-1].text = " ".join(new_texts[:-1])
title = new_texts[-1]
elements = []
for chunk_buttons, last in self.batch(keyboard[0:30], 3):
elements.append(Element(title=title, buttons=chunk_buttons))
generic_template = GenericTemplate(elements)
attachment = TemplateAttachment(generic_template)
msgs.append(messages.Message(attachment=attachment))
for msg in msgs:
try:
logger.debug("Message to send:(%s)" % msg.to_dict())
recipient = messages.Recipient(recipient_id=chat_id)
self._bot.send(messages.MessageRequest(recipient, msg))
logger.debug("Message sent OK:(%s)" % msg.to_dict())
except:
exctype, value = sys.exc_info()[:2]
logger.error("Error trying to send message:(%s): %s:%s" %
(msg.to_dict(), exctype, value))
class MessengerBot(IntegrationBot):
"""
Facebook Messenger integration.
Permabots only uses Page Access Token but webhook is not set. It mus be set manually in Facebook dev platform using
UUID generated as id of the messenger bot after creation in Permabots.
This bot is used to Verify Token and generate url https://domain/processing/messengerbot/permabots_messenger_bot_id/
Read Messenger documentation `<https://developers.facebook.com/docs/messenger-platform/quickstart>` _.
"""
token = models.CharField(_('Messenger Token'), max_length=512, db_index=True)
class Meta:
verbose_name = _('Messenger Bot')
verbose_name_plural = _('Messenger Bots')
def __init__(self, *args, **kwargs):
super(MessengerBot, self).__init__(*args, **kwargs)
self._bot = None
self.webhook = False
if self.token:
self.init_bot()
def __str__(self):
return "%s" % self.token
def __repr__(self):
return "(%s, %s)" % (self.id, self.token)
def init_bot(self):
self._bot = MessengerClient(self.token)
def set_webhook(self, url):
# Url is set in facebook dashboard. Just subscribe
self._bot.subscribe_app()
@property
def hook_url(self):
return 'permabots:messengerbot'
@property
def hook_id(self):
return str(self.id)
@property
def null_url(self):
# not used
return "https://example.com"
@property
def identity(self):
return 'messenger'
def message_text(self, message):
return message.data
def get_chat_state(self, message):
try:
return MessengerChatState.objects.select_related('state').get(chat=message.sender, state__bot=self.bot)
except MessengerChatState.DoesNotExist:
return None
def build_keyboard(self, keyboard):
def traverse(o, tree_types=(list, tuple)):
if isinstance(o, tree_types):
for value in o:
for subvalue in traverse(value, tree_types):
yield subvalue
else:
yield o
built_keyboard = None
if keyboard:
# same payload as title
built_keyboard = [PostbackButton(element[0:20], element) for element in traverse(ast.literal_eval(keyboard))]
return built_keyboard
def create_chat_state(self, message, target_state, context):
MessengerChatState.objects.create(chat=message.sender,
state=target_state,
ctx=context)
def get_chat_id(self, message):
return message.sender
def send_message(self, chat_id, text, keyboard, reply_message=None, user=None):
texts = text.strip().split('\\n')
msgs = []
for txt in texts:
for chunk in textwrap.wrap(txt, 320):
msgs.append(messages.Message(text=chunk))
if keyboard:
if len(msgs[-1].text) <= 45:
title = msgs.pop().text
else:
new_texts = textwrap.wrap(msgs[-1].text, 45)
msgs[-1].text = " ".join(new_texts[:-1])
title = new_texts[-1]
elements = []
for chunk_buttons, last in self.batch(keyboard[0:30], 3):
elements.append(Element(title=title, buttons=chunk_buttons))
generic_template = GenericTemplate(elements)
attachment = TemplateAttachment(generic_template)
msgs.append(messages.Message(attachment=attachment))
for msg in msgs:
try:
logger.debug("Message to send:(%s)" % msg.to_dict())
recipient = messages.Recipient(recipient_id=chat_id)
self._bot.send(messages.MessageRequest(recipient, msg))
logger.debug("Message sent OK:(%s)" % msg.to_dict())
except:
exctype, value = sys.exc_info()[:2]
logger.error("Error trying to send message:(%s): %s:%s" % (msg.to_dict(), exctype, value))
class Replies(object):
def __init__(self, access_token, recipient_id):
self.access_token = access_token
self.recipient_id = recipient_id
self.messenger = MessengerClient(access_token=access_token)
@property
def recipient(self):
return messages.Recipient(recipient_id=self.recipient_id)
def reply_msg(self, text):
message = messages.Message(text=text)
request = messages.MessageRequest(self.recipient, message)
self.messenger.send(request)
def ask_where(self):
self.reply_msg('Xin chào! Bạn muốn ăn ở đâu?')
def give_suggestions(self, location, negations):
text = 'kết quả từ {district} nè'.format(district=location)
self.reply_msg(text)
results = quanbot.quan.Quan.search(location, negations)
if results:
elements = list(map(self._quan_element, results))
self._send_carousel(elements)
return True
else:
self.reply_msg("Không tìm thành công thì restart")
return False
def failed_read_location(self):
self.reply_msg('Vui lòng viết lại')
def say_restarting(self):
self.reply_msg('Rồi, hãy thử một lần nữa nha')
def _quan_element(self, qd):
web_button = elements.WebUrlButton(
title='xem trên Foody.vn',
url=qd['URL'],
)
postback_button = elements.PostbackButton(
title="Không " + qd['maindish'],
payload="Không " + qd['maindish'],
)
element = elements.Element(
title=qd['Name'],
item_url=qd['URL'],
image_url=qd['Pic'],
subtitle=qd['dish'],
buttons=[
# web_button,
postback_button,
],
)
return element
def _send_carousel(self, elements):
generic_template = templates.GenericTemplate(
elements=elements
)
attachment = attachments.TemplateAttachment(template=generic_template)
message = messages.Message(attachment=attachment)
request = messages.MessageRequest(self.recipient, message)
self.messenger.send(request)