in_dir = sys.argv[1]
except:
print "Please specify input_file."
- exit
+ sys.exit()
for dir in dirs:
srcdir = os.path.join (in_dir, dir)
- destdir = os.path.join (os.getcwd(), 'lsr', dir)
+ destdir = os.path.join ('input', 'lsr', dir)
+ if not(os.path.isdir(destdir)):
+ print "Please run this script from the head of the source tree,"
+ print " and/or check that you have the right categories."
+ sys.exit()
file_names = os.listdir (destdir)
for file in file_names:
if (file.endswith ('.ly')):
- if (file[:3] != 'AAA'): # or whatever we use to denote the first file
+ if (file[:3] != 'AAA'):
os.remove( os.path.join(destdir,file) )
file_names = os.listdir (in_dir + dir)