def action_card():
message = Message()
message.add_card(
Card(
CardHeader(
title="Timmy's Action Menu",
subtitle="Top tips for Timmy Timesheets",
image_url='https://timesheets.servian.fun/static/img/timmy.2fafd88.png',
image_style=ImageStyle.AVATAR),
Section(
ButtonList(
TextButton(text="VIEW LAST WEEK'S TIMESHEET").add_action(
action_method=ActionMethod.SHOW_LAST_WEEKS_TIMESHEET)),
TextParagraph("Show last week's timesheet")),
Section(
ButtonList(
TextButton(text="VIEW THIS WEEK'S TIMESHEET").add_action(
action_method=ActionMethod.SHOW_THIS_WEEKS_TIMESHEET)),
TextParagraph("Show this week's timesheet")),
Section(
ButtonList(
TextButton(text="PROPOSE THIS WEEK'S TIMESHEET").add_action(
action_method=ActionMethod.PROPOSE_TIMESHEET)),
TextParagraph(
"Show Timmy's Best Guessâ„¢ at this week's timesheet. "
"You also have the option to make it real by updating your timesheet "
"right from this here chat. Rock on 🤘")),
Section(
TextParagraph('Bring up this menu anytime by typing <b>help</b>')
)
))
return message
def handle_card_clicked(self, action_method, action_parameters, event):
if action_method == self.ActionMethod.COPY_TIMESHEET:
self.handle_event_asynchronously(event)
msg = MessageTemplate.timesheet_updated_successfully().output()
msg['actionResponse'] = {'type': 'UPDATE_MESSAGE'}
return msg
elif action_method == self.ActionMethod.HELP:
msg = MessageTemplate.action_card().output()
msg['actionResponse'] = {'type': 'UPDATE_MESSAGE'}
return msg
elif action_method == self.ActionMethod.SHOW_THIS_WEEKS_TIMESHEET:
self.handle_event_asynchronously(event)
return Message(
text="I'm off to track down this week's timesheet... ⌛",
response_type=ResponseType.UPDATE_MESSAGE).output()
elif action_method == self.ActionMethod.SHOW_LAST_WEEKS_TIMESHEET:
self.handle_event_asynchronously(event)
return Message(
text="I'm off to track down last week's timesheet... ⌛",
response_type=ResponseType.UPDATE_MESSAGE).output()
elif action_method == self.ActionMethod.PROPOSE_TIMESHEET:
self.handle_event_asynchronously(event)
return Message(
text="Thinking cap is on! I'm off to divine this week's timesheet... ⌛",
response_type=ResponseType.UPDATE_MESSAGE).output()
elif action_method == self.ActionMethod.REFRESH_TIMESHEET_DATA:
self.handle_event_asynchronously(event)
return Message(
text="Getting latest timesheet data... ⌛",
response_type=ResponseType.UPDATE_MESSAGE).output()
def simple_text_card(text, show_menu_button=False, button_label="BACK"):
message = Message()
card = Card(Section(TextParagraph(text)))
if show_menu_button:
card.add_section(
Section(ButtonList(TextButton(button_label).add_action(ActionMethod.HELP))))
message.add_card(card)
return message
def timesheet_updated_successfully():
message = Message(action_response=Message.ResponseType.UPDATE_MESSAGE)
message.add_card(
Card(
Section(
TextParagraph('Timesheet updated sucessfully 🎉'),
ButtonList(
TextButton('VIEW ON TIMESHEETS.COM.AU').add_link('https://timesheets.com.au/login.asp'))),
Section(
ButtonList(
TextButton('BACK').add_action(ActionMethod.HELP)))))
return message
def handle_message(self, message, event):
message_text = message['text'].lower()
user_name = event['user']['name']
user = self.authenticate_user(user_name)
if not user:
return self.request_config(event)
if message_text == 'help':
return MessageTemplate.action_card().output()
elif message_text in ['remind_everyone', 'remind_everyone_meme']:
self.handle_event_asynchronously(event)
return Message(text='Sending a reminder to everyone!').output()
elif message_text == 'logout':
user.delete()
return Message(text="You've been logged out and your TimePro credentials deleted.").output()
def process_refresh_timesheet_data(self, parameters, event):
start_date, end_date = utils.get_week_dates(weeks=-3, week_span=4)
user = models.User.get(event['user']['name'])
date_entries = self._get_timesheet_date_entries(
start_date, end_date, user)
models.Timesheet.bulk_create_from_date_entries(user=user, date_entries=date_entries)
message = Message(text="Timesheet data refreshed!").output()
return message
def timesheet_card(date_entries, user, title='Timesheet Summary', show_buttons=False):
start_date = min(date_entries.keys())
end_date = max(date_entries.keys())
start_date_label = start_date.strftime('%d-%b-%Y')
end_date_label = end_date.strftime('%d-%b-%Y')
message = Message()
card = Card(CardHeader(
title=title,
subtitle=f'{start_date_label} - {end_date_label}',
image_url=user.picture,
image_style=ImageStyle.AVATAR))
# Loop through timesheet and construct widget_list
widget_list = list()
for date, entries in date_entries.items():
date_label = date.strftime('%A, %d %B')
widget_list.append(TextParagraph(f'<b>{date_label}</b>'))
if not entries:
widget_list.append(KeyValue(top_label='Incomplete', content='0.0', icon=Icon.CLOCK))
for entry in entries:
widget_list.append(KeyValue(
top_label=entry.get('customer_description'),
content=str(entry.get('hours')),
bottom_label=entry.get('project_description'),
icon=Icon.CLOCK))
timesheet_section = Section(*widget_list)
card.add_section(timesheet_section)
if show_buttons:
action_parameters = {
'start_date': str(start_date),
'end_date': str(end_date)}
button_section = Section(
ButtonList(
TextButton('SAVE TIMESHEET').add_action(
action_method=ActionMethod.COPY_TIMESHEET,
parameters=action_parameters)))
card.add_section(button_section)
back_button = Section(
ButtonList(
TextButton('BACK').add_action(ActionMethod.HELP)))
card.add_section(back_button)
message.add_card(card)
return message
def test_request_config():
message = Message.request_config('https://example.com')
expected = {
'actionResponse': {
'type': 'REQUEST_CONFIG',
'url': 'https://example.com'
}
}
assert message.output() == expected
def handle_added_to_space(self, space_type, event):
# Register new space in DB
space = models.Space(
hash_key=event['space']['name'],
type=event['space']['type'],
username=event['user']['name'])
space.save()
if space_type == SpaceType.DIRECT_MESSAGE:
return Message(text='Hey, thanks for adding me 🎉! Type *help* to get started...').output()
def test_pizza_bot_full_example(pizza_bot_message):
message = Message()
message.add_card(
Card(
CardHeader(
title='Pizza Bot Customer Support',
subtitle='*****@*****.**',
image_url='https://goo.gl/aeDtrS'),
Section(
KeyValue(top_label='Order No.', content='12345'),
KeyValue(top_label='Status', content='In Delivery')),
Section(
'Location',
Image(image_url='https://maps.googleapis.com/...')),
Section(
ButtonList(
TextButton(text='OPEN ORDER').add_link(url='https://example.com/orders/...')))))
assert message.output() == pizza_bot_message
def request_config(self, event):
client_secret = self.parameters.get('google_auth_client_secret')
callback_url = self.parameters.get('config_callback_url')
state = {
'user_name': event['user']['name'],
'user_email': event['user']['email'],
'space_name': event['space']['name'],
'thread_name': event['message']['thread']['name'],
'redirect_url': event['configCompleteRedirectUrl']
}
oauth2_url = auth.get_authorization_url(callback_url, state, client_secret)
return Message.request_config(url=oauth2_url).output()
def handle_message(self, message, event):
message_text = message['text'].lower()
if message_text in ['remind_everyone', 'remind_everyone_meme']:
spaces = models.Space.scan()
for space in spaces:
user = models.User.get(space.username)
if message_text == 'remind_everyone':
message = Message(text=(
f"Hey {user.given_name}! Just a friendly reminder "
"that it's time to do your timesheet 😄")).output()
elif message_text == 'remind_everyone_meme':
message = MessageTemplate.reminder_meme().output()
self.chat.create_message(message=message, space_name=space.name)
def test_text_message():
message = Message(text='hello')
assert message.output() == {'text': 'hello'}
def reminder_meme():
random_integer = randint(0, 4) + 1
meme_url = f'https://timesheets.servian.fun/images/{random_integer}.jpg'
message = Message()
message.add_card(Card(Section(Image(image_url=meme_url))))
return message