75028d0427
# What this PR does See #2173 Also, closes #2187 . All of the new files under `type_stubs/icalendar` were autogenerated by running: ```bash stubgen -p icalendar -o type_stubs ``` ## Checklist - [ ] Unit, integration, and e2e (if applicable) tests updated - [ ] Documentation added (or `pr:no public docs` PR label added if not required) - [ ] `CHANGELOG.md` updated (or `pr:no changelog` PR label added if not required)
63 lines
1.5 KiB
Python
63 lines
1.5 KiB
Python
import typing
|
|
|
|
from django.contrib import admin
|
|
from django.core.exceptions import FieldDoesNotExist
|
|
from django.db.models import ForeignKey, Model
|
|
|
|
|
|
class RawForeignKeysMixin:
|
|
model: Model
|
|
|
|
@property
|
|
def raw_id_fields(self) -> typing.Tuple[str, ...]:
|
|
fields = self.model._meta.fields
|
|
fk_field_names = tuple(str(field.name) for field in fields if isinstance(field, ForeignKey))
|
|
|
|
return fk_field_names
|
|
|
|
|
|
class SearchableByIdsMixin:
|
|
model: Model
|
|
|
|
@property
|
|
def search_fields(self) -> typing.Tuple[str, ...]:
|
|
search_fields = (
|
|
"id",
|
|
"public_primary_key",
|
|
)
|
|
|
|
existing_fields: typing.List[str] = []
|
|
|
|
for field in search_fields:
|
|
try:
|
|
self.model._meta.get_field(field)
|
|
except FieldDoesNotExist:
|
|
continue
|
|
|
|
existing_fields.append(field)
|
|
|
|
return tuple(existing_fields)
|
|
|
|
|
|
class SelectRelatedMixin:
|
|
model: Model
|
|
list_display: typing.Tuple[str, ...]
|
|
|
|
@property
|
|
def list_select_related(self) -> typing.Tuple[str, ...]:
|
|
fk_field_names = []
|
|
|
|
for field_name in self.list_display:
|
|
try:
|
|
field = self.model._meta.get_field(field_name)
|
|
except FieldDoesNotExist:
|
|
continue
|
|
|
|
if isinstance(field, ForeignKey):
|
|
fk_field_names.append(str(field.name))
|
|
|
|
return tuple(fk_field_names)
|
|
|
|
|
|
class CustomModelAdmin(SearchableByIdsMixin, RawForeignKeysMixin, SelectRelatedMixin, admin.ModelAdmin):
|
|
pass
|