Merge pull request #512 from ynput/get-used-project-tags

Helpers: get_used_project_tags helper

Author: martastain

ayon_server/graphql/nodes/project.py

@@ -18,6 +18,7 @@
 from ayon_server.graphql.resolvers.versions import get_version, get_versions
 from ayon_server.graphql.resolvers.workfiles import get_workfile, get_workfiles
 from ayon_server.graphql.utils import parse_attrib_data
+from ayon_server.helpers.tags import get_used_project_tags
 from ayon_server.lib.postgres import Postgres
 from ayon_server.utils import json_dumps
 
@@ -203,6 +204,10 @@ async def product_types(self) -> list[ProductType]:
             )
         ]
 
+    @strawberry.field(description="List of tags used in the project")
+    async def used_tags(self) -> list[str]:
+        return await get_used_project_tags(self.project_name)
+
 
 def project_from_record(
     project_name: str | None, record: dict[str, Any], context: dict[str, Any]

ayon_server/helpers/tags.py

@@ -0,0 +1,30 @@
+from ayon_server.lib.postgres import Postgres
+
+
+async def get_used_project_tags(project_name: str) -> list[str]:
+    """Returns a list of tags that are used in the project.
+
+    This function is used to get all the tags that are used in the project.
+    The tags are collected from all the entities, but don't necessarily
+    contain tags defined in the project anatomy.
+    """
+    result = []
+    project_schema = f"project_{project_name.lower()}"
+    query = f"""
+    SELECT DISTINCT tag FROM (
+        SELECT unnest(tags) AS tag FROM {project_schema}.folders
+        UNION ALL
+        SELECT unnest(tags) FROM {project_schema}.products
+        UNION ALL
+        SELECT unnest(tags) FROM {project_schema}.tasks
+        UNION ALL
+        SELECT unnest(tags) FROM {project_schema}.versions
+        UNION ALL
+        SELECT unnest(tags) FROM {project_schema}.representations
+        UNION ALL
+        SELECT unnest(tags) FROM {project_schema}.workfiles
+    ) t
+    """
+    async for row in Postgres.iterate(query):
+        result.append(row["tag"])
+    return result