1 # -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2008 CEA/DEN, EDF R&D, OPEN CASCADE
4 # Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
5 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
7 # This library is free software; you can redistribute it and/or
8 # modify it under the terms of the GNU Lesser General Public
9 # License as published by the Free Software Foundation; either
10 # version 2.1 of the License.
12 # This library is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 # Lesser General Public License for more details.
17 # You should have received a copy of the GNU Lesser General Public
18 # License along with this library; if not, write to the Free Software
19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
25 __date__ ="$21 mai 2010 18:00:23$"
28 def findFiles(rootpath, excludes=[]):
30 This looks after files recursively from the specified rootpath,
31 but without visiting directories whose basename is in the list
34 if not os.path.exists(rootpath):
35 raise RuntimeError("the path %s does not exists"%rootpath)
38 for excludepath in excludes:
39 exclude_options+="-e %s "%excludepath
42 stream=os.popen("find %s -type f | grep -v -e '\.svn' %s 2>/dev/null"%(rootpath,exclude_options))
43 for line in stream.readlines():
44 listfiles.append(line.split('\n')[0])
49 from stat import ST_MODE, S_ISDIR, S_ISREG
50 def walktree(rootpath, callback, **kwargs):
52 This recursively descends the directory tree rooted at rootpath,
53 calling the callback function for each regular file
55 for f in os.listdir(rootpath):
56 pathname = os.path.join(rootpath, f)
58 mode = os.stat(pathname)[ST_MODE]
60 # It probably means that the file is a broken inode
63 # It's a directory, recurse into it
64 walktree(pathname, callback, **kwargs)
66 # It's a file, call the callback function
67 callback(pathname, **kwargs)
69 # Unknown file type, print a message
70 print 'Skipping %s' % pathname
74 # =============================================================
75 # Use cases and unit test functions
76 # =============================================================
80 TESTDOCDIR=os.path.join(os.environ["KERNEL_ROOT_DIR"],"share")
85 print "########## find 1"
87 listfiles=findFiles(rootpath)
88 for filename in listfiles:
91 print "########## find 2"
92 excludes=[os.path.join(TESTDOCDIR,"doc")]
93 listfiles=findFiles(rootpath,excludes)
94 for filename in listfiles:
99 # This is the test callback function
100 def visitfile_withargs(file, rootid):
101 print 'visiting file %s (rootid=%s)'%(file,str(rootid))
103 def visitfile_withoutargs(file):
104 print 'visiting file %s'%(file)
107 #walktree(TESTDOCDIR, visitfile_withargs, rootid=2)
108 walktree(TESTDOCDIR, visitfile_withoutargs)
111 if __name__ == "__main__":
113 unittester.run("syshelper", "TEST_findFiles")
114 unittester.run("syshelper", "TEST_walktree")