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

update tree script

parent 66cb0a27

create_analysis_tree.py

+11 −9
Original line number Original line Diff line number Diff line
@@ -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(): 
    )

    )





    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__":