U {hm @sTddlmZddlmZddlmZddlmZddlm Z dZ GdddeZ d S) )apps) BaseCommand) transaction)ReferenceIndex)#disable_reference_index_auto_updateic@sFeZdZddZddZddZddZd d Zej e fd d Z d S)CommandcOs|jdkr|jj||dS)zK Helper function that writes based on verbosity parameter rN) verbositystdoutwrite)selfargskwargsr|/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/management/commands/rebuild_references_index.pyr s z Command.writecCs|jdddttdddS)Nz --chunk_sizestore chunk_sizezHSet number of records to be fetched at once for inserting into the index)actiondestdefaulttypehelp) add_argumentDEFAULT_CHUNK_SIZEint)r parserrrr add_argumentsszCommand.add_argumentsc Ks|d|_|d}d}|dtttj}|j |j dW5QRXt D]j}t |snq^|t||||jd|D]$}|D]}t|q|t|7}q|q^W5QRX|d||dS)NrrrzRebuilding reference index)usingpkzIndexed %d objects)rgetr ratomicrrobjectsallZ _raw_deletedbrZ get_modelsZ is_indexedstrprint_iter_progressqueryset_chunksZorder_byZcreate_or_update_for_objectlen print_newline)r optionsrZ object_countZall_referencesmodelchunkinstancerrrhandles*        zCommand.handlecCs|ddS)N)r )r rrrr'AszCommand.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. )start.r-)ending(rz#  N) enumerater r'r flush)r iterableivaluerrrr$Ds   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. rNr.)list)r qsrr8itemsrrrr%\s zCommand.queryset_chunksN) __name__ __module__ __qualname__r rr,r'r$rrrr%rrrrr s  "rN) Z django.appsrZdjango.core.management.baserZ django.dbrZwagtail.modelsrZwagtail.signal_handlersrrrrrrrs