mirror of
https://github.com/ansible/awx.git
synced 2026-04-27 04:35:22 -02:30
ad85b176f4
Add the new credential type to the Credential page object and helper methods to manage the galaxy credentials for the Organization page object.
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
from awxkit.api.mixins import HasCreate, HasInstanceGroups, HasNotifications, DSAdapter
|
|
from awxkit.utils import random_title, suppress, PseudoNamespace
|
|
from awxkit.api.resources import resources
|
|
import awxkit.exceptions as exc
|
|
from . import base
|
|
from . import page
|
|
|
|
|
|
class Organization(HasCreate, HasInstanceGroups, HasNotifications, base.Base):
|
|
|
|
NATURAL_KEY = ('name',)
|
|
|
|
def add_admin(self, user):
|
|
if isinstance(user, page.Page):
|
|
user = user.json
|
|
with suppress(exc.NoContent):
|
|
self.related.admins.post(user)
|
|
|
|
def add_user(self, user):
|
|
if isinstance(user, page.Page):
|
|
user = user.json
|
|
with suppress(exc.NoContent):
|
|
self.related.users.post(user)
|
|
|
|
def add_galaxy_credential(self, credential):
|
|
if isinstance(credential, page.Page):
|
|
credential = credential.json
|
|
with suppress(exc.NoContent):
|
|
self.related.galaxy_credentials.post({
|
|
"id": credential.id,
|
|
})
|
|
|
|
def remove_galaxy_credential(self, credential):
|
|
if isinstance(credential, page.Page):
|
|
credential = credential.json
|
|
with suppress(exc.NoContent):
|
|
self.related.galaxy_credentials.post({
|
|
"id": credential.id,
|
|
"disassociate": True,
|
|
})
|
|
|
|
|
|
def payload(self, **kwargs):
|
|
payload = PseudoNamespace(name=kwargs.get('name') or 'Organization - {}'.format(random_title()),
|
|
description=kwargs.get('description') or random_title(10))
|
|
return payload
|
|
|
|
def create_payload(self, name='', description='', **kwargs):
|
|
payload = self.payload(name=name, description=description, **kwargs)
|
|
payload.ds = DSAdapter(self.__class__.__name__, self._dependency_store)
|
|
return payload
|
|
|
|
def create(self, name='', description='', **kwargs):
|
|
payload = self.create_payload(name=name, description=description, **kwargs)
|
|
return self.update_identity(Organizations(self.connection).post(payload))
|
|
|
|
|
|
page.register_page([resources.organization,
|
|
(resources.organizations, 'post')], Organization)
|
|
|
|
|
|
class Organizations(page.PageList, Organization):
|
|
|
|
pass
|
|
|
|
|
|
page.register_page([resources.organizations,
|
|
resources.user_organizations,
|
|
resources.project_organizations], Organizations)
|