From 675aa28eacf61bdc8798110b40406981da2e6c9f Mon Sep 17 00:00:00 2001
From: iLLiCiTiT <jakub.trllo@gmail.com>
Date: Wed, 6 Jan 2021 16:38:58 +0100
Subject: [PATCH] replaced `rootless` kwarg with `template_key`

---
 pype/lib/avalon_context.py | 26 +++++++++++++++++---------
 1 file changed, 17 insertions(+), 9 deletions(-)

diff --git a/pype/lib/avalon_context.py b/pype/lib/avalon_context.py
index 4a60cbc214..6e3ba1880c 100644
--- a/pype/lib/avalon_context.py
+++ b/pype/lib/avalon_context.py
@@ -276,7 +276,7 @@ def get_workdir_data(project_doc, asset_doc, task_name, host_name):
 
 
 def get_workdir_with_workdir_data(
-    workdir_data, anatomy=None, project_name=None, rootless=False
+    workdir_data, anatomy=None, project_name=None, template_key=None
 ):
     """Fill workdir path from entered data and project's anatomy.
 
@@ -290,7 +290,8 @@ def get_workdir_with_workdir_data(
             `project_name` is entered.
         project_name (str): Project's name. Optional if `anatomy` is entered
             otherwise Anatomy object is created with using the project name.
-        rootless (bool): Return workdir without filled root key when `True`.
+        template_key (str): Key of work templates in anatomy templates. By
+            default is seto to `"work"`.
 
     Returns:
         str: Workdir path.
@@ -306,15 +307,21 @@ def get_workdir_with_workdir_data(
 
     if not anatomy:
         anatomy = Anatomy(project_name)
+
+    if not template_key:
+        template_key = "work"
+
     anatomy_filled = anatomy.format(workdir_data)
-    result = anatomy_filled["work"]["folder"]
-    if rootless:
-        result = result.rootless
-    return os.path.normpath(result)
+    return anatomy_filled[template_key]["folder"]
 
 
 def get_workdir(
-    project_doc, asset_doc, task_name, host_name, anatomy=None, rootless=False
+    project_doc,
+    asset_doc,
+    task_name,
+    host_name,
+    anatomy=None,
+    template_key=None
 ):
     """Fill workdir path from entered data and project's anatomy.
 
@@ -329,7 +336,8 @@ def get_workdir(
             project name from `project_doc`. It is preffered to pass this
             argument as initialization of a new Anatomy object may be time
             consuming.
-        rootless (bool): Return workdir without filled root key when `True`.
+        template_key (str): Key of work templates in anatomy templates. Default
+            value is defined in `get_workdir_with_workdir_data`.
 
     Returns:
         str: Workdir path.
@@ -340,7 +348,7 @@ def get_workdir(
     workdir_data = get_workdir_data(
         project_doc, asset_doc, task_name, host_name
     )
-    return get_workdir_with_workdir_data(workdir_data, anatomy, rootless)
+    return get_workdir_with_workdir_data(workdir_data, anatomy, template_key)
 
 
 class BuildWorkfile: