From 0359ec6da4e03bf9c7df49060a423c028195d86c Mon Sep 17 00:00:00 2001
From: "clement.hector" <clement.hector@fixstudio.com>
Date: Thu, 7 Apr 2022 15:46:40 +0200
Subject: [PATCH] create kitsu collector

---
 .../publish/collect_kitsu_credential.py       | 18 ++++++
 .../plugins/publish/collect_kitsu_entities.py | 64 +++++++++++++++++++
 2 files changed, 82 insertions(+)
 create mode 100644 openpype/modules/kitsu/plugins/publish/collect_kitsu_credential.py
 create mode 100644 openpype/modules/kitsu/plugins/publish/collect_kitsu_entities.py

diff --git a/openpype/modules/kitsu/plugins/publish/collect_kitsu_credential.py b/openpype/modules/kitsu/plugins/publish/collect_kitsu_credential.py
new file mode 100644
index 0000000000..bd0af16c8b
--- /dev/null
+++ b/openpype/modules/kitsu/plugins/publish/collect_kitsu_credential.py
@@ -0,0 +1,18 @@
+import os
+
+import gazu
+
+import pyblish.api
+
+
+class CollectKitsuSession(pyblish.api.ContextPlugin):
+    """Collect Kitsu session using user credentials"""
+
+    order = pyblish.api.CollectorOrder
+    label = "Kitsu user session"
+
+
+    def process(self, context):
+
+        gazu.client.set_host(os.environ["KITSU_SERVER"])
+        gazu.log_in(os.environ["KITSU_LOGIN"], os.environ["KITSU_PWD"])
\ No newline at end of file
diff --git a/openpype/modules/kitsu/plugins/publish/collect_kitsu_entities.py b/openpype/modules/kitsu/plugins/publish/collect_kitsu_entities.py
new file mode 100644
index 0000000000..e4773f7b2a
--- /dev/null
+++ b/openpype/modules/kitsu/plugins/publish/collect_kitsu_entities.py
@@ -0,0 +1,64 @@
+import os
+
+import gazu
+
+import pyblish.api
+
+
+class CollectKitsuEntities(pyblish.api.ContextPlugin):
+    """Collect Kitsu entities according to the current context"""
+
+    order = pyblish.api.CollectorOrder + 0.499
+    label = "Kitsu entities"
+
+    def process(self, context):
+
+        os.environ["AVALON_PROJECT"],
+        os.environ["AVALON_ASSET"],
+        os.environ["AVALON_TASK"],
+        os.environ["AVALON_APP_NAME"]
+
+        asset_data = context.data["assetEntity"]["data"]
+        zoo_asset_data = asset_data.get("zou")
+        if not zoo_asset_data:
+            raise
+
+        kitsu_project = gazu.project.get_project(zoo_asset_data["project_id"])
+        if not kitsu_project:
+            raise
+        context.data["kitsu_project"] = kitsu_project
+
+        kitsu_asset = gazu.asset.get_asset(zoo_asset_data["entity_type_id"])
+        if not kitsu_asset:
+            raise
+        context.data["kitsu_asset"] = kitsu_asset
+
+        # kitsu_task_type = gazu.task.get_task_type_by_name(instance.data["task"])
+        # if not kitsu_task_type:
+        #     raise
+        # context.data["kitsu_task_type"] = kitsu_task_type
+
+        zoo_task_data = asset_data["tasks"][os.environ["AVALON_TASK"]].get("zou")
+        kitsu_task = gazu.task.get_task(
+            asset_data["zou"],
+            kitsu_task_type
+        )
+        if not kitsu_task:
+            raise
+        context.data["kitsu_task"] = kitsu_task
+
+        wip = gazu.task.get_task_status_by_short_name("wip")
+
+        task = gazu.task.get_task_by_name(asset, modeling)
+        comment = gazu.task.add_comment(task, wip, "Change status to work in progress")
+
+        person = gazu.person.get_person_by_desktop_login("john.doe")
+
+        # task_type = gazu.task.get_task_type_by_name(instance.data["task"])
+
+        # entity_task = gazu.task.get_task_by_entity(
+        #     asset_data["zou"],
+        #     task_type
+        # )
+
+        raise
\ No newline at end of file