U |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).0Z index_nameZ index_modelsindicesrw/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/search/management/commands/update_index.py 1sz)group_models_by_index..) collections OrderedDictZget_index_for_model setdefaultnameappenditems)backendmodelsZmodels_by_indexmodelindexrr r group_models_by_index s  rc@sTeZdZddZdefddZddZdd Zd d Zd d Z e j efddZ dS)CommandcOs|jdkr|jj||dS)z6Helper function that respects verbosity when printing.rN) verbositystdoutwrite)selfargskwargsrrr r9s z Command.writeFc Cs8|d|t|}|js.|d|dSt|t}|sP||d|D]\}}||d|j||}|}|D]} || qd} |s|D]j} |jd || j j | j  ddd ||| d |D]} || | | t| 7} q|q|||d | |qTdS) NzUpdating backend: z'Backend '%s' doesn't require rebuildingz: No indices to rebuildz: Rebuilding index %srz {}: {}.{} #endingpkz: indexed %d objects)rrZrebuilder_classrrrrstartZ add_modelformat_metaZ app_label__name__ljustprint_iter_progressqueryset_chunksZget_indexed_objectsZorder_byZ add_itemslen print_newlinefinish) r backend_name schema_only chunk_sizerZmodels_grouped_by_indexrrZ rebuilderrZ object_countchunkrrr update_backend>sT      zCommand.update_backendcCsB|jdddddd|jdddd d d|jd dd ttd ddS)Nz --backendstorer.zSpecify a backend to update)actiondestdefaulthelpz --schema-only store_truer/Fz(Prevents loading any data into the indexz --chunk_sizer0zHSet number of records to be fetched at once for inserting into the index)r4r5r6typer7) add_argumentDEFAULT_CHUNK_SIZEint)rparserrrr add_argumentsvs,zCommand.add_argumentscKsf|d|_|dr|dg}nttdr4tj}ndg}|D]"}|j||dd|ddq>dS) Nrr.WAGTAILSEARCH_BACKENDSr6r/Fr0)r/r0)rhasattrrr?keysr2get)roptionsZ backend_namesr.rrr handles     zCommand.handlecCs|ddS)Nr )r)rrrr r,szCommand.print_newlineccsvt|ddD]d\}}|V|jddd|ddkrL||jdddn|d dkrf|jd dd|jq d 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) enumeraterr,rflush)riterableivaluerrr r)s   zCommand.print_iter_progressccs:d}t|||dd|}|s&q6|V|d7}qdS)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. rNrE)list)rqsr0rMrrrr r*s zCommand.queryset_chunksN) r' __module__ __qualname__rr;r2r>rDr,r)rZatomicr*rrrr r8s 8r)r Z django.confrZdjango.core.management.baserZ django.dbrZwagtail.search.backendsrZwagtail.search.indexrr;rrrrrr s     +