148 lines
3.6 KiB
Python
148 lines
3.6 KiB
Python
from auditlog.registry import auditlog
|
|
from django.conf import settings
|
|
from loguru import logger
|
|
|
|
from .book.models import Edition, EditionInSchema, EditionSchema, Series, Work
|
|
from .collection.models import Collection as CatalogCollection
|
|
from .common.models import (
|
|
AvailableItemCategory,
|
|
ExternalResource,
|
|
IdType,
|
|
Item,
|
|
ItemCategory,
|
|
ItemInSchema,
|
|
ItemSchema,
|
|
ItemType,
|
|
SiteName,
|
|
item_categories,
|
|
item_content_types,
|
|
)
|
|
from .game.models import Game, GameInSchema, GameSchema
|
|
from .movie.models import Movie, MovieInSchema, MovieSchema
|
|
from .music.models import Album, AlbumInSchema, AlbumSchema
|
|
from .performance.models import (
|
|
Performance,
|
|
PerformanceProduction,
|
|
PerformanceProductionSchema,
|
|
PerformanceSchema,
|
|
)
|
|
from .podcast.models import Podcast, PodcastEpisode, PodcastInSchema, PodcastSchema
|
|
from .tv.models import (
|
|
TVEpisode,
|
|
TVEpisodeSchema,
|
|
TVSeason,
|
|
TVSeasonInSchema,
|
|
TVSeasonSchema,
|
|
TVShow,
|
|
TVShowInSchema,
|
|
TVShowSchema,
|
|
)
|
|
|
|
from .search.models import Indexer, ExternalSearchResultItem # isort:skip
|
|
|
|
|
|
# class Exhibition(Item):
|
|
|
|
# class Meta:
|
|
# proxy = True
|
|
|
|
|
|
# class Fanfic(Item):
|
|
|
|
# class Meta:
|
|
# proxy = True
|
|
|
|
|
|
def init_catalog_search_models():
|
|
if settings.DISABLE_MODEL_SIGNAL:
|
|
logger.warning(
|
|
"Catalog models are not being indexed with DISABLE_MODEL_SIGNAL configuration"
|
|
)
|
|
return
|
|
# skip indexing if the item type should never show up in search
|
|
Indexer.update_model_indexable(Edition)
|
|
# Indexer.update_model_indexable(Work)
|
|
Indexer.update_model_indexable(Movie)
|
|
Indexer.update_model_indexable(TVShow)
|
|
Indexer.update_model_indexable(TVSeason)
|
|
Indexer.update_model_indexable(Album)
|
|
Indexer.update_model_indexable(Game)
|
|
Indexer.update_model_indexable(Podcast)
|
|
Indexer.update_model_indexable(Performance)
|
|
# Indexer.update_model_indexable(PerformanceProduction)
|
|
# Indexer.update_model_indexable(CatalogCollection)
|
|
|
|
|
|
def init_catalog_audit_log():
|
|
for cls in Item.__subclasses__():
|
|
auditlog.register(
|
|
cls,
|
|
exclude_fields=[
|
|
"id",
|
|
"item_ptr",
|
|
"polymorphic_ctype",
|
|
"metadata",
|
|
"created_time",
|
|
"edited_time",
|
|
# related fields are not supported in django-auditlog yet
|
|
"lookup_ids",
|
|
"external_resources",
|
|
"merged_from_items",
|
|
"focused_comments",
|
|
],
|
|
)
|
|
|
|
auditlog.register(
|
|
ExternalResource, include_fields=["item", "id_type", "id_value", "url"]
|
|
)
|
|
|
|
# logger.debug(f"Catalog audit log initialized for {item_content_types().values()}")
|
|
|
|
|
|
__all__ = [
|
|
"CatalogCollection",
|
|
"AvailableItemCategory",
|
|
"ExternalResource",
|
|
"IdType",
|
|
"Item",
|
|
"ItemCategory",
|
|
"ItemInSchema",
|
|
"ItemSchema",
|
|
"ItemType",
|
|
"SiteName",
|
|
"item_categories",
|
|
"item_content_types",
|
|
"Edition",
|
|
"EditionInSchema",
|
|
"EditionSchema",
|
|
"Series",
|
|
"Work",
|
|
"Game",
|
|
"GameInSchema",
|
|
"GameSchema",
|
|
"Movie",
|
|
"MovieInSchema",
|
|
"MovieSchema",
|
|
"Album",
|
|
"AlbumInSchema",
|
|
"AlbumSchema",
|
|
"Performance",
|
|
"PerformanceProduction",
|
|
"PerformanceProductionSchema",
|
|
"PerformanceSchema",
|
|
"Podcast",
|
|
"PodcastEpisode",
|
|
"PodcastInSchema",
|
|
"PodcastSchema",
|
|
"TVEpisode",
|
|
"TVEpisodeSchema",
|
|
"TVSeason",
|
|
"TVSeasonInSchema",
|
|
"TVSeasonSchema",
|
|
"TVShow",
|
|
"TVShowInSchema",
|
|
"TVShowSchema",
|
|
"Indexer",
|
|
"init_catalog_search_models",
|
|
"init_catalog_audit_log",
|
|
]
|