From 87c2524397966e9f2859d7b3b4048cfd95972906 Mon Sep 17 00:00:00 2001
From: iLLiCiTiT <jakub.trllo@gmail.com>
Date: Mon, 13 Sep 2021 18:53:13 +0200
Subject: [PATCH] added auto creator

---
 .../testhost/plugins/create/auto_creator.py   | 65 +++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 openpype/hosts/testhost/plugins/create/auto_creator.py

diff --git a/openpype/hosts/testhost/plugins/create/auto_creator.py b/openpype/hosts/testhost/plugins/create/auto_creator.py
new file mode 100644
index 0000000000..40cab97839
--- /dev/null
+++ b/openpype/hosts/testhost/plugins/create/auto_creator.py
@@ -0,0 +1,65 @@
+from openpype import resources
+from openpype.hosts.testhost import api
+from openpype.pipeline import (
+    AutoCreator,
+    CreatedInstance,
+    lib
+)
+from avalon import io
+
+
+class MyAutoCreator(AutoCreator):
+    family = "workfile"
+
+    def create(self, options=None):
+        existing_instance = None
+        for instance in self.create_context.instances:
+            if instance.family == self.family:
+                existing_instance = instance
+                break
+
+        variant = "Main"
+        project_name = io.Session["AVALON_PROJECT"]
+        asset_name = io.Session["AVALON_ASSET"]
+        task_name = io.Session["AVALON_TASK"]
+        host_name = io.Session["AVALON_APP"]
+
+        if existing_instance is None:
+            asset_doc = io.find_one({"type": "asset", "name": asset_name})
+            subset_name = self.get_subset_name(
+                variant, task_name, asset_doc, project_name, host_name
+            )
+            data = {
+                "asset": asset_name,
+                "task": task_name,
+                "variant": variant
+            }
+            data.update(self.get_dynamic_data(
+                variant, task_name, asset_doc, project_name, host_name
+            ))
+
+            existing_instance = CreatedInstance(
+                self.family, subset_name, data, self
+            )
+            api.pipeline.HostContext.add_instance(
+                existing_instance.data_to_store()
+            )
+
+        elif (
+            existing_instance.data["asset"] != asset_name
+            or existing_instance.data["task"] != task_name
+        ):
+            asset_doc = io.find_one({"type": "asset", "name": asset_name})
+            subset_name = self.get_subset_name(
+                variant, task_name, asset_doc, project_name, host_name
+            )
+            existing_instance.data["asset"] = asset_name
+            existing_instance.data["task"] = task_name
+
+        return existing_instance
+
+    def get_attribute_defs(self):
+        output = [
+            lib.NumberDef("number_key", label="Number")
+        ]
+        return output