Kaynağa Gözat

Merge remote-tracking branch 'origin/master'

zhangyong 1 yıl önce
ebeveyn
işleme
9ab4a00d6a
2 değiştirilmiş dosya ile 8 ekleme ve 9 silme
  1. 1 5
      changsha_bot.py
  2. 7 4
      common/limit.py

+ 1 - 5
changsha_bot.py

@@ -5,7 +5,6 @@ import json
 import datetime
 import time
 
-import schedule
 import requests
 
 from common.db import RedisClient
@@ -86,7 +85,4 @@ def monitor():
 
 
 if __name__ == '__main__':
-    schedule.every().day.at("08:00").do(monitor)
-    while True:
-        schedule.run_pending()
-        time.sleep(60)
+    monitor()

+ 7 - 4
common/limit.py

@@ -31,7 +31,7 @@ class AuthorLimit(object):
     def __init__(self, mode, platform):
         self.mode = mode
         self.platform = platform
-        self.limit_tag_dict = {"352": "余海涛", "353": "罗情", "53": "范军", "51": "鲁涛"}
+        self.limit_tag_dict = {"352": "余海涛", "353": "罗情", "53": "范军", "51": "鲁涛", "131": False}
 
     def find_tag(self, uid):
         """
@@ -44,9 +44,12 @@ class AuthorLimit(object):
         tags = result[0]["tag"]
         if tags:
             tags = tags.split(",")
-            for tag in tags:
-                if self.limit_tag_dict.get(tag):
-                    return tag
+            if "131" in tags:
+                return None
+            else:
+                for tag in tags:
+                    if self.limit_tag_dict.get(tag):
+                        return tag
         return None
 
     def author_limitation(self, user_id):