|
|
#!PYTHONEXE
|
|
|
|
|
|
"""registrationAdmin.py is an script for Madrigal administrators to register
|
|
|
or unregister users from receiving emails for experiments or instruments.
|
|
|
|
|
|
$Id: registrationAdmin.py 7046 2019-10-07 19:57:14Z brideout $
|
|
|
"""
|
|
|
|
|
|
usage = """registrationAdmin.py [--unregister | --register]
|
|
|
[--inst=<kinst> | --exp=<expDir>] userEmail
|
|
|
|
|
|
Either unregister or register must be set.
|
|
|
Either --inst=<instrument code> or --exp=<expDir> must be set.
|
|
|
expDir is in form <experiments/2010/mlh/18jan10>
|
|
|
"""
|
|
|
|
|
|
import os, os.path, sys
|
|
|
import getopt
|
|
|
|
|
|
import madrigal.ui.userData
|
|
|
|
|
|
|
|
|
# parse command line
|
|
|
unregister = None
|
|
|
register = None
|
|
|
inst = None
|
|
|
expDir = None
|
|
|
|
|
|
try:
|
|
|
opts, args = getopt.getopt(sys.argv[1:], "", ["unregister", "register",
|
|
|
"inst=", "exp="])
|
|
|
except getopt.GetoptError as err:
|
|
|
print(str(err)) # will print something like "option -a not recognized"
|
|
|
print(usage)
|
|
|
sys.exit(2)
|
|
|
for o, a in opts:
|
|
|
if o == "--unregister":
|
|
|
unregister = True
|
|
|
if register != None:
|
|
|
raise ValueError("cannot set both register and unregister")
|
|
|
elif o == "--register":
|
|
|
register = True
|
|
|
if unregister != None:
|
|
|
raise ValueError("cannot set both register and unregister")
|
|
|
elif o == "--inst":
|
|
|
inst = int(a)
|
|
|
if expDir != None:
|
|
|
raise ValueError("cannot set both inst and exp")
|
|
|
elif o == "--exp":
|
|
|
expDir = a
|
|
|
if inst != None:
|
|
|
raise ValueError("cannot set both inst and exp")
|
|
|
else:
|
|
|
assert False, "unhandled option"
|
|
|
if len(args) != 1:
|
|
|
print(usage)
|
|
|
print('userEmail argument required')
|
|
|
sys.exit(-1)
|
|
|
email = args[0]
|
|
|
if email.find('@') == -1:
|
|
|
print(usage)
|
|
|
print(('userEmail argument <%s> seems invalid' % (email)))
|
|
|
sys.exit(-1)
|
|
|
|
|
|
if register == None and unregister == None:
|
|
|
raise ValueError("must set either register or unregister")
|
|
|
|
|
|
if inst == None and expDir == None:
|
|
|
raise ValueError("must set either inst or exp")
|
|
|
|
|
|
madUserObj = madrigal.ui.userData.MadrigalUserData()
|
|
|
|
|
|
if register:
|
|
|
if inst:
|
|
|
madUserObj.registerInstrument(email, inst)
|
|
|
elif expDir:
|
|
|
madUserObj.registerExperiment(email, expDir)
|
|
|
elif unregister:
|
|
|
if inst:
|
|
|
madUserObj.unregisterInstrument(email, inst)
|
|
|
elif expDir:
|
|
|
madUserObj.unregisterExperiment(email, expDir)
|
|
|
|
