Add a `json_assignment` parameter

[StephenSorriaux]

When draining some brokers from its topics, it could be useful to have a json_assignment that lets the user specify how to reassign topic partitions in the cluster.