mirror of
https://github.com/ansible/awx.git
synced 2026-02-24 22:46:01 -03:30
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
from awxkit.api.mixins import HasCreate, DSAdapter
|
|
from awxkit.utils import random_title, PseudoNamespace
|
|
from awxkit.api.resources import resources
|
|
from awxkit.config import config
|
|
|
|
from . import base
|
|
from . import page
|
|
|
|
|
|
class User(HasCreate, base.Base):
|
|
|
|
NATURAL_KEY = ('username',)
|
|
|
|
def payload(self, **kwargs):
|
|
payload = PseudoNamespace(
|
|
username=kwargs.get('username') or 'User-{}'.format(random_title(non_ascii=False)),
|
|
password=kwargs.get('password') or config.credentials.default.password,
|
|
is_superuser=kwargs.get('is_superuser', False),
|
|
is_system_auditor=kwargs.get('is_system_auditor', False),
|
|
first_name=kwargs.get('first_name', random_title()),
|
|
last_name=kwargs.get('last_name', random_title()),
|
|
email=kwargs.get('email', '{}@example.com'.format(random_title(5, non_ascii=False))),
|
|
)
|
|
return payload
|
|
|
|
def create_payload(self, username='', password='', **kwargs):
|
|
payload = self.payload(username=username, password=password, **kwargs)
|
|
payload.ds = DSAdapter(self.__class__.__name__, self._dependency_store)
|
|
return payload
|
|
|
|
def create(self, username='', password='', organization=None, **kwargs):
|
|
payload = self.create_payload(username=username, password=password, **kwargs)
|
|
self.password = payload.password
|
|
|
|
self.update_identity(Users(self.connection).post(payload))
|
|
|
|
if organization:
|
|
organization.add_user(self)
|
|
|
|
return self
|
|
|
|
|
|
page.register_page([resources.user, (resources.users, 'post')], User)
|
|
|
|
|
|
class Users(page.PageList, User):
|
|
|
|
pass
|
|
|
|
|
|
page.register_page(
|
|
[resources.users, resources.organization_admins, resources.related_users, resources.credential_owner_users, resources.user_admin_organizations], Users
|
|
)
|
|
|
|
|
|
class Me(Users):
|
|
|
|
pass
|
|
|
|
|
|
page.register_page(resources.me, Me)
|