def user(user_password):
user = User(username="******")
user.set_password(user_password)
user.save()
yield user
user.delete()
def get_user(self, request: HTTPServerRequest) -> Optional[User]:
"""Gets the User based on certificates supplied with in the request body
Args:
request: tornado HTTPServerRequest object
Returns:
User: The User object for the user specified by the certificates
None: If no User was found
"""
authenticated_user: Optional[User] = None
if request.headers and self.group_mapping:
username = request.headers.get(self.username_header)
groups = self._groups_from_headers(request.headers)
if username and groups:
try:
authenticated_user = User.objects.get(username=username)
except User.DoesNotExist:
if self.create_users:
authenticated_user = User(username=username)
# TODO: Really we should just have an option on User to disable
# password logins. For now, just set a random-ish value.
authenticated_user.set_password(str(uuid4()))
if authenticated_user:
authenticated_user.role_assignments = (
self._role_assignments_from_groups(groups))
authenticated_user.save()
return authenticated_user
def user_admin(user_admin_role):
role_assignment = RoleAssignment(role=user_admin_role,
domain={"scope": "Global"})
user = User(username="******")
user.set_password("password")
user.role_assignments = [role_assignment]
user.save()
yield user
user.delete()
def ensure_users():
"""Create the default admin user if necessary"""
if User.objects.count() == 0:
username = config.get("auth.default_admin.username")
password = config.get("auth.default_admin.password")
superuser_role = Role.objects.get(name="superuser")
logger.info("Creating default admin user with username: %s", username)
admin = User(username=username)
admin.set_password(password)
admin.role_assignments = [
RoleAssignment(role=superuser_role, domain={"scope": "Global"})
]
admin.save()