o hu@sdddlZddlmZddlmZddlmZddlmZddl m Z dZ dd Z Gd d d eZ dS) N)settings) BaseCommand) transaction)get_search_backend)get_indexed_modelsicslit}|D]}||}|r'|j|||jg||j|qtfdd|DS)a; This takes a search backend and a list of models. By calling the get_index_for_model method on the search backend, it groups the models into the indices that they will be indexed into. It returns an ordered mapping of indices to lists of models within each index. For example, Elasticsearch 2 requires all page models to be together, but separate from other content types (eg, images and documents) to prevent field mapping collisions: >>> group_models_by_index(elasticsearch2_backend, [ ... wagtailcore.Page, ... myapp.HomePage, ... myapp.StandardPage, ... wagtailimages.Image ... ]) { : [wagtailcore.Page, myapp.HomePage, myapp.StandardPage], : [wagtailimages.Image], } csg|] \}}||fqSr).0 index_name index_modelsindicesro/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/search/management/commands/update_index.py 1s z)group_models_by_index..) collections OrderedDictget_index_for_model setdefaultnameappenditems)backendmodelsmodels_by_indexmodelindexrr r group_models_by_index s  rc@sTeZdZddZdefddZddZdd Zd d Zd d Z e j efddZ dS)CommandcOs$|jdkr|jj|i|dSdS)z6Helper function that respects verbosity when printing.rN) verbositystdoutwrite)selfargskwargsrrr r9s z Command.writeFc Cs6|d|t|}|js|d|dSt|t}|s(||d|D]n\}}||d|j||}|}|D]} || qCd} |s|D]5} |jd || j j | j  ddd ||| d |D]} || | | t| 7} qs|qQ|||d | |q*dS) NzUpdating backend: z'Backend '%s' doesn't require rebuildingz: No indices to rebuildz: Rebuilding index %srz {}: {}.{} #endingpkz: indexed %d objects)rrrebuilder_classrrrrstart add_modelformat_meta app_label__name__ljustprint_iter_progressqueryset_chunksget_indexed_objectsorder_by add_itemslen print_newlinefinish) r backend_name schema_only chunk_sizermodels_grouped_by_indexrr rebuilderr object_countchunkrrr update_backend>sP       zCommand.update_backendcCsB|jdddddd|jdddd d d|jd dd ttd ddS)Nz --backendstorer8zSpecify a backend to update)actiondestdefaulthelpz --schema-only store_truer9Fz(Prevents loading any data into the indexz --chunk_sizer:zHSet number of records to be fetched at once for inserting into the index)rArBrCtyperD) add_argumentDEFAULT_CHUNK_SIZEint)r parserrrr add_argumentsvs, zCommand.add_argumentscKsf|d|_|dr|dg}nttdrtj}ndg}|D]}|j||dd|ddqdS) Nrr8WAGTAILSEARCH_BACKENDSrCr9Fr:)r9r:)rhasattrrrLkeysr?get)r options backend_namesr8rrr handles     zCommand.handlecCs|ddS)Nr$)r)r rrr r6szCommand.print_newlineccsxt|ddD]2\}}|V|jddd|ddkr'||jdddn |d dkr4|jd dd|jqd S) aq Print a progress meter while iterating over an iterable. Use it as part of a ``for`` loop:: for item in self.print_iter_progress(big_long_list): self.do_expensive_computation(item) A ``.`` character is printed for every value in the iterable, a space every 10 items, and a new line every 50 items. )r).r$r%(rz#  N) enumeraterr6rflush)r iterableivaluerrr r0s    zCommand.print_iter_progressccs<d} t|||dd|}|sdS|V|d7}q)a  Yield a queryset in chunks of at most ``chunk_size``. The chunk yielded will be a list, not a queryset. Iterating over the chunks is done in a transaction so that the order and count of items in the queryset remains stable. rTNrS)list)r qsr:r[rrrr r1szCommand.queryset_chunksN) r. __module__ __qualname__rrHr?rKrRr6r0ratomicr1rrrr r8s 8r)r django.confrdjango.core.management.baser django.dbrwagtail.search.backendsrwagtail.search.indexrrHrrrrrr s     +