[Gban] make spamwatch ignore wolves and tigers

aceleonister001 · Apr 11, 2021 · 98bd86b · 98bd86b
1 parent 713615b
commit 98bd86b
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/SaitamaRobot/modules/global_bans.py b/SaitamaRobot/modules/global_bans.py
@@ -415,11 +415,13 @@ def gbanlist(update: Update, context: CallbackContext):
 
 def check_and_ban(update, user_id, should_message=True):
 
-    chat = update.effective_chat  # type: Optional[Chat]
-    try:
-        sw_ban = sw.get_ban(int(user_id))
-    except:
+    if user_id in TIGERS or user_id in WOLVES:
         sw_ban = None
+    else:
+        try:
+            sw_ban = sw.get_ban(int(user_id))
+        except:
+            sw_ban = None
 
     if sw_ban:
         update.effective_chat.kick_member(user_id)