From d7202e8f5142b5a9a4d31d57f0303c12f8272119 Mon Sep 17 00:00:00 2001
From: Paul McCarthy <pauldmccarthy@gmail.com>
Date: Sat, 10 Aug 2019 16:17:49 +0100
Subject: [PATCH] RF: Add tree attribute to FileTreeQuery for convenience
---
fsl/utils/filetree/query.py | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/fsl/utils/filetree/query.py b/fsl/utils/filetree/query.py
index 3e2733747..49433fa54 100644
--- a/fsl/utils/filetree/query.py
+++ b/fsl/utils/filetree/query.py
@@ -149,6 +149,7 @@ class FileTreeQuery(object):
tarr[tuple(idx)] = match
+ self.__tree = tree
self.__allvars = allvars
self.__templatevars = templatevars
self.__matches = matches
@@ -179,6 +180,14 @@ class FileTreeQuery(object):
return {var : list(self.__allvars[var]) for var in varnames}
+ @property
+ def tree(self):
+ """Returns the :class:`.FileTree` associated with this
+ ``FileTreeQuery``.
+ """
+ return self.__tree
+
+
@property
def templates(self) -> List[str]:
"""Returns a list containing all templates of the ``FileTree`` that
--
GitLab