From 32e02701a17ab058a9efeb9d66017a5673922531 Mon Sep 17 00:00:00 2001
From: Jakub Trllo <jakub.trllo@gmail.com>
Date: Thu, 14 Apr 2022 12:22:43 +0200
Subject: [PATCH] replaced avalon imports in testhost

---
 openpype/hosts/testhost/api/__init__.py       |  6 +++---
 openpype/hosts/testhost/api/pipeline.py       |  5 ++---
 .../testhost/plugins/create/auto_creator.py   | 20 ++++++++++++-------
 openpype/hosts/traypublisher/api/pipeline.py  | 11 +++++-----
 4 files changed, 24 insertions(+), 18 deletions(-)

diff --git a/openpype/hosts/testhost/api/__init__.py b/openpype/hosts/testhost/api/__init__.py
index 7840b25892..a929a891aa 100644
--- a/openpype/hosts/testhost/api/__init__.py
+++ b/openpype/hosts/testhost/api/__init__.py
@@ -1,8 +1,8 @@
 import os
 import logging
 import pyblish.api
-import avalon.api
-from openpype.pipeline import BaseCreator
+
+from openpype.pipeline import register_creator_plugin_path
 
 from .pipeline import (
     ls,
@@ -27,7 +27,7 @@ def install():
     log.info("OpenPype - Installing TestHost integration")
     pyblish.api.register_host("testhost")
     pyblish.api.register_plugin_path(PUBLISH_PATH)
-    avalon.api.register_plugin_path(BaseCreator, CREATE_PATH)
+    register_creator_plugin_path(CREATE_PATH)
 
 
 __all__ = (
diff --git a/openpype/hosts/testhost/api/pipeline.py b/openpype/hosts/testhost/api/pipeline.py
index 1f5d680705..285fe8f8d6 100644
--- a/openpype/hosts/testhost/api/pipeline.py
+++ b/openpype/hosts/testhost/api/pipeline.py
@@ -1,5 +1,6 @@
 import os
 import json
+from openpype.pipeline import legacy_io
 
 
 class HostContext:
@@ -16,9 +17,7 @@ class HostContext:
         if not asset_name:
             return project_name
 
-        from avalon import io
-
-        asset_doc = io.find_one(
+        asset_doc = legacy_io.find_one(
             {"type": "asset", "name": asset_name},
             {"data.parents": 1}
         )
diff --git a/openpype/hosts/testhost/plugins/create/auto_creator.py b/openpype/hosts/testhost/plugins/create/auto_creator.py
index 4c22eea9dd..06b95375b1 100644
--- a/openpype/hosts/testhost/plugins/create/auto_creator.py
+++ b/openpype/hosts/testhost/plugins/create/auto_creator.py
@@ -1,7 +1,7 @@
-from avalon import io
 from openpype.lib import NumberDef
 from openpype.hosts.testhost.api import pipeline
 from openpype.pipeline import (
+    legacy_io,
     AutoCreator,
     CreatedInstance,
 )
@@ -38,13 +38,16 @@ class MyAutoCreator(AutoCreator):
                 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"]
+        project_name = legacy_io.Session["AVALON_PROJECT"]
+        asset_name = legacy_io.Session["AVALON_ASSET"]
+        task_name = legacy_io.Session["AVALON_TASK"]
+        host_name = legacy_io.Session["AVALON_APP"]
 
         if existing_instance is None:
-            asset_doc = io.find_one({"type": "asset", "name": asset_name})
+            asset_doc = legacy_io.find_one({
+                "type": "asset",
+                "name": asset_name
+            })
             subset_name = self.get_subset_name(
                 variant, task_name, asset_doc, project_name, host_name
             )
@@ -66,7 +69,10 @@ class MyAutoCreator(AutoCreator):
             existing_instance["asset"] != asset_name
             or existing_instance["task"] != task_name
         ):
-            asset_doc = io.find_one({"type": "asset", "name": asset_name})
+            asset_doc = legacy_io.find_one({
+                "type": "asset",
+                "name": asset_name
+            })
             subset_name = self.get_subset_name(
                 variant, task_name, asset_doc, project_name, host_name
             )
diff --git a/openpype/hosts/traypublisher/api/pipeline.py b/openpype/hosts/traypublisher/api/pipeline.py
index 24175883d9..954a0bae47 100644
--- a/openpype/hosts/traypublisher/api/pipeline.py
+++ b/openpype/hosts/traypublisher/api/pipeline.py
@@ -3,11 +3,12 @@ import json
 import tempfile
 import atexit
 
-from avalon import io
-import avalon.api
 import pyblish.api
 
-from openpype.pipeline import register_creator_plugin_path
+from openpype.pipeline import (
+    register_creator_plugin_path,
+    legacy_io,
+)
 
 ROOT_DIR = os.path.dirname(os.path.dirname(
     os.path.abspath(__file__)
@@ -175,6 +176,6 @@ def install():
 def set_project_name(project_name):
     # TODO Deregister project specific plugins and register new project plugins
     os.environ["AVALON_PROJECT"] = project_name
-    avalon.api.Session["AVALON_PROJECT"] = project_name
-    io.install()
+    legacy_io.Session["AVALON_PROJECT"] = project_name
+    legacy_io.install()
     HostContext.set_project_name(project_name)