Update CharacterService to handle relationship sets properly

College-Kings · May 25, 2024 · 4b2451b · 4b2451b
1 parent ea688b3
commit 4b2451b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/CharacterService_ren.py b/CharacterService_ren.py
@@ -54,6 +54,14 @@ def get_relationship(
         if not hasattr(character, "relationships"):
             character.relationships = {}
 
+        if isinstance(target.relationships, set):
+            target.relationships = {r: Relationship.FWB for r in target.relationships}
+
+        if isinstance(character.relationships, set):
+            character.relationships = {
+                r: Relationship.FWB for r in character.relationships
+            }
+
         if (
             target.relationships.get(character, Relationship.STRANGER).value
             > character.relationships.get(target, Relationship.STRANGER).value