implemented helper private methods for creators

This commit is contained in:
iLLiCiTiT 2021-10-01 19:30:48 +02:00
parent 3e8c97e9d6
commit d15240d2a3
4 changed files with 26 additions and 20 deletions

View file

@ -26,14 +26,14 @@ class MyAutoCreator(AutoCreator):
instance = CreatedInstance(
self.family, subset_name, instance_data, self
)
self.create_context.add_instance(instance)
self._add_instance_to_context(instance)
elif instance_data["family"] == self.identifier:
instance_data["creator_identifier"] = self.identifier
instance = CreatedInstance.from_existing(
instance_data, self, attr_plugins
)
self.create_context.add_instance(instance)
self._add_instance_to_context(instance)
def update_instances(self, update_list):
pipeline.update_instances(update_list)
@ -83,4 +83,4 @@ class MyAutoCreator(AutoCreator):
existing_instance.data["asset"] = asset_name
existing_instance.data["task"] = task_name
return existing_instance
self._add_instance_to_context(existing_instance)

View file

@ -24,26 +24,28 @@ class TestCreatorOne(Creator):
instance = CreatedInstance(
self.family, subset_name, instance_data, self
)
self.create_context.add_instance(instance)
self._add_instance_to_context(instance)
elif instance_data["family"] == self.identifier:
instance_data["creator_identifier"] = self.identifier
instance = CreatedInstance.from_existing(
instance_data, self, attr_plugins
)
self.create_context.add_instance(instance)
self._add_instance_to_context(instance)
def update_instances(self, update_list):
pipeline.update_instances(update_list)
def remove_instances(self, instances):
pipeline.remove_instances(instances)
for instance in instances:
self._remove_instance_from_context(instance)
def create(self, subset_name, data, options=None):
avalon_instance = CreatedInstance(self.family, subset_name, data, self)
pipeline.HostContext.add_instance(avalon_instance.data_to_store())
self.log.info(avalon_instance.data)
return avalon_instance
new_instance = CreatedInstance(self.family, subset_name, data, self)
pipeline.HostContext.add_instance(new_instance.data_to_store())
self.log.info(new_instance.data)
self._add_instance_to_context(new_instance)
def get_default_variants(self):
return [

View file

@ -15,10 +15,10 @@ class TestCreatorTwo(Creator):
return "cube"
def create(self, subset_name, data, options=None):
avalon_instance = CreatedInstance(self.family, subset_name, data, self)
pipeline.HostContext.add_instance(avalon_instance.data_to_store())
self.log.info(avalon_instance.data)
return avalon_instance
new_instance = CreatedInstance(self.family, subset_name, data, self)
pipeline.HostContext.add_instance(new_instance.data_to_store())
self.log.info(new_instance.data)
self._add_instance_to_context(new_instance)
def collect_instances(self, attr_plugins):
for instance_data in pipeline.list_instances():
@ -29,20 +29,22 @@ class TestCreatorTwo(Creator):
instance = CreatedInstance(
self.family, subset_name, instance_data, self
)
self.create_context.add_instance(instance)
self._add_instance_to_context(instance)
elif instance_data["family"] == self.identifier:
instance_data["creator_identifier"] = self.identifier
instance = CreatedInstance.from_existing(
instance_data, self, attr_plugins
)
self.create_context.add_instance(instance)
self._add_instance_to_context(instance)
def update_instances(self, update_list):
pipeline.update_instances(update_list)
def remove_instances(self, instances):
pipeline.remove_instances(instances)
for instance in instances:
self._remove_instance_from_context(instance)
def get_attribute_defs(self):
output = [