stop adding private chats to groups collection

drforse · Jan 29, 2020 · 5172d57 · 5172d57
1 parent 59c9039
commit 5172d57
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/bot.py b/bot.py
@@ -15,17 +15,20 @@
 
 
 async def check_group_and_user(chat_id, user_id):
-    group = groups_col.find_one({'group_id': chat_id})
-    if not group:
-        groups_col.insert_one({'group_id': chat_id,
-                               'users': [user_id]})
-        group = groups_col.find_one({'group_id': chat_id})
     user = users_col.find_one({'user_id': user_id})
     if not user:
         users_col.insert_one({'user_id': user_id,
                               'years': 100,
                               'games_finished': 0,
                               'loses': 0})
+    grp = await bot.get_chat(chat_id)
+    if grp.stats == 'private':
+        return
+    group = groups_col.find_one({'group_id': chat_id})
+    if not group:
+        groups_col.insert_one({'group_id': chat_id,
+                               'users': [user_id]})
+        group = groups_col.find_one({'group_id': chat_id})
     if user_id not in group['users']:
         groups_col.update_one({'group_id': chat_id},
                               {'$push': {'users': user_id}})