62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
|
|
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
|
|
|
|
|