hdZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z Gd d eZGd d ejj!eZeZeZy) z Custom generic managers N)ImproperlyConfigured)models)QuerySet) get_language) appsettings)get_active_language_choicesc`eZdZdZfdZfdZfdZfdZfdZd dZ dZ d d Z xZ S) TranslatableQuerySetz An enhancement of the QuerySet which sets the objects language before they are returned. When using this class in combination with *django-polymorphic*, make sure this class is first in the chain of inherited classes. c2t||i|d|_yN)super__init__ _language)selfargskwargs __class__s ?/home/dcms/DCMS/lib/python3.12/site-packages/parler/managers.pyrzTranslatableQuerySet.__init__s $)&)cFt|}|j|_|Sr )r _cloner)rcrs rrzTranslatableQuerySet._clones GN nn rc X|jr|j|d<t|di|S)N_current_language)rr create)rrrs rrzTranslatableQuerySet.creates- >>*...F& 'w~'''rct||jb|jrUt |jdt j r-|jD]}|j|jyyyy)Nr)r _fetch_allr _result_cache isinstancerModelset_current_language)robjrs rrzTranslatableQuerySet._fetch_all&ss  NN &""4--a0&,,?)) 9((8 9@# 'rc i}|r>|jjjD]} |j|||<t ||fi|}|j||S#t$rYHwxYwr )model _parler_metaget_all_fieldspopKeyErrorr _extract_model_paramsupdate)rdefaultsrtranslated_defaultsfieldparamsrs rr*z*TranslatableQuerySet._extract_model_params3s! 00??A 19e1D'.  .xB6B )*  sA'' A32A3cT|tjj}||_|S)z[ Set the language code to assign to objects retrieved using this QuerySet. )rPARLER_LANGUAGESget_default_languager)r language_codes rlanguagezTranslatableQuerySet.languageBs*  '88MMOM& rc~|jjj}|s tf}i}|j D])\}}|j dr |||dd< |||d|<+t |dk(r|d||dz<|jd i|S|||dz<|jd i|jS) a~ Only return translated objects which of the given languages. When no language codes are given, only the currently active language is returned. .. note:: Due to Django `ORM limitations `_, this method can't be combined with other filters that access the translated fields. As such, query the fields in one filter: .. code-block:: python qs.translated('en', name="Cheese Omelette") This will query the translated model for the ``name`` field. master__N__r__language_code__language_code__inr) r%r& root_rel_nameritems startswithlenfilterdistinct)rlanguage_codestranslated_fieldsrelnamefilters field_namevals r translatedzTranslatableQuerySet.translatedLs"**))77*n.N0668 :OJ$$Z0*- 12'697)2j\23  : ~ ! #3A!3DGG// 04;;)) )7EGG33 44;;))224 4rc <t|}|j|i|S)aN Only return objects which are translated, or have a fallback that should be displayed. Typically that's the currently active language and fallback language. This should be combined with ``.distinct()``. When ``hide_untranslated = True``, only the currently active language will be returned. )rrH)rr3rCrBs ractive_translationsz(TranslatableQuerySet.active_translationsps%5]CtD2CDDrr ) __name__ __module__ __qualname____doc__rrrrr*r4rHrJ __classcell__rs@rr r s2 ( 9 "5H Err c"eZdZdZfdZxZS)TranslatableManagerz[ The manager class which ensures the enhanced TranslatableQuerySet object is used. ct|}t|ts"t |j j d|S)Nz;._queryset_class does not inherit from TranslatableQuerySet)r get_querysetr r rrrK)rqsrs rrTz TranslatableManager.get_querysetsE W ! #"23&>>**++fg  r)rKrLrMrNrTrOrPs@rrRrRsrrR)rNdjangodjango.core.exceptionsr django.dbrdjango.db.models.queryrdjango.utils.translationrparlerr parler.utilsrr Manager from_querysetrRTranslationQuerysetTranslationManagerrrrras\7+14nE8nEb &..667KL +(r