U d @sdZddlZddlZddlZddlmZddlmZmZm Z ddl m Z e ej e ddZdd Zdd d Zd ddZe ejfddZejededejdeddddejdeddddejdeddddddZdS)!zGThe ``celery upgrade`` command, used to upgrade from previous versions.N)defaults) CeleryCommand CeleryOptionhandle_preload_options)pass1cCsdS)z!Perform upgrade between versions.N)ctxrr6/tmp/pip-unpacked-wheel-mu1yl971/celery/bin/upgrade.pyupgrade sr c Cs4t|dd}dd|DW5QRSQRXdS)Nrutf-8cSsg|]}|qSrr.0linerrr sz_slurp..)codecsopen)filenameread_fhrrr _slurpsrCELERYcCs$|}||s d||g}|S)N_)upper startswithjoin)key namespacerrr _compat_keys r.origc Cs~g}d||g}td|dtjdt|dd>}t|dd$}|D]}||||qLW5QRXW5QRX|S)Nzwriting backup to ...filer r w)rprintsysstderrrrwriteappend)rsuffixlinesZbackup_filenamerZ backup_fhrrrr _backup!s   r+cCsRtt|dddD]4}|||||}||krd|krd|fSqd|fS)NcSst|S)N)len)xrrr 1z_to_new_key..)rZ CELERY_CELERYr)reversedsortedreplace)r keyfiltersourceZold_keynew_linerrr _to_new_key.s r7)clsrz--djangoTzUpgrading OptionszUpgrade Django project.)r8Zis_flagZ help_grouphelpz--compatz!Maintain backwards compatibility.z --no-backupzDon't backup original files.c st|}|s|rtnttd|dtjdfdd|D}tdd|Dr|s^t|t |dd }|D]\}}| |qrW5QRXtd tj dntd tj dd S) z/Migrate settings from Celery 3.x to Celery 4.x.z processing r r!csg|]}t|qSr)r7r r4rr rOszsettings..css|]}|dVqdS)rNr)rnrrr Rszsettings..r#r z'Changes to your setting have been made!z(Does not seem to require any changes :-)N) rrrr$r%r&anyr+rrr'stdout) rZdjangocompatZ no_backupr* new_linesZwrite_fhrrrr:r settings8s$  rA)r)r)__doc__rr%ZclickZ celery.apprZcelery.bin.baserrrZcelery.utils.functionalrgroupZ pass_contextr rrr+Z _TO_NEW_KEYr7commandargumentoptionrArrrr sF