From f685397dcdb01c222c383d5564ec3ee7f8fe858c Mon Sep 17 00:00:00 2001
From: SiHuan <sihuan@sakuya.love>
Date: Fri, 30 Oct 2020 10:13:44 +0800
Subject: [PATCH] fix lsp keyword search

---
 worker/lsp.py | 34 +++++++++++++++++++---------------
 1 file changed, 19 insertions(+), 15 deletions(-)

diff --git a/worker/lsp.py b/worker/lsp.py
index a1d99e7..bd7b688 100644
--- a/worker/lsp.py
+++ b/worker/lsp.py
@@ -3,12 +3,15 @@ import requests
 
 from config import LOLIKEYS
 
+
 class Ans(StdAns):
-    AllowGroup = [973510746, 805197917,343700338,125733077,1084566280,920863253,798595664,655057127,196268763, 247022495, 474907856, 940333876]
+    AllowGroup = [973510746, 805197917, 343700338, 125733077, 1084566280,
+                  920863253, 798595664, 655057127, 196268763, 247022495, 474907856, 940333876]
+
     def GETMSG(self):
 
         keys = LOLIKEYS
-        
+
         try:
             nowkey = int(self.DATAGET()['nowkey'])
         except:
@@ -20,43 +23,43 @@ class Ans(StdAns):
             keyword = self.parms[1]
             if mysakuya(self, keyword) == False:
                 return "不许你们看咲夜的涩图！！"
-            
+
         code, quota, picurl = getsetu(keys[nowkey], keyword)
-        
+
         msg = ''
         if code == -1 or code == -2:
             msg += f'>_< 天啦喽，loli 不见了 Code:{code}'
         elif code == 3:
             msg += '咱没查到,也有可能是Pixiv坏掉惹'
         elif code == 0:
-            if len(self.parms) < 2 or (len(self.parms > 2) and self.parms[2] == 'p'):
+            if len(self.parms) < 2 or (len(self.parms) > 2 and self.parms[2] == 'p'):
                 msg += f'[CQ:image,file={picurl}]'
             else:
                 msg += f'[CQ:reply,id={self.mid}]咱帮你🔍{keyword}找到了这个\n{picurl}'
-                
+
         if quota == 0:
-            nowkey = (nowkey+1)%(len(keys))
+            nowkey = (nowkey+1) % (len(keys))
             if code == 429:
                 msg += f'>_< 已经没有 loli 啦，帮你换到了key{nowkey}，再试一下吧'
             else:
                 msg += f'额度用光了，下次将使用 key{nowkey}'
-        self.DATASET({'nowkey':nowkey})
+        self.DATASET({'nowkey': nowkey})
         return msg
-        
-        
-def getsetu(apikey, keyword = ''):
+
+
+def getsetu(apikey, keyword=''):
     url = 'https://api.lolicon.app/setu/'
     params = {
         'apikey': apikey,
     }
     if keyword != '':
         params['keyword'] = keyword
-        
+
     try:
-        resp = requests.get(url=url,params=params).json()
+        resp = requests.get(url=url, params=params).json()
     except:
-        return -1, -1 ,''
-    
+        return -1, -1, ''
+
     #quota = str(resp['quota'])
     #seconds = resp['quota_min_ttl']
     #m, s = divmod(seconds, 60)
@@ -74,4 +77,5 @@ def getsetu(apikey, keyword = ''):
     elif resp['code'] == 429:
         return 429, 0, ''
     else:
+        # print(resp['code'])
         return -2, -1, ''