mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
45c99e4477
* 9856 base strawberry integration * 9856 user and group * 9856 user and circuits base * 9856 extras and mixins * 9856 fk * 9856 update strawberry version * 9856 update imports * 9856 compatability fixes * 9856 compatability fixes * 9856 update strawberry types * 9856 update strawberry types * 9856 core schema * 9856 dcim schema * 9856 extras schema * 9856 ipam and tenant schema * 9856 virtualization, vpn, wireless schema * 9856 fix old decorator * 9856 cleanup * 9856 cleanup * 9856 fixes to circuits type specifiers * 9856 fixes to circuits type specifiers * 9856 update types * 9856 GFK working * 9856 GFK working * 9856 _name * 9856 misc fixes * 9856 type updates * 9856 _name to types * 9856 update types * 9856 update types * 9856 update types * 9856 update types * 9856 update types * 9856 update types * 9856 update types * 9856 update types * 9856 update types * 9856 GraphQLView * 9856 GraphQLView * 9856 fix OrganizationalObjectType * 9856 single item query for schema * 9856 circuits graphql tests working * 9856 test fixes * 9856 test fixes * 9856 test fixes * 9856 test fix vpn * 9856 test fixes * 9856 test fixes * 9856 test fixes * 9856 circuits test sans DjangoModelType * 9856 core test sans DjangoModelType * 9856 temp checkin * 9856 fix extas FK * 9856 fix tenancy FK * 9856 fix virtualization FK * 9856 fix vpn FK * 9856 fix wireless FK * 9856 fix ipam FK * 9856 fix partial dcim FK * 9856 fix dcim FK * 9856 fix virtualization FK * 9856 fix tests / remove debug code * 9856 fix test imagefield * 9856 cleanup graphene * 9856 fix plugin schema * 9856 fix requirements * 9856 fix requirements * 9856 fix docs * 9856 fix docs * 9856 temp fix tests * 9856 first filterset * 9856 first filterset * 9856 fix tests * 9856 fix tests * 9856 working auto filter generation * 9856 filter types * 9856 filter types * 9856 filter types * 9856 fix graphiql test * 9856 fix counter fields and merge feature * 9856 temp fix tests * 9856 fix tests * 9856 fix tenancy, ipam filter definitions * 9856 cleanup * 9856 cleanup * 9856 cleanup * 9856 review changes * 9856 review changes * 9856 review changes * 9856 fix base-requirements * 9856 add wrapper to graphiql * 9856 remove old graphiql debug toolbar * 9856 review changes * 9856 update strawberry * 9856 remove superfluous check --------- Co-authored-by: Jeremy Stretch <jstretch@netboxlabs.com>
83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
from typing import TYPE_CHECKING, Annotated, List
|
|
|
|
import strawberry
|
|
import strawberry_django
|
|
from django.contrib.contenttypes.models import ContentType
|
|
|
|
from extras.models import ObjectChange
|
|
|
|
__all__ = (
|
|
'ChangelogMixin',
|
|
'ConfigContextMixin',
|
|
'ContactsMixin',
|
|
'CustomFieldsMixin',
|
|
'ImageAttachmentsMixin',
|
|
'JournalEntriesMixin',
|
|
'TagsMixin',
|
|
)
|
|
|
|
if TYPE_CHECKING:
|
|
from .types import ImageAttachmentType, JournalEntryType, ObjectChangeType, TagType
|
|
from tenancy.graphql.types import ContactAssignmentType
|
|
|
|
|
|
@strawberry.type
|
|
class ChangelogMixin:
|
|
|
|
@strawberry_django.field
|
|
def changelog(self, info) -> List[Annotated["ObjectChangeType", strawberry.lazy('.types')]]:
|
|
content_type = ContentType.objects.get_for_model(self)
|
|
object_changes = ObjectChange.objects.filter(
|
|
changed_object_type=content_type,
|
|
changed_object_id=self.pk
|
|
)
|
|
return object_changes.restrict(info.context.request.user, 'view')
|
|
|
|
|
|
@strawberry.type
|
|
class ConfigContextMixin:
|
|
|
|
@strawberry_django.field
|
|
def config_context(self) -> strawberry.scalars.JSON:
|
|
return self.get_config_context()
|
|
|
|
|
|
@strawberry.type
|
|
class CustomFieldsMixin:
|
|
|
|
@strawberry_django.field
|
|
def custom_fields(self) -> strawberry.scalars.JSON:
|
|
return self.custom_field_data
|
|
|
|
|
|
@strawberry.type
|
|
class ImageAttachmentsMixin:
|
|
|
|
@strawberry_django.field
|
|
def image_attachments(self, info) -> List[Annotated["ImageAttachmentType", strawberry.lazy('.types')]]:
|
|
return self.images.restrict(info.context.request.user, 'view')
|
|
|
|
|
|
@strawberry.type
|
|
class JournalEntriesMixin:
|
|
|
|
@strawberry_django.field
|
|
def journal_entries(self, info) -> List[Annotated["JournalEntryType", strawberry.lazy('.types')]]:
|
|
return self.journal_entries.all()
|
|
|
|
|
|
@strawberry.type
|
|
class TagsMixin:
|
|
|
|
@strawberry_django.field
|
|
def tags(self) -> List[Annotated["TagType", strawberry.lazy('.types')]]:
|
|
return self.tags.all()
|
|
|
|
|
|
@strawberry.type
|
|
class ContactsMixin:
|
|
|
|
@strawberry_django.field
|
|
def contacts(self) -> List[Annotated["ContactAssignmentType", strawberry.lazy('tenancy.graphql.types')]]:
|
|
return list(self.contacts.all())
|