def list(color):
if not is_dbenv_loaded():
load_dbenv()
from aiida.orm.implementation import User
from aiida.common.utils import get_configured_user_email
from aiida.common.exceptions import ConfigurationError
try:
current_user = get_configured_user_email()
except ConfigurationError:
current_user = None
if current_user is not None:
pass
else:
click.echo(
"### No default user configured yet, run 'verdi install'! ###",
err=True)
for user in User.get_all_users():
name_pieces = []
if user.first_name:
name_pieces.append(user.first_name)
if user.last_name:
name_pieces.append(user.last_name)
full_name = " ".join(name_pieces)
if full_name:
full_name = " {}".format(full_name)
institution_str = " ({})".format(
user.institution) if user.institution else ""
color_id = 39 # Default foreground color
permissions_list = []
if user.is_staff:
permissions_list.append("STAFF")
if user.is_superuser:
permissions_list.append("SUPERUSER")
if not user.has_usable_password():
permissions_list.append("NO_PWD")
color_id = 90 # Dark gray
else:
color_id = 34 # Blue
permissions_str = ",".join(permissions_list)
if permissions_str:
permissions_str = " [{}]".format(permissions_str)
if user.email == current_user:
symbol = ">"
color_id = 31
else:
symbol = "*"
if color:
start_color = "\x1b[{}m".format(color_id)
end_color = "\x1b[0m"
bold_sequence = "\x1b[1;{}m".format(color_id)
nobold_sequence = "\x1b[0;{}m".format(color_id)
else:
start_color = ""
end_color = ""
bold_sequence = ""
nobold_sequence = ""
click.echo("{}{} {}{}{}:{}{}{}{}".format(start_color, symbol,
bold_sequence, user.email,
nobold_sequence, full_name,
institution_str,
permissions_str, end_color))
def do_configure(email,
first_name,
last_name,
institution,
no_password,
force_reconfigure=False):
if not is_dbenv_loaded():
load_dbenv()
from aiida.orm import User as UserModel
import readline
import getpass
configure_user = False
user, created = get_or_new_user(email=email)
if not created:
click.echo("\nAn AiiDA user for email '{}' is already present "
"in the DB:".format(email))
if force_reconfigure or click.confirm(
"Do you want to reconfigure it?"):
configure_user = True
else:
configure_user = True
click.echo("Configuring a new user with email '{}'".format(email))
if configure_user:
try:
kwargs = {}
for field in UserModel.REQUIRED_FIELDS:
verbose_name = field.capitalize()
readline.set_startup_hook(
lambda: readline.insert_text(getattr(user, field)))
if field == 'first_name' and first_name:
kwargs[field] = first_name
elif field == 'last_name' and last_name:
kwargs[field] = last_name
elif field == 'institution' and institution:
kwargs[field] = institution
else:
kwargs[field] = raw_input('{}: '.format(verbose_name))
finally:
readline.set_startup_hook(lambda: readline.insert_text(""))
for k, v in kwargs.iteritems():
setattr(user, k, v)
change_password = False
if no_password:
user.password = None
else:
if user.has_usable_password():
reply = raw_input(
"Do you want to replace the user password? [y/N] ")
reply = reply.strip()
if not reply:
pass
elif reply.lower() == 'n':
pass
elif reply.lower() == 'y':
change_password = True
else:
click.echo("Invalid answer, assuming answer was 'NO'")
else:
reply = raw_input(
"The user has no password, do you want to set one? [y/N] ")
reply = reply.strip()
if not reply:
pass
elif reply.lower() == 'n':
pass
elif reply.lower() == 'y':
change_password = True
else:
click.echo("Invalid answer, assuming answer was 'NO'")
if change_password:
match = False
while not match:
new_password = getpass.getpass("Insert the new password: "******"Insert the new password (again): ")
if new_password == new_password_check:
match = True
else:
click.echo("ERROR, the two passwords do not match.")
## Set the password here
user.password = new_password
else:
user.password = None
user.force_save()
click.echo(">> User {} {} saved. <<".format(user.first_name,
user.last_name))
if not user.has_usable_password():
click.echo("** NOTE: no password set for this user, ")
click.echo(" so he/she will not be able to login")
click.echo(" via the REST API and the Web Interface.")
def do_configure(email, first_name, last_name, institution, no_password):
from aiida.backends.utils import is_dbenv_loaded
if not is_dbenv_loaded():
load_dbenv()
import readline
import getpass
from aiida.orm.implementation import User
# if len(args) != 1:
# print >> sys.stderr, ("You have to pass (only) one parameter after "
# "'user configure', the email of")
# print >> sys.stderr, "the user to be configured."
# sys.exit(1)
# email = args[0]
user_list = User.search_for_users(email=email)
if user_list is not None and len(user_list) >= 1:
user = user_list[0]
print ""
print ("An AiiDA user for email '{}' is already present "
"in the DB:".format(email))
# print "First name: {}".format(user.first_name)
# print "Last name: {}".format(user.last_name)
# print "Institution: {}".format(user.institution)
configure_user = False
reply = click.confirm("Do you want to reconfigure it?")
# if not reply:
# pass
# elif reply.lower() == 'n':
# pass
if reply:
configure_user = True
# else:
# print "Invalid answer, assuming answer was 'NO'"
else:
configure_user = True
user = User(email=email)
print "Configuring a new user with email '{}'".format(email)
if configure_user:
try:
kwargs = {}
# for field in models.DbUser.REQUIRED_FIELDS:
for field in User.REQUIRED_FIELDS:
verbose_name = field.capitalize()
readline.set_startup_hook(lambda: readline.insert_text(
getattr(user, field)))
if field == 'first_name' and first_name:
kwargs[field] = first_name
elif field == 'last_name' and last_name:
kwargs[field] = last_name
elif field == 'institution' and institution:
kwargs[field] = institution
else:
kwargs[field] = raw_input('{}: '.format(verbose_name))
finally:
readline.set_startup_hook(lambda: readline.insert_text(""))
for k, v in kwargs.iteritems():
setattr(user, k, v)
change_password = False
if no_password:
user.password = None
else:
if user.has_usable_password():
reply = raw_input("Do you want to replace the user password? [y/N] ")
reply = reply.strip()
if not reply:
pass
elif reply.lower() == 'n':
pass
elif reply.lower() == 'y':
change_password = True
else:
print "Invalid answer, assuming answer was 'NO'"
else:
reply = raw_input("The user has no password, do you want to set one? [y/N] ")
reply = reply.strip()
if not reply:
pass
elif reply.lower() == 'n':
pass
elif reply.lower() == 'y':
change_password = True
else:
print "Invalid answer, assuming answer was 'NO'"
if change_password:
match = False
while not match:
new_password = getpass.getpass("Insert the new password: "******"Insert the new password (again): ")
if new_password == new_password_check:
match = True
else:
print "ERROR, the two passwords do not match."
## Set the password here
user.password = new_password
else:
user.password = None
user.force_save()
print ">> User {} {} saved. <<".format(user.first_name,
user.last_name)
if not user.has_usable_password():
print "** NOTE: no password set for this user, "
print " so he/she will not be able to login"
print " via the REST API and the Web Interface."