Commit 0df4a512 authored by Alice Donini's avatar Alice Donini
Browse files

update tree script

parent 66cb0a27
...@@ -37,7 +37,7 @@ def makedir(name): ...@@ -37,7 +37,7 @@ def makedir(name):
# Create target Directory if don't exist # Create target Directory if don't exist
if not os.path.exists(name): if not os.path.exists(name):
os.makedirs(name) os.makedirs(name)
print("Directory ", name, " Created ") print("Directory ", name, " created ")
else: else:
print("Directory ", name, " already exists") print("Directory ", name, " already exists")
return None return None
...@@ -64,31 +64,33 @@ def main(): ...@@ -64,31 +64,33 @@ def main():
) )
parser.add_argument( parser.add_argument(
'--night', type=str, required=True, dest='night', '--night', nargs='+', type=int, required=True, dest='night',
help='Night date' help='Night date'
) )
parser.add_argument( parser.add_argument(
'--version', type=str, required=False, dest='version', '--version', type=str, required=False, dest='version',
default='v0.9.2', help='lstchain version' default='v0.9.2', help='lstchain version (default: %(default)s)'
) )
parser.add_argument( parser.add_argument(
'--cleaning', type=str, required=False, dest='cleaning', '--cleaning', type=str, required=False, dest='cleaning',
default='tailcut84', help='Cleaning type' default='tailcut84', help='Cleaning type (default: %(default)s)'
) )
args = parser.parse_args() args = parser.parse_args()
# Define analysis directories and subdirectories # Define analysis directories and subdirectories
data_dir = ['DL1', 'DL2', 'DL3'] data_dir = ['DL1', 'DL2', 'DL3']
structure = f'{args.source}/{args.night}/{args.version}/{args.cleaning}' nights = args.night
for folder in data_dir: for night in nights:
dirName = os.path.join(f'{args.main_dir}', str(folder), structure) structure = f'{args.source}/{night}/{args.version}/{args.cleaning}'
makedir(dirName) for folder in data_dir:
dirName = os.path.join(f'{args.main_dir}', str(folder), structure)
makedir(dirName)
print(f'Directory structure ready for analysis on {args.source}.') print(f'Directory structure for analysis on {args.source} was created.')
if __name__ == "__main__": if __name__ == "__main__":
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment