1 # -*- coding: utf-8 -*-
2 ##############################################################################
4 # OpenERP, Open Source Management Solution
5 # Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
7 # This program is free software: you can redistribute it and/or modify
8 # it under the terms of the GNU Affero General Public License as
9 # published by the Free Software Foundation, either version 3 of the
10 # License, or (at your option) any later version.
12 # This program 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
15 # GNU Affero General Public License for more details.
17 # You should have received a copy of the GNU Affero General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
20 ##############################################################################
23 Some functions related to the os and os.path module
27 from os.path import join as opj
29 def listdir(dir, recursive=False):
30 """Allow to recursively get the file listing"""
31 dir = os.path.normpath(dir)
33 return os.listdir(dir)
36 for root, dirs, files in walksymlinks(dir):
37 root = root[len(dir)+1:]
38 res.extend([opj(root, f) for f in files])
41 def walksymlinks(top, topdown=True, onerror=None):
43 same as os.walk but follow symlinks
44 attention: all symlinks are walked before all normals directories
46 for dirpath, dirnames, filenames in os.walk(top, topdown, onerror):
48 yield dirpath, dirnames, filenames
50 symlinks = filter(lambda dirname: os.path.islink(os.path.join(dirpath, dirname)), dirnames)
52 for x in walksymlinks(os.path.join(dirpath, s), topdown, onerror):
56 yield dirpath, dirnames, filenames
60 if __name__ == '__main__':
61 from pprint import pprint as pp
62 pp(listdir('../report', True))
64 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: