소스 검색

offer refactored way to remove subject collection

Bastien Sevajol 6 년 전
부모
커밋
b6ab205fb0
1개의 변경된 파일7개의 추가작업 그리고 0개의 파일을 삭제
  1. 7 0
      synergine2/simulation.py

+ 7 - 0
synergine2/simulation.py 파일 보기

@@ -120,6 +120,13 @@ class Subject(IdentifiedObject):
120 120
         for collection in self.collections:
121 121
             self.simulation.collections.setdefault(collection, []).append(self.id)
122 122
 
123
+    def remove_collection(self, collection_name: str) -> None:
124
+        self.collections.remove(collection_name)
125
+        # Manipulate as shared property
126
+        simulation_collection = self.simulation.collections[collection_name]
127
+        simulation_collection.remove(self.id)
128
+        self.simulation.collections[collection_name] = simulation_collection
129
+
123 130
     def __str__(self):
124 131
         return self.__repr__()
125 132