-
Notifications
You must be signed in to change notification settings - Fork 1
/
Make_Directory_Structure.py
76 lines (55 loc) · 2.37 KB
/
Make_Directory_Structure.py
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
#!/usr/bin/env python
# coding: utf-8
import os
import numpy as np
from zipfile import ZipFile
basepath = '/homes/picaro/bellt/research/'
planets = ['CoRoT-2b', 'HAT-P-7b', 'KELT-1b', 'KELT-16b', 'KELT-9b', 'MASCARA-1b', 'Qatar-1b', 'WASP-103b', 'WASP-12b', 'WASP-12b_old', 'WASP-14b', 'WASP-18b', 'WASP-19b', 'WASP-33b', 'WASP-43b', 'HD189733b', 'HD209458b', 'HD149026b']
for planet in planets:
rootpath = basepath+planet+'/'
print('Starting', rootpath)
zips = os.listdir(rootpath)
zips = [zips[i] for i in range(len(zips)) if '.zip' in zips[i]]
channels = []
for i in range(len(zips)):
with ZipFile(rootpath+zips[i], 'r') as zipObj:
channels.append(zipObj.namelist()[0].split('/')[1])
channels_uni = np.unique(channels)
aors = []
for i in range(len(zips)):
with ZipFile(rootpath+zips[i], 'r') as zipObj:
aors.extend([zipObj.namelist()[j].split('/')[0] for j in range(len(zipObj.namelist()))])
aors = np.unique(aors)
aors = [aor for aor in aors if '.txt' not in aor]
os.mkdir(rootpath+'data')
for channel in channels_uni:
os.mkdir(rootpath+'data/'+channel)
for i in range(len(zips)):
with ZipFile(rootpath+zips[i], 'r') as zipObj:
zipObj.extractall(rootpath+'data/'+channels[i])
os.mkdir(rootpath+'analysis')
for channel in channels_uni:
os.mkdir(rootpath+'analysis/'+channel)
os.mkdir(rootpath+'analysis/frameDiagnostics')
for channel in channels_uni:
os.mkdir(rootpath+'analysis/frameDiagnostics/'+channel)
os.mkdir(rootpath+'analysis/photometryComparison')
for channel in channels_uni:
os.mkdir(rootpath+'analysis/photometryComparison/'+channel)
os.mkdir(rootpath+'raw')
for i in range(len(zips)):
os.rename(rootpath+zips[i], rootpath+'raw/'+zips[i])
aorSnip = ''
if np.all([aor[:5]==aors[0][:5] for aor in aors[1:]]):
aorSnip = aors[0][:5]
elif np.all([aor[:4]==aors[0][:4] for aor in aors[1:]]):
aorSnip = aors[0][:4]
elif np.all([aor[:3]==aors[0][:3] for aor in aors[1:]]):
aorSnip = aors[0][:3]
elif np.all([aor[:2]==aors[0][:2] for aor in aors[1:]]):
aorSnip = aors[0][:2]
else:
aorSnip = aors[0][:1]
print('Your AOR snippet is', aorSnip)
with open(rootpath+'analysis/aorSnippet.txt', 'w') as f:
f.write(aorSnip)