import tornado.ioloop
import tornado.web
import tornado.escape
-import os, hmac, base64, hashlib, re
-import json, pytz, pymongo
+import os, re, glob
+import pytz, pymongo
from datetime import datetime
-from linebot import LineBotApi, WebhookParser, WebhookHandler
+from linebot import LineBotApi, WebhookParser
from linebot.exceptions import InvalidSignatureError
from linebot.models import TextSendMessage
-from multiprocessing.connection import answer_challenge
-class WebHookHandler(tornado.web.RequestHandler):
- def get(self):
- mes = self.get_argument('code','')
- self.write(self.main(mes))
-
- def main(self,no):
+class WebHookHandler(tornado.web.RequestHandler):
+ 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'仕事中.'
- db = pymongo.MongoClient(uri)[ac]
- table = db['glove']
+ table, na = self.users()
item = table.find({'no':re.compile(no,re.IGNORECASE)})
if item.count() == 1:
x = item[0]
else:
break
else:
- return ans
- ans = ''
- for x in sorted(list1, key=lambda k:k['no']):
- ans += x['no']+'\n'
+ return ans
+ ans = self.itr(sorted(list1, key=lambda k:k['no']))
else:
- ans = ''
- for x in table.find().sort('no'):
- ans += x['no']+'\n'
+ 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 help(self):
+ s = '-*-database names-*-\n'
+ for x in self.database.collection_names(include_system_collections=True):
+ s += x+'\n'
+ return s
+
+ def setting(self, dbname):
+ dbname = dbname.lower()
+ if dbname in self.database.collection_names(include_system_collections=False):
+ db = self.database['users']
+ item = db.find_one({'name':self.uid})
+ if item['dbname'] == dbname:
+ return False
+ else:
+ db.update({'name':self.uid}, {'name':self.uid, 'dbname':dbname})
+ return True
+ return False
+
+ def users(self):
+ db = self.database['users']
+ item = db.find_one({'name':self.uid})
+ x = item['dbname']
+ return self.database[x], x
+
def post(self):
- header = json.load(self.request.headers)
- body = json.load(self.request.body)
- hash = hmac.new(header['X-LINE-SIGNATURE'].encode('utf-8'),
- body.encode('utf-8'), hashlib.sha256).digest()
- signature = base64.b64encode(hash)
+ '''
+ signature = self.request.headers['X-Line-Signature']
+ body = self.request.body
+ parser = WebhookParser(ch)
try:
- events = webhook.parse(body, signature)
+ parser.parse(body, signature)
except InvalidSignatureError:
- raise tornado.web.HTTPError(400)
+ tornado.web.HTTPError(404)
return
- for event in events:
- if (event['type'] == 'text')and(event['message']['type'] == 'text'):
- linebot.reply_message(
- event.reply_token,
- TextSendMessage(text=self.main(event.Message.text))
- )
+ '''
+ dic = tornado.escape.json_decode(self.request.body)
+ for event in dic['events']:
+ if 'replyToken' in event.keys():
+ self.uid = event['source']['userId']
+ self.database = pymongo.MongoClient(uri)[ac]
+ if event['type'] == 'unfollow':
+ self.database['users'].remove({'name':self.uid})
+ return
+ elif event['type'] == 'join':
+ db = self.database['users']
+ if not db.find_one({'name':self.uid}):
+ db.insert({'name':self.uid, 'dbname':'glove'})
+ return
+ x = event['message']['text']
+ if self.setting(x):
+ linebot.reply_message(event['replyToken'],
+ TextSendMessage(text=u'設定完了.'))
+ elif x == '?':
+ linebot.reply_message(event['replyToken'],
+ TextSendMessage(text=self.help())
+ )
+ else:
+ linebot.reply_message(event['replyToken'],
+ TextSendMessage(text=self.main(x))
+ )
class DummyHandler(tornado.web.RequestHandler):
- def get(self):
- f = open('data.txt')
- data = f.read()
- f.close()
- db = pymongo.MongoClient(uri)[ac]
- table = db['glove']
+ 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 = []
+ dic = None
for x in data.split('\n'):
- if x[0] == '@':
+ if len(x) > 0 and x[0] == '@':
dic = {}
dic['name'] = x[1:]
- else:
+ elif dic:
dic['no'] = x
item.append(dic)
+ table = self.db[name]
table.remove()
for x in item:
table.insert(x)