Merge pull request #696 from python-cmd2/script_refactor

Script refactor

1 files changed, 29 insertions, 0 deletions

diff --git a/cmd2/utils.py b/cmd2/utils.py
index 54ad763d..3500ba7a 100644
--- a/cmd2/utils.py
+++ b/cmd2/utils.py
@@ -2,6 +2,7 @@
 """Shared utility functions"""
 
 import collections
+import glob
 import os
 import re
 import subprocess
@@ -319,6 +320,34 @@ def find_editor() -> str:
     return editor
 
 
+def files_from_glob_pattern(pattern: str, access=os.F_OK) -> List[str]:
+    """Return a list of file paths based on a glob pattern.
+
+    Only files are returned, not directories, and optionally only files for which the user has a specified access to.
+
+    :param pattern: file name or glob pattern
+    :param access: file access type to verify (os.* where * is F_OK, R_OK, W_OK, or X_OK)
+    :return: list of files matching the name or glob pattern
+    """
+    return [f for f in glob.glob(pattern) if os.path.isfile(f) and os.access(f, access)]
+
+
+def files_from_glob_patterns(patterns: List[str], access=os.F_OK) -> List[str]:
+    """Return a list of file paths based on a list of glob patterns.
+
+    Only files are returned, not directories, and optionally only files for which the user has a specified access to.
+
+    :param patterns: list of file names and/or glob patterns
+    :param access: file access type to verify (os.* where * is F_OK, R_OK, W_OK, or X_OK)
+    :return: list of files matching the names and/or glob patterns
+    """
+    files = []
+    for pattern in patterns:
+        matches = files_from_glob_pattern(pattern, access=access)
+        files.extend(matches)
+    return files
+
+
 class StdSim(object):
     """
     Class to simulate behavior of sys.stdout or sys.stderr.