Update the partition management to allow it to be only run against a
subset of the available databases. This is needed for rake tasks where
we might only interact with `main` or `ci` at one time.