collection names

[simple-tornado-bot/simple-tornado-bot.git] / bot.py

diff --git a/bot.py b/bot.py
index 5f872b8..8f2fe7b 100644 (file)
--- a/bot.py
+++ b/bot.py
@@ -52,6 +52,12 @@ class WebHookHandler(tornado.web.RequestHandler):
             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):
@@ -98,6 +104,10 @@ class WebHookHandler(tornado.web.RequestHandler):
                 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))
@@ -116,11 +126,12 @@ class DummyHandler(tornado.web.RequestHandler):
         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]