rydberg_experiment_tools/makeAnalysisFolders.py at master · KillianRice/rydberg_experiment_tools · GitHub

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
import os
import datetime
import shutil
import sys

ver = sys.version_info[0]
#get date and create timestamps
dateWithPeriods = datetime.datetime.today().strftime('%Y.%m.%d')
dateWithoutPeriods = datetime.datetime.today().strftime('%Y%m%d')

#ask for base name and isotope
baseName = input('Enter base name: ')
isotope = input('Enter Isotope: ')
while isotope not in ['84','86','87','88']:
    isotope = input('Please select valid isotope: ')

#base folder should be wherever you are storing your data
baseFolder = 'F:\\'
path_to_notepad = '"C:\\Program Files (x86)\\Notepad++\\notepad++.exe"'
repository_url = 'https://github.com/KillianRice/rydberg_analysis.git'

#change to the base folder
os.chdir(baseFolder)

#verify that there is a folder for the isotope you selected, if not
#make one
try:
    os.chdir(baseFolder+'\\Analysis\\'+isotope+'Sr')
except:
    os.mkdir(baseFolder+'\\Analysis\\'+isotope+'Sr')

try:
    os.chdir(baseFolder+'\\Raw_Data\\'+isotope+'Sr')
except:
    os.mkdir(baseFolder+'\\Raw_Data\\'+isotope+'Sr')

analysisFolder = baseFolder+'\\Analysis\\'+isotope+'Sr'
dataFolder = baseFolder+'\\Raw_Data\\'+isotope+'Sr'

#create the directories that will contain the data and analysis
try:
	os.listdir(analysisFolder+'\\'+dateWithPeriods+'_'+baseName)
except:
	os.mkdir(analysisFolder+'\\'+dateWithPeriods+'_'+baseName)
try:
	os.listdir(dataFolder+'\\'+dateWithPeriods)
except:
    os.mkdir(dataFolder+'\\'+dateWithPeriods)

todayAnalysisFolder = analysisFolder+'\\'+dateWithPeriods+'_'+baseName
todayDataFolder = dataFolder+'\\'+dateWithPeriods

#clone most up to date version of analysis code
if os.listdir(analysisFolder+'\\'+dateWithPeriods+'_'+baseName) == []:
	os.system('git clone ' + repository_url + ' ' + todayAnalysisFolder)

#create the master batch files
master_batch_atom = todayDataFolder + '\\Files_' + dateWithoutPeriods + '.txt.'
try:
	f = open(master_batch_atom,'r')
	f.close()
except:
	f = open(master_batch_atom,'w+')
	f.close()

try:
	f = open(todayDataFolder+'\\Files_'+dateWithoutPeriods+'_Bg.txt','r')
	f.close()
except:
	f = open(todayDataFolder+'\\Files_'+dateWithoutPeriods+'_Bg.txt','w+')
	f.close()

try:
	f = open(todayDataFolder+'\\Files_'+dateWithoutPeriods+'_Counts.txt','r')
	f.close()
except:
	f = open(todayDataFolder+'\\Files_'+dateWithoutPeriods+'_Counts.txt','w+')
	f.close()

try:
	f = open(todayDataFolder+'\\custom_scan.csv','r')
	f.close()
except:
	f = open(todayDataFolder+'\\custom_scan.csv','w+')
	f.close()

os.system("pause")

#open the master batch file for convenience
os.system('START /B ' + path_to_notepad + ' ' + master_batch_atom)