-
Notifications
You must be signed in to change notification settings - Fork 3
/
LineHandler.py
68 lines (56 loc) · 2.37 KB
/
LineHandler.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import datetime
import os
from WolframAPIClient import *
from functions import *
from linebot import (
LineBotApi, WebhookHandler
)
from linebot.models import (
MessageEvent, TextMessage, TextSendMessage,JoinEvent, LeaveEvent, ImageSendMessage)
load_dotenv(verbose=True)
line_bot_api = LineBotApi(os.getenv('CHANNEL_ACCESS_TOKEN'))
handler = WebhookHandler(os.getenv('CHANNEL_SECRET'))
def reply_with_text(event_param, reply_text_param):
line_bot_api.reply_message(
event_param.reply_token,
TextSendMessage(text=reply_text_param)
)
def reply_with_image(event_param, reply_image_param):
line_bot_api.reply_message(
event_param.reply_token,
ImageSendMessage(original_content_url=reply_image_param)
)
def push_message(destination, message_type, message):
if message_type == "text":
line_bot_api.push_message(
destination, TextSendMessage(text=message)
)
elif message_type == "image":
line_bot_api.push_message(
destination, ImageSendMessage(original_content_url=message, preview_image_url=message)
)
@handler.add(MessageEvent, message=TextMessage)
def handle_message(event):
msg_source = ""
if event.source.type == "group":
msg_source = event.source.group_id
elif event.source.type == "user":
msg_source = event.source.user_id
profile = line_bot_api.get_profile(event.source.user_id)
text = str(event.message.text).lower()
command = text.strip('/')
if text.startswith('/') is True:
if command == "help":
reply_with_text(event, get_help())
elif command.startswith("solve") is True:
# TODO: Make the Full Result API more functional (look at WolframAPIClient.py)
reply_with_text(event, "Connecting to WolframAlpha...")
query = command.strip("solve ")
result_list = get_results(query)
push_message(msg_source, "text", "Real Solution:" + "\n" + " , ".join(result_list[:-1]))# Get all but last element
push_message(msg_source, "text", "Graphical plot:")
push_message(msg_source, "image", result_list[-1])
else:
reply_with_text(event, "This command hasn\'t been implemented yet..")
else:
reply_with_text(event, "Unknown message. Command message should start with '/'. Reply with /help to get more information.")