1HEADmaster

1 files changed, 61 insertions, 0 deletions

diff --git a/devtools/WildcardSearch.py b/devtools/WildcardSearch.py
new file mode 100644
index 0000000..7477644
--- /dev/null
+++ b/devtools/WildcardSearch.py
@@ -0,0 +1,61 @@
+
+from __future__ import generators
+import os
+import re
+import stat
+
+
+# This takes a DOS filename wildcard like *abc.t?t and returns a regex string that will match it.
+def GetRegExForDOSWildcard( wildcard ):
+	# First find the base directory name.
+	iLast = wildcard.rfind( "/" )
+	if iLast == -1:
+		iLast = wildcard.rfind( "\\" )
+
+	if iLast == -1:
+		dirName = "."
+		dosStyleWildcard = wildcard
+	else:
+		dirName = wildcard[0:iLast]
+		dosStyleWildcard = wildcard[iLast+1:]
+		
+	# Now generate a regular expression for the search.
+	# DOS     -> RE
+	# *       -> .*
+	# .       -> \.
+	# ?       -> .
+	reString = dosStyleWildcard.replace( ".", r"\." ).replace( "*", ".*" ).replace( "?", "." )
+	return reString
+
+
+#
+# Useful function to return a list of files in a directory based on a dos-style wildcard like "*.txt"
+#
+# for name in WildcardSearch( "d:/hl2/src4/dt*.cpp", 1 ):
+#	print name
+#
+def WildcardSearch( wildcard, bRecurse=0 ):
+	reString = GetRegExForDOSWildcard( wildcard )
+	matcher = re.compile( reString, re.IGNORECASE )
+
+	return __GetFiles_R( matcher, dirName, bRecurse )
+
+def __GetFiles_R( matcher, dirName, bRecurse ):
+	fileList = []
+	# For each file, see if we can find the regular expression.
+	files = os.listdir( dirName )
+	for baseName in files:
+		filename = dirName + "/" + baseName
+
+		mode = os.stat( filename )[stat.ST_MODE]
+		if stat.S_ISREG( mode ):
+			# Make sure the file matches the search string.
+			if matcher.match( baseName ):
+				fileList.append( filename )
+
+		elif bRecurse and stat.S_ISDIR( mode ):
+			fileList += __GetFiles_R( matcher, filename, bRecurse )
+
+	return fileList				
+	
+