From 9f197dea76995fc2916884e341fb385775e39d2c Mon Sep 17 00:00:00 2001 From: Vijay Kumar Banerjee Date: Mon, 20 May 2019 16:53:51 +0530 Subject: rtemstoolkit/path : Add listdir function --- rtemstoolkit/path.py | 29 ++++++++++++++++++++++++++++- 1 file changed, 28 insertions(+), 1 deletion(-) (limited to 'rtemstoolkit') diff --git a/rtemstoolkit/path.py b/rtemstoolkit/path.py index 2b569ae..6b63c8a 100644 --- a/rtemstoolkit/path.py +++ b/rtemstoolkit/path.py @@ -157,6 +157,33 @@ def expanduser(path): path = os.path.expanduser(path) return shell(path) +def listdir(path): + path = host(path) + files = [] + if not exists(path): + raise error.general('path does not exist : %s' % (path)) + elif not isdir(path): + raise error.general('path is not a directory: %s' % (path)) + else: + if windows: + try: + files = os.listdir(host(path)) + except IOError: + raise error.general('Could not list files: %s' % (path)) + except OSError as e: + raise error.general('Could not list files: %s: %s' % (path, str(e))) + except WindowsError as e: + raise error.general('Could not list files: %s: %s' % (path, str(e))) + else: + try: + files = os.listdir(host(path)) + except IOError: + raise error.general('Could not list files: %s' % (path)) + except OSError as e: + raise error.general('Could not list files: %s: %s' % (path, str(e))) + + return files + def collect_files(path_): # # Convert to shell paths and return shell paths. @@ -184,7 +211,7 @@ def copy_tree(src, dst): hdst = host(dst) if os.path.exists(src) and os.path.isdir(src): - names = os.listdir(src) + names = listdir(src) else: names = [basename(src)] src = dirname(src) -- cgit v1.2.3