-
Notifications
You must be signed in to change notification settings - Fork 0
/
insertDB.py
39 lines (31 loc) · 1.04 KB
/
insertDB.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
import io
import os
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--crys',dest='crys')
parser.add_argument('--runs',dest='runs')
parser.add_argument('--tag',dest='tag')
parser.add_argument('--id',dest='xtalid')
parser.add_argument('--ledruns',dest='ledRuns')
parser.add_argument('--refruns',dest='refRuns')
parser.add_argument('--prod',dest='producer')
parser.add_argument('--geom',dest='geometry')
parser.add_argument('--db',dest='db')
args = parser.parse_args()
crys=str(args.crys)
tag=str(args.tag)
xtalid=int(args.xtalid)
runs=str(args.runs).split(',')
refRuns=str(args.refRuns).split(',')
ledRuns=str(args.ledRuns).split(',')
prod=str(args.producer)
geo=str(args.geometry)
if (int(crys.split('BAR0')[1]) != xtalid):
print('ERROR. Check that ID %s matches crystal number %d'%(crys,xtalid))
exit(-1)
import measDB
f=measDB.MeasDB(args.db)
f.insertMeas(crys,prod,geo,runs,refRuns,ledRuns,tag)
print('Runs for this xtal %s'%str(f.getMeas(crys)))
print('------ Number of crystals in DB %d ------'%len(f.getXtals()))
f.save()