add func

api_functions.py

@@ -1,3 +1,6 @@
+from random import shuffle
+
+
 def isEnd(cur, trueChatID):
     # cur.execute(f"SELECT ChatID FROM Chats WHERE TrueChatID = '{trueChatID}'")
     # row = cur.fetchall()
@@ -24,3 +27,14 @@ def isEnd(cur, trueChatID):
         return "mafia"
 
     return ""
+
+
+def role_cnt(user_count):
+    ls = [0, 0, 1, 1, 0]
+    ls[4] = user_count//4
+    ls[1] = user_count - ls[4]
+    return ls
+
+
+def role_distribution(userIDs, count_r):
+    pass

app.py

@@ -57,5 +57,6 @@ def start():  # Команда принимает на вход
         db.query_ins_user(id, curs)
     db.close()
 
+
 if __name__ == '__main__':
     app.run()

db.py

@@ -51,4 +51,3 @@ def query_change(x):
 
 def close(cur):
     cursor.close()
-