import tornado.ioloop
import tornado.web
-import json, os
+import tornado.escape
+import os, re, glob
+import pytz, pymongo
+from datetime import datetime
from linebot import LineBotApi, WebhookParser
from linebot.exceptions import InvalidSignatureError
-from linebot.models import MessageEvent, TextMessage, TextSendMessage
+from linebot.models import TextSendMessage
-class WebHookHandler(tornado.web.RequestHandler):
+class WebHookHandler(tornado.web.RequestHandler):
+ def get(self):
+ mes = self.get_argument('code', '')
+ self.uid = ''
+ self.write(self.main(mes))
+
+ def main(self, no):
+ pz = pytz.timezone('Asia/Tokyo')
+ now = datetime.now(pz)
+ t = now.hour
+ w = now.weekday()
+ if (w < 5)and(t >= 9)and(t < 16):
+ return u'仕事中.'
+ table, na = self.users
+ item = table.find({'no':re.compile(no,re.IGNORECASE)})
+ if item.count() == 1:
+ x = item[0]
+ ans = x['name']+'\n'+x['no']
+ elif item.count() > 1:
+ ans = ''
+ obj = list(item)
+ list1 = sorted(obj, key=lambda k:k['name'])
+ for x in list1:
+ if x['name'] == list1[0]['name']:
+ ans += x['name']+'\n'+x['no']+'\n'
+ else:
+ break
+ else:
+ return ans
+ ans = self.itr(sorted(list1, key=lambda k:k['no']))
+ else:
+ ans = self.itr(table.find().sort('no'))
+ ans = '-*-'+na+' list-*-\n'+ans
+ return ans
+
+ def itr(self, item):
+ ans = ''
+ for x in item:
+ ans += '【'+x['no']+'】 '
+ return ans
+
+ def setting(self, dbname):
+ client = pymongo.MongoClient(uri)[ac]
+ if dbname in client.collection_names(include_system_collections=False):
+ db = client['users']
+ item = db.find_one({'name':self.uid})
+ if not item:
+ db.insert({'user':self.id, 'dbname':dbname})
+ elif item['dbname'] == dbname:
+ return False
+ else:
+ db.update({'user':self.uid, 'dbname':dbname})
+ return True
+
+ def users(self):
+ client = pymongo.MongoClient(uri)[ac]
+ db = client['users']
+ item = db.find_one({'name':self.uid})
+ if item:
+ x = item['dbname']
+ return client[x], x
+ else:
+ db.insert({'name':self.uid, 'dbname':'glove'})
+ return client['glove'], 'glove'
+
def post(self):
- signature = json.load(self.request.headers['X-Line-Signature'])
- data = json.load(self.request.body)
+ '''
+ signature = self.request.headers['X-Line-Signature']
+ body = self.request.body
+ parser = WebhookParser(ch)
try:
- events = webhook.parse(data, signature)
+ parser.parse(body, signature)
except InvalidSignatureError:
- self.abort(400)
- for event in events:
- if not isinstance(event,MessageEvent):
- continue
- if not isinstance(event.message,TextMessage):
- continue
- linebot.reply_message(
- event.rply_token,
- TextSendMessage(text=event.message.text)
- )
- self.abort(200)
+ tornado.web.HTTPError(404)
+ return
+ '''
+ dic = tornado.escape.json_decode(self.request.body)
+ for event in dic['events']:
+ if 'replyToken' in event:
+ x = event['message']['text']
+ self.uid = event['source']['userId']
+ if self.setting(x):
+ linebot.reply_message(event['replyToken'],
+ TextSendMessage(text=u'設定完了.'))
+ else:
+ linebot.reply_message(event['replyToken'],
+ TextSendMessage(text=self.main(x))
+ )
+
+class DummyHandler(tornado.web.RequestHandler):
+ def get(self):
+ self.db = pymongo.MongoClient(uri)[ac]
+ for x in glob.glob('./*.txt'):
+ f = open(x)
+ data = f.read()
+ f.close()
+ self.main(x[2:-4], data)
+
+ def main(self, name, data):
+ if name == 'requirements':
+ return
+ item = []
+ for x in data.split('\n'):
+ if x[0] == '@':
+ dic = {}
+ dic['name'] = x[1:]
+ else:
+ dic['no'] = x
+ item.append(dic)
+ table = self.db[name]
+ table.remove()
+ for x in item:
+ table.insert(x)
-application = tornado.web.Application([(r'/callback',WebHookHandler)],{
- 'debug':True
- })
+application = tornado.web.Application([(r'/callback',WebHookHandler),(r'/init',DummyHandler)])
if __name__ == '__main__':
- ch_id = os.environ['Channel_ID']
+ token = os.environ['Access_Token']
ch = os.environ['Channel_Secret']
- linebot = LineBotApi(ch_id)
+ uri = os.environ['MONGODB_URI']
+ ac = os.environ['ACCOUNT']
+ port = int(os.environ.get('PORT',5000))#important in heroku
+ linebot = LineBotApi(token)
webhook = WebhookParser(ch)
- application.listen(5000)
- tornado.ioloop.IOLoop.instance().start()
\ No newline at end of file
+ application.listen(port)
+ tornado.ioloop.IOLoop.instance().start()
+
\ No newline at end of file
OSDN Git Service
