U Z+d[@sdZddlZddlmZddlmZmZmZddlm Z ej ede de dej d d ee d d d ej ddeed dd ej ddedd ddej dded ddej dded ddej ddedd ddejed d!ZdS)"zAThe ``celery migrate`` command, used to filter and move messages.N) Connection) CeleryCommand CeleryOptionhandle_preload_options) migrate_tasks)clssource destinationz-nz--limitzMigration OptionszNumber of tasks to consume.)rtype help_grouphelpz-tz --timeoutz%Timeout in seconds waiting for tasks.z-az--ack-messagesTz Ack messages from source broker.)rZis_flagr r z-Tz--tasksz List of task names to filter on.)rr r z-Qz--queueszList of queues to migrate.z-Fz --foreverz'Continually migrate tasks until killed.c s.fdd}tt|t|fd|i|dS)zMigrate tasks from one broker to another. Warning: This command is experimental, make sure you have a backup of the tasks before you continue. cs&jd|jd|jd|dS)NzMigrating task /z: )objZechocountZstrtotal)statebodymessagectx6/tmp/pip-unpacked-wheel-ucduq0nd/celery/bin/migrate.pyon_migrate_task:sz migrate..on_migrate_taskcallbackN)rr)rrr kwargsrrrrmigrate s0 r)__doc__ZclickZkomburZcelery.bin.baserrrZcelery.contrib.migratercommandargumentoptionintfloatZ pass_contextrrrrrsd