final_vers

app/plugin/shuffle.py

@@ -1,6 +1,155 @@
-from plugin.user import User
+from plugin.user import User, users_reg
+from typing import List, Tuple
+import random
+
+already_matched: List[Tuple[User, User]] = []
 
 
-def Shuffle(users: list[User]):
+def Shuffle() -> List[Tuple[User, User]]:
+    users = users_reg.copy()
+    random.shuffle(users)
+    pairs = []
 
-    return
+    if len(users) % 2 != 0:
+        pairs.append((users.pop(),))
+
+    # Step 1: Match men with women by compatibility and age
+    for user in users:
+        for potential_match in users:
+            if user == potential_match:
+                continue
+            if (
+                user.gender != potential_match.gender
+                and potential_match.type in compatibility[user.type]
+                and abs(user.age - potential_match.age) <= 5
+                and (user, potential_match) not in already_matched
+                and (potential_match, user) not in already_matched
+            ):
+                pairs.append((user, potential_match))
+                users.remove(user)
+                users.remove(potential_match)
+                break
+
+    # Step 2: Match men with women with compatibility
+    for user in users:
+        for potential_match in users:
+            if user == potential_match:
+                continue
+            if (
+                user.gender != potential_match.gender
+                and potential_match.type in compatibility[user.type]
+                and (user, potential_match) not in already_matched
+                and (potential_match, user) not in already_matched
+            ):
+                pairs.append((user, potential_match))
+                users.remove(user)
+                users.remove(potential_match)
+                break
+
+    # Step 3: Match men with women
+    for user in users:
+        for potential_match in users:
+            if user == potential_match:
+                continue
+            if (
+                user.gender != potential_match.gender
+                and (user, potential_match) not in already_matched
+                and (potential_match, user) not in already_matched
+            ):
+                pairs.append((user, potential_match))
+                users.remove(user)
+                users.remove(potential_match)
+                break
+
+    # Step 4: Match people of the same age
+    for user in users:
+        for potential_match in users:
+            if user == potential_match:
+                continue
+            if (
+                user.age == potential_match.age
+                and (user, potential_match) not in already_matched
+                and (potential_match, user) not in already_matched
+            ):
+                pairs.append((user, potential_match))
+                users.remove(user)
+                users.remove(potential_match)
+                break
+
+    # Step 5: Pair all remaining users
+    while len(users) > 1:
+        pairs.append((users.pop(), users.pop()))
+
+    already_matched.extend(pairs)
+
+    return pairs
+
+
+compatibility = {
+    "INTJ": (
+        "ENFP",
+        "ENTP",
+    ),
+    "INTP": (
+        "ENTJ",
+        "ESTJ",
+    ),
+    "INFJ": (
+        "ENFP",
+        "ENTP",
+    ),
+    "INFP": (
+        "ENFJ",
+        "ENTJ",
+    ),
+    "ISTJ": (
+        "ESFP",
+        "ESTP",
+    ),
+    "ISTP": (
+        "ESFJ",
+        "ESTJ",
+    ),
+    "ISFJ": (
+        "ESFP",
+        "ESTP",
+    ),
+    "ISFP": (
+        "ENFJ",
+        "ESFJ",
+        "ESTJ",
+    ),
+    "ENTJ": (
+        "INFP",
+        "INTP",
+    ),
+    "ENTP": (
+        "INTJ",
+        "ENTP",
+    ),
+    "ENFJ": (
+        "INFP",
+        "ISFP",
+        "ENFJ",
+    ),
+    "ENFP": (
+        "INFJ",
+        "INTJ",
+    ),
+    "ESTJ": (
+        "ISFP",
+        "ISTP",
+    ),
+    "ESTP": (
+        "ISFJ",
+        "ISTJ",
+    ),
+    "ESFJ": (
+        "ISFP",
+        "ISTP",
+    ),
+    "ESFP": (
+        "ISTJ",
+        "ISFJ",
+    ),
+}