Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
"""Create the directory structure for an analysis.
Change the default value for the Parent folder in the argparse
Parent_Folder
└──DL1
├── source
│ └── night
│ └── version
│ └── cleaning
DL2
├── source
│ └── night
│ └── version
│ └── cleaning
DL3
└── source
└── night
└── version
└── cleaning
"""
import os
import argparse
from argparse import RawTextHelpFormatter
def makedir(name):
"""
Create folder if non-existent and output OS error if any.
Parameters
----------
name : str
Name of the analysis.
"""
# Create target Directory if don't exist
if not os.path.exists(name):
os.makedirs(name)
print("Directory ", name, " Created ")
else:
print("Directory ", name, " already exists")
return None
def main():
# define command-line arguments
description = 'Create a directory structure'
parser = argparse.ArgumentParser(
description=description,
formatter_class=RawTextHelpFormatter
)
parser.add_argument(
'--main_dir', type=str, required=False, dest='main_dir',
default='/fefs/aswg/workspace/alice.donini/Analysis/data',
help='Path to parent folder'
)
parser.add_argument(
'--source', type=str, required=True, dest='source',
help='Source name'
)
parser.add_argument(
'--night', type=str, required=True, dest='night',
help='Night date'
)
parser.add_argument(
'--version', type=str, required=False, dest='version',
default='v0.9.2', help='lstchain version'
)
parser.add_argument(
'--cleaning', type=str, required=False, dest='cleaning',
default='tailcut84', help='Cleaning type'
)
args = parser.parse_args()
# Define analysis directories and subdirectories
data_dir = ['DL1', 'DL2', 'DL3']
structure = f'{args.source}/{args.night}/{args.version}/{args.cleaning}'
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}.')
if __name__ == "__main__":
main()