o 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|i|dSdS)zK Helper function that writes based on verbosity parameter rN) verbositystdoutwrite)selfargskwargsrt/var/www/html/ndineBlogger/venv/lib/python3.10/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_argumentss zCommand.add_argumentsc Ks|d|_|d}d}|dtattj}|j |j dWdn1s0wYt D]5}t |sAq9|t||||jd|D]}|D]}t|q[|t|7}qW|q9Wdn1sywY|d||dS)NrrrzRebuilding reference index)usingpkzIndexed %d objects)rgetr ratomicrrobjectsall _raw_deletedbr get_models is_indexedstrprint_iter_progressqueryset_chunksorder_bycreate_or_update_for_objectlen print_newline)r optionsr object_countall_referencesmodelchunkinstancerrrhandles0          zCommand.handlecCs|ddS)N)r )r rrrr,AszCommand.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. )start.r4)ending(rz#  N) enumerater r,r flush)r iterableivaluerrrr'Ds    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. rTNr5)list)r qsrr?itemsrrrr(\szCommand.queryset_chunksN) __name__ __module__ __qualname__r rr3r,r'rrrr(rrrrr s  "rN) django.appsrdjango.core.management.baser django.dbrwagtail.modelsrwagtail.signal_handlersrrrrrrrs