def switchSession(session=None, user=None):
"""Switch the current session, using the global connection object"""
global s
if user is None:
user = s._dir[1]
s = workflow.switchSession(cxn, user, session)
powers = range(-5,-85,-5)
powers = [power*dBm for power in powers]
hHPowers = [power*dBm for power in range(0,15,5)]
hPowers = powers[:5]
lPowers = powers[5:]
print 'What user are you? (fpga_test (FPGA), paramp (PA), PAold (old),' \
' or scope_calibration (SC), or (SCold)?'
name = raw_input()
if name == 'fpga' or name == 'FPGA':
print '*******************'
print 'You are now logged in to fpga'
print '*******************'
cxn = labrad.connect(host='more-mesa.sba.corp.google.com')
s = switchSession(cxn, user='******',sample = 'sampleFpgaTest')
elif name == 'paramp' or name == 'PA':
print '*******************'
print 'You are now logged in to PA'
print '*******************'
cxn = labrad.connect(host='more-mesa.sba.corp.google.com')
s = switchSession(cxn, user='******',sample = 'samplePA')
elif name == 'old' or name == 'PAold':
print '*******************'
print 'You are now logged in to PA old'
print '*******************'
cxn = labrad.connect(host='more-mesa.sba.corp.google.com')
s = switchSession(cxn, user='******',sample = 'samplePAold')
reslist = ["Res0" + str(i) for i in range(1, 10)] + ["Res" + str(i) for i in range(10, 17)]
powers = range(-5, -85, -5)
powers = [power * dBm for power in powers]
hHPowers = [power * dBm for power in range(0, 15, 5)]
hPowers = powers[:5]
lPowers = powers[5:]
print "What user are you? (Brooks, paramp (PA) or fpga)"
name = raw_input()
if name == "fpga" or name == "FPGA":
print "*******************"
print "You are now logged in to fpga"
print "*******************"
cxn = labrad.connect()
s = switchSession(cxn, user="******", sample="sampleFpgaTest")
elif name == "paramp" or name == "PA":
print "*******************"
print "You are now logged in to PA"
print "*******************"
cxn = labrad.connect()
s = switchSession(cxn, user="******", sample="samplePA")
else:
print "*******************"
print "You are now logged in to Brooks"
print "*******************"
cxn = labrad.connect()
s = switchSession(cxn, user="******", sample="sample")
