From 5365b81d146415bca65893325934a2acbb542c41 Mon Sep 17 00:00:00 2001
From: Timur Vasyunin <kpod13@gmail.com>
Date: Thu, 10 Nov 2016 15:00:27 +0300
Subject: [PATCH] Fix for import 'display' in
 lib/ansible/plugins/lookup/filetree.py

---
 lib/ansible/plugins/lookup/filetree.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/lib/ansible/plugins/lookup/filetree.py b/lib/ansible/plugins/lookup/filetree.py
index 7ff5621214..3cc6796f0c 100644
--- a/lib/ansible/plugins/lookup/filetree.py
+++ b/lib/ansible/plugins/lookup/filetree.py
@@ -32,8 +32,11 @@ except ImportError:
 from ansible.plugins.lookup import LookupBase
 from ansible.module_utils._text import to_native
 
-from __main__ import display
-
+try:
+    from __main__ import display
+except ImportError:
+    from ansible.utils.display import Display
+    display = Display()
 
 # If selinux fails to find a default, return an array of None
 def selinux_context(path):