feat(*.py): add limit count of words that Google Home will response. …

…fix #419 @2.5h

domain.py

@@ -25,6 +25,7 @@
 ASK_CONTINUE_EVENT = u'ASK_CONTINUE_EVENT'
 ASK_WORD_EVENT = u'ASK_WORD_EVENT'
 DECLARE_USER_LOSE_EVENT = u'DECLARE_USER_LOSE_EVENT'
+DECLARE_GOOGLE_HOME_LOSE_EVENT = u'DECLARE_GOOGLE_HOME_LOSE_EVENT'
 
 
 def ask_continue(obj):
@@ -52,7 +53,12 @@ def response_word(obj):
     queryResult = obj[u'queryResult']
     userId = originalDetectIntentRequest[u'payload'][u'user'][u'userId']
     queryText = queryResult[u'queryText']
+
+    user = infra.load_user(userId)
+
     if queryText == ASK_WORD_EVENT:
+        infra.reset_datastore(user)
+        infra.save_word_datastore(user, u'シリトリ')
         return {
             u'fulfillmentText': u'しりとり、の、リ',
         }
@@ -62,36 +68,70 @@ def response_word(obj):
         if reading:
             logging.info(reading)
             reading_end = reading[-1]
-            if infra.check_last_word_datastore(userId, reading):
+            if infra.check_last_word_datastore(user, reading):
                 if reading_end == u'ン':
-                    infra.reset_datastore(userId)
+                    infra.reset_datastore(user)
                     return {
                         u'followupEventInput': {
                             u'name': DECLARE_USER_LOSE_EVENT,
                             u'languageCode': queryResult[u'languageCode'],
                         }
                     }
-                elif infra.check_word_datastore(userId, reading):
-                    infra.save_word_datastore(userId, reading)
-                    word_record = infra.search_word_record_from_dic(
-                        userId, reading_end)
-                    logging.info(word_record)
-                    word = word_record[u'org'][0]
-                    infra.save_word_datastore(userId, word_record[u'key'])
-                    fulfillmentText = word + u'、の、' + word_record[u'end']
-                    logging.info(fulfillmentText)
-                    return {
-                        u'fulfillmentText': fulfillmentText,
-                    }
+                elif infra.check_word_datastore(user, reading):
+                    infra.save_word_datastore(user, reading)
+                    if infra.is_need_google_home_lose(user):
+                        return {
+                            u'followupEventInput': {
+                                u'name': DECLARE_GOOGLE_HOME_LOSE_EVENT,
+                                u'languageCode': queryResult[u'languageCode'],
+                            }
+                        }
+                    else:
+                        word_record = infra.search_word_record_from_dic(
+                            user, reading_end)
+                        logging.info(word_record)
+                        word = word_record[u'org'][0]
+                        infra.save_word_datastore(user, word_record[u'key'])
+                        fulfillmentText = word + u'、の、' + word_record[u'end']
+                        logging.info(fulfillmentText)
+                        return {
+                            u'fulfillmentText': fulfillmentText,
+                        }
                 else:
                     return {
                         u'fulfillmentText': u'それは使用済みの言葉です',
                     }
             else:
                 return {
-                    u'fulfillmentText': infra.get_last_word_datastore(userId) + u'で始まる言葉を使ってください',
+                    u'fulfillmentText': infra.get_last_word_datastore(user) + u'で始まる言葉を使ってください',
                 }
         else:
             return {
                 u'fulfillmentText': u'それは知らない言葉です',
             }
+
+
+def response_lose_word(obj):
+    originalDetectIntentRequest = obj['originalDetectIntentRequest']
+    userId = originalDetectIntentRequest[u'payload'][u'user'][u'userId']
+
+    user = infra.load_user(userId)
+    reading_end = user.last_word[-1]
+
+    word_record = infra.search_lose_word_record_from_dic(
+        user, reading_end)
+    logging.info(word_record)
+    fulfillmentText = u''
+    if word_record:
+        word = word_record[u'org'][0]
+        infra.save_word_datastore(user, word_record[u'key'])
+        fulfillmentText += word + u'。'
+        fulfillmentText += u'あら、「ん」で終わってしまいました。'
+    else:
+        fulfillmentText += u'うーん、「' + reading_end + u'」で始まる言葉が思いつきません。'
+    fulfillmentText += u'私の負けです。'
+    logging.info(fulfillmentText)
+    infra.reset_datastore(user)
+    return {
+        u'fulfillmentText': fulfillmentText,
+    }

infra.py

@@ -23,6 +23,10 @@
 import json
 import random
 
+# しりとりが WORDS_COUNT_LIMIT 以上続いたら AI が降参する
+WORDS_COUNT_LIMIT = 10
+
+
 with open('data/dict.json') as json_file:
     json_dic = json.load(json_file)
 
@@ -34,36 +38,51 @@ class User(ndb.Model):
     date = ndb.DateTimeProperty(auto_now_add=True)
 
 
-def reset_datastore(user_id):
+def load_user(user_id):
     try:
         user = User.get_by_id(user_id)
+        if user:
+            return user
+    except Exception:
+        pass
+
+    user = User(id=user_id)
+    user.words = None
+    user.last_word = None
+    user.count = 0
+    user.date = None
+    return user
+
+
+def reset_datastore(user):
+    try:
         user.key.delete()
+        user.words = None
+        user.last_word = None
+        user.count = 0
+        user.date = None
     except Exception:
         pass
 
 
-def get_last_word_datastore(user_id):
+def get_last_word_datastore(user):
     try:
-        user = User.get_by_id(user_id)
-        if user.last_word == u'':
-            return u'リ'
         return user.last_word[-1]
     except Exception:
-        return u'リ'
+        pass
 
 
-def check_last_word_datastore(user_id, check_word):
+def check_last_word_datastore(user, check_word):
     try:
-        if check_word[0] == get_last_word_datastore(user_id):
+        if check_word[0] == get_last_word_datastore(user):
             return True
         return False
     except Exception:
         return True
 
 
-def check_word_datastore(user_id, check_word):
+def check_word_datastore(user, check_word):
     try:
-        user = User.get_by_id(user_id)
         words = user.words.split(u',')
         if check_word in words:
             return False
@@ -72,15 +91,15 @@ def check_word_datastore(user_id, check_word):
         return True
 
 
-def save_word_datastore(user_id, save_word):
+def save_word_datastore(user, save_word):
     try:
-        user = User.get_by_id(user_id)
-        user.words += u',' + save_word
+        if user.words:
+            user.words += u',' + save_word
+        else:
+            user.words = save_word
         user.count += 1
     except Exception:
-        user = User(id=user_id)
-        user.words = save_word
-        user.count = 1
+        pass
     user.last_word = save_word
     user.put()
 
@@ -99,7 +118,7 @@ def search_reading_from_dic(search_word):
     return u''
 
 
-def search_word_record_from_dic(user_id, search_first):
+def search_word_record_from_dic(user, search_first):
     """json形式の辞書ファイルを全探索し、読みが search_first で始まる適当な単語レコードを返す。
     無効な単語（既出の単語・存在しない単語・「ん」で終わる単語）のレコードは返さない。
 
@@ -116,10 +135,42 @@ def search_word_record_from_dic(user_id, search_first):
         if dict_record[u'end'] == u'ン':
             continue
         # Google Home は既出の単語を言わない
-        if not check_word_datastore(user_id, dict_record[u'key']):
+        if not check_word_datastore(user, dict_record[u'key']):
             continue
         dict_record_list.append(dict_record)
     if dict_record_list:
         return dict_record_list[random.randint(0, len(dict_record_list) - 1)]
     else:
         return {}
+
+
+def search_lose_word_record_from_dic(user, search_first):
+    """json形式の辞書ファイルを全探索し、読みが search_first で始まり
+    かつ、「ん」で終わる単語レコードを返す。
+
+    :param unicode search_first: カタカナ 1 文字
+    :rtype: dict
+    :return: 検索した単語レコード(辞書にない場合は空の辞書)
+    """
+    dict_record_list = []
+    for dict_record in json_dic:
+        # search_first で始まる単語レコードを検索する
+        if dict_record[u'first'] != search_first:
+            continue
+        # 「ん」で終わる単語を検索する
+        if dict_record[u'end'] == u'ン':
+            dict_record_list.append(dict_record)
+    if dict_record_list:
+        return dict_record_list[random.randint(0, len(dict_record_list) - 1)]
+    else:
+        return {}
+
+
+def is_need_google_home_lose(user):
+    """Google Home の負け処理が必要であるかを返す。
+
+    :param infra.User user: ユーザ
+    :rtype: bool
+    :return: Google Home の負け処理が必要であるか
+    """
+    return user.count >= WORDS_COUNT_LIMIT

main.py

@@ -27,6 +27,7 @@
 GOOGLE_ASSISTANT_WELCOME_INTENT = u'Google Assistant Welcome Intent'
 ASK_CONTINUE_INTENT = u'Ask Continue Intent'
 ASK_WORD_INTENT = u'Ask Word Intent'
+DECLARE_GOOGLE_HOME_LOSE_INTENT = u'Declare Google Home Lose Intent'
 
 
 class MainPage(webapp2.RequestHandler):
@@ -51,6 +52,8 @@ def post(self):
             obj = domain.set_continue(obj)
         elif intentDisplayName == ASK_WORD_INTENT:
             obj = domain.response_word(obj)
+        elif intentDisplayName == DECLARE_GOOGLE_HOME_LOSE_INTENT:
+            obj = domain.response_lose_word(obj)
         else:
             obj = {
                 u'fulfillmentText': queryResult[u'queryText'],