mirror of
https://github.com/ansible/awx.git
synced 2026-03-18 01:17:35 -02:30
2c61e8f6de
In python3, we can use the built-in suppress from contextlib https://docs.python.org/3/library/contextlib.html#contextlib.suppress
76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
from contextlib import suppress
|
|
|
|
from awxkit.api.mixins import HasCreate, HasInstanceGroups, HasNotifications, DSAdapter
|
|
from awxkit.utils import random_title, set_payload_foreign_key_args, 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)
|
|
)
|
|
|
|
payload = set_payload_foreign_key_args(payload, ('default_environment',), kwargs)
|
|
|
|
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)
|