-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
115 lines (84 loc) · 2.77 KB
/
app.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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
# encoding: utf-8
import os
##########################################
# Init db connection.
##########################################
conn = None
# Remove the comment notations. Make a connenction to real db. [TODO_DB]
"""
import psycopg2
import urllib.parse as urlparse
urlparse.uses_netloc.append("postgres")
url = urlparse.urlparse(os.environ["DATABASE_URL"])
conn = psycopg2.connect(
database=url.path[1:],
user=url.username,
password=url.password,
host=url.hostname,
port=url.port
)
"""
##########################################
# Init bot.
##########################################
from DataManager import DataManager
data_manager = DataManager(conn)
from CianCianBot import CianCianBot
bot = CianCianBot(data_manager)
##########################################
# Init flask backend and linebot facility.
##########################################
from flask import Flask, request, abort
from linebot import (
LineBotApi, WebhookHandler
)
from linebot.exceptions import (
InvalidSignatureError
)
from linebot.models import (
MessageEvent, TextMessage, TextSendMessage,
)
app = Flask(__name__)
channel_secret = '71e40cc8f86e10bf58698084c39f1192'
channel_access_token = 'niVSl54pEQPH3YwsHHnL3oZpTKZiMMJsIp4qRKq6X/i6iimSjJf5hJA0Vt4EhkLxzmPiuusLET5IfgZuiSroiD3Qem6ypgqXW6HXOO3SkJMURH0ZOD2KyTwMPSg2V2PF5JE3PDanRwsa98Vrx6YBrgdB04t89/1O/w1cDnyilFU='
handler = WebhookHandler(channel_secret)
line_bot_api = LineBotApi(channel_access_token)
@app.route('/')
def index():
return "<p>Hello World!</p>"
@app.route("/callback", methods=['POST'])
def callback():
# get X-Line-Signature header value
signature = request.headers['X-Line-Signature']
# get request body as text
body = request.get_data(as_text=True)
app.logger.info("Request body: " + body)
# handle webhook body
try:
handler.handle(body, signature)
except InvalidSignatureError:
abort(400)
return 'OK'
@handler.add(MessageEvent, message=TextMessage) # default
def handle_text_message(event): # default
# User's message
msg = event.message.text # message from user
# User's chatting window id, could be `user_id`, `room_id`, `group_id`.
if event.source.type == "user":
src_id = event.source.user_id
elif event.source.type == "room":
src_id = event.source.room_id
elif event.source.type == "group":
src_id = event.source.group_id
else:
src_id = "error"
unique_id = str(event.source.type) + "_" + src_id
# Responding algorithm
bot_response = bot.respond(msg, unique_id)
# Reply
line_bot_api.reply_message(
event.reply_token,
TextSendMessage(text=bot_response)
)
if __name__ == "__main__":
app.run(host='0.0.0.0',port=os.environ['PORT'])