Refactor CMD argument parsing
It would be nice to use the built-in Python library argparse to parse arguments from the command line. The current approach looks a little messy, not declarative, and not flexible enough.
Let's say, we want to add an extra parameter to parse (e.g., data-partition-id), and this is difficult to do since there are a lot places where we have to add the changes.
But if we have a single file/entry-point where we describe expected commands with argparse the action of adding a new parameter will be much more pleasant.