marcel/pylot
3
0
Fork 0
Code Issues 7 Pull Requests 1 Releases Wiki Activity

[enhancement] added capability of searching for files in subdirectories

Browse Source
This commit is contained in:
Sebastian Wehling-Benatelli 2016-08-09 13:10:25 +02:00
parent a4c7e61e1a
commit f8623fc454
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pylot/core/pick/compare.py
View File

@ -345,6 +345,7 @@ class PDFstatistics(object):
''' '''
def __init__(self, directory): def __init__(self, directory):
self.directory = directory self.directory = directory
self.evtlist = list()
self.return_phase = None self.return_phase = None
def readTheta(self, arname, dir, fnpattern): def readTheta(self, arname, dir, fnpattern):
@ -358,10 +359,15 @@ class PDFstatistics(object):
exec('self.' + arname + ' += list') exec('self.' + arname + ' += list')
fid.close() fid.close()
def makeFileList(self, fn_pattern='*.xml'):
def makeFileList(self, fn_pattern='*'): evtlist = list()
self.evtlist = glob.glob1((os.path.join(self.directory)), '*.xml') evtlist = glob.glob1((os.path.join(self.directory)), fn_pattern)
if not evtlist:
for root, _, files in os.walk(self.directory):
for file in files:
if file.endswith(fn_pattern[1:]):
evtlist.append(os.path.join(root, file))
self.evtlist = evtlist
def __iter__(self): def __iter__(self):
assert isinstance(self.return_phase, str), 'phase has to be set before being able to iterate over items...' assert isinstance(self.return_phase, str), 'phase has to be set before being able to iterate over items...'