Use `--build-arg` in `build.py`

[guedressel]

build.py is a solution to build different versions of this image. Today this can be easily achieved by using build-args. Refactoring build.py to use this feature will reduce code and complexity.

https://docs.docker.com/engine/reference/commandline/build/#set-build-time-variables---build-arg