def dir_list5(dir_name,
*args):
fileList
= []
for file in os.listdir(dir_name):
dirfile
= os.path.join(dir_name,
file)
if os.path.isfile(dirfile): # skip
directories
if len(args) == 0:
fileList.append('%s\n' %(dirfile))
else:
if os.path.splitext(dirfile)[1][1:]
in args:
fileList.append('%s\n' %(dirfile))
return
"".join(fileList)
if __name__ == '__main__':
def run_script():
dir_name = (os.path.join('H:\\',
'TEMP', 'temsys'))
fList
= dir_list2(dir_name, 'txt')
data_file = (os.path.join('H:\\', 'TEMP', 'temsys',
'student_data.dat'))
try:
fData = open(data_file, 'w')
print
for f in fList.split():
ff =
open(f, "r")
print
f
for
line in ff:
if
'student ID:' in line:
# print
line[line.index('student ID:')+11:]
print
line.split('student ID:')[1]
fData.write("".join([os.path.basename(f),'----',line.split('student ID:')[1]]))
break
ff.close()
fData.close()
except:
raise IOError, "Could not open data file or file open
error"
#import profile
#profile.run('run_script()')
run_script()