o -Fj/@sHddlmZddlmZddlmZddlmZmZGdddeZ dS)) defaultdict) BaseCommand) transaction) Craptitude GameSessionc@s0eZdZdZddZddZddZdd Zd S) Commanda$Remove duplicate craptitudes, keeping ONE row per (category, roundNumber). GameSession references pointing at a discarded duplicate are reassigned to the kept row first, so no game sessions are cascade-deleted. Run this before applying the unique (category, roundNumber) constraint migration.cCs|jdddddS)Nz --dry-run store_truez2Report what would change without writing anything.)actionhelp) add_argument)selfparserrH/var/www/html/premium_crap/api/management/commands/dedupe_craptitudes.py add_argumentss  zCommand.add_argumentsc s|d}tt}tjdD] }||j|jf|qdd| D}|s5|j |j ddSd}d}tt| D]z\\} } } || fdd| D} d d| D} || }tjj| d }|}|j d | d | d jdt| d |rd|dnd|rd|nd|s|rj|d|r|jdtjj| d|t| 7}||7}qD|rtdWdn1swY|rdnd}|j |j |d|dt|d|d|s|j ddSdS)Ndry_runpkcSs"i|] \}}t|dkr||qS))len).0keyrowsrrr s"z"Command.handle..u#✅ No duplicate craptitudes found.rcsg|] }|jjkr|qSrrrrkeeperrr +sz"Command.handle..cSsg|]}|jqSrrrrrrr,s)craptitude_id__inz category=z round=z : keep id=z , delete z duplicate(s)z , reassign z session(s)z, carried prefixes ) update_fields) craptitude)pk__inTz[dry-run] would deleteu ✅ Deleted z duplicate craptitude(s) across z slot(s); reassigned z game session(s).zNext: run `python manage.py makemigrations api` and `python manage.py migrate` to add the unique (category, roundNumber) constraint.)rlistrobjectsallorder_by category_id roundNumberappenditemsstdoutwritestyleSUCCESSratomicsorted _pick_keeper_backfill_prefixesrfiltercountrrsaveupdatedelete set_rollback)r argsoptionsrgroupsc dup_groups total_deletedtotal_reassignedr) round_numberrdupesdupe_idscarriedsessions session_countprefixrrrhandlesl         "zCommand.handlecs^g}dD](t|p drqtfdd|Dd}|r,t|t||q|S)z Fill any empty prefix on the keeper from a duplicate that has it. Mutates keeper in memory and returns the list of field names to save (empty if nothing changed). ) diaperPrefixdumpsterPrefixr c3s&|]}t|p dr|VqdS)r N)getattrstrip)rdfieldrr `s$z-Command._backfill_prefixes..N)rLrMnextsetattrr+)r rrCrEdonorrrOrr4Ws zCommand._backfill_prefixescCsdd}t||dS)z Choose which duplicate to keep. Preference order: 1. active (deletedAt is None) over soft-deleted, 2. has prefix text over empty, 3. lowest primary key. cSs8|jdu}t|jp dp|jpd}|||j fS)Nr ) deletedAtboolrJrMrKr)ractive has_prefixrrrscorems  z#Command._pick_keeper..score)r)max)r rrYrrrr3fs zCommand._pick_keeperN)__name__ __module__ __qualname__r rrIr4r3rrrrr s? rN) collectionsrdjango.core.management.baser django.dbr api.modelsrrrrrrrs