Add attachment pairs into BDDL

bddl/knowledge_base/models.py

@@ -3,7 +3,7 @@
 import itertools
 import json
 import networkx as nx
-from typing import Dict, Set
+from typing import Dict, Set, List, Tuple
 from bddl.object_taxonomy import ObjectTaxonomy
 from bddl.knowledge_base.orm import (
     Model,
@@ -15,7 +15,7 @@
     OneToManyField,
     UUIDField,
 )
-from bddl.knowledge_base.utils import *
+from bddl.knowledge_base.utils import SynsetState
 from collections import defaultdict
 
 
@@ -89,6 +89,16 @@ class Meta:
         pk = "name"
 
 
+@dataclass(eq=False, order=False)
+class AttachmentPair(Model):
+    name: str
+    female_object_fk: ManyToMany = ManyToManyField("Object", "female_attachment_pairs")
+    male_object_fk: ManyToMany = ManyToManyField("Object", "male_attachment_pairs")
+
+    class Meta:
+        pk = "name"
+
+
 @dataclass(eq=False, order=False)
 class Predicate(Model):
     name: str
@@ -196,6 +206,10 @@ class Object(Model):
     # roomobject counts of this object
     roomobjects_fk: OneToMany = OneToManyField("RoomObject", "object")
 
+    # attachment pairs this object participates in
+    female_attachment_pairs_fk: ManyToMany = ManyToManyField(AttachmentPair, "female_objects")
+    male_attachment_pairs_fk: ManyToMany = ManyToManyField(AttachmentPair, "male_objects")
+
     def __str__(self):
         return self.category.name + "-" + self.name