def test_insert_constants(expname=TEST_EXPNAME, detname=TEST_DETNAME, ctype='test_ctype', runnum=10, data='test text sampele'):
""" Inserts constants using direct MongoDB interface from MDBUtils.
"""
import psana.pyalgos.generic.Utils as gu
print('test_delete_database 1:', database_names())
#txt = '%s\nThis is a string\n to test\ncalibration storage' % gu.str_tstamp()
#data, ctype = txt, 'testtext'; logger.debug('txt: %s' % str(data))
#data, ctype = get_test_nda(), 'testnda'; logger.debug(info_ndarr(data, 'nda'))
#data, ctype = get_test_dic(), 'testdict'; logger.debug('dict: %s' % str(data))
kwa = {'user' : gu.get_login()}
t0_sec = time()
ts = gu.str_tstamp(fmt='%Y-%m-%dT%H:%M:%S%z', time_sec=t0_sec)
mu.insert_constants('%s - saved at %s'%(data,ts), expname, detname, ctype, runnum+int(tname), int(t0_sec),\
time_stamp=ts, **kwa)
print('test_delete_database 2:', database_names())
def test_xtcav_calib_constants(fname=
'/reg/d/psdm/XPP/xpptut15/calib/Xtcav::CalibV1/XrayTransportDiagnostic.0:Opal1000.0/pedestals/101-102.data',\
add_constants_to_db=False) :
_, exp, _, cvers, detname, ctype, cfname = fname.rsplit('/',6)
resp = parse_calib_file_name(cfname)
begin, end, ext = resp if resp is not None else (None, None, None)
det = detname_conversion(detname)
run = begin
dbname_exp = dbu.db_prefixed_name(exp)
dbname_det = dbu.db_prefixed_name(det)
print('LCLS1 Xtcav calibration file: %s' % fname)
print('Parameters form path: exp:%s det:%s ctype:%s run:%s dbname_exp:%s dbname_det:%s'%\
(exp, det, ctype, run, dbname_exp, dbname_det))
#Save(ct,fname)
o1 = Load(fname)
d1 = dict_from_xtcav_calib_object(o1)
print('Xtcav calibration constants as dict:\n', d1)
if not add_constants_to_db : return
#==================================
#---- Delete databases for experiment and detector
client = dbu.connect_to_server(HOST, PORT)
print('Open client on host:%s port:%s' % (HOST, PORT))
print('Delete database %s'% dbname_exp)
dbu.delete_database(client, dbname_exp)
print('Delete database %s'% dbname_det)
dbu.delete_database(client, dbname_det)
#---- Add data to experiment and detector dbs
print('Add Xtcav constants')
kwargs = {'host' : HOST,\
'port' : PORT,\
'version' : 'V01',\
'comment' : 'test of add-retrieve xtcav constants'
}
#insert_calib_data(data, *kwargs)
dbu.insert_constants(o1, exp, det, ctype, run, time_sec='1000000000', **kwargs)
#msg = dbu.database_info(client, dbname_exp, level=10)
#print(msg)
print('Xtcav constants inserted, now retrieve them from db:%s collection:%s' % (dbname_exp, det))
db, fs = dbu.db_and_fs(client, dbname_exp)
col = dbu.collection(db, det)
#for doc in col.find() :
# print(doc)
doc = dbu.find_doc(col, query={'ctype':ctype, 'run':run})
print('Found doc:\n', doc)
o2 = dbu.get_data_for_doc(fs, doc)
d2 = dict_from_xtcav_calib_object(o2)
print('Xtcav calib object converted to dict:\n', d2)
#print('cmp(d1,d2) :', str(d1==d2))
print('\nCompare dictionaries for Xtcav calib objects loaded directly from calib file and passed through the CDB')
compare_dicts(d1,d2)
client.close()
return