Swift: pep8 clean rewrite.py

Change-Id: I71c8af8db5e1d48a52e953f43e64041007e44ebb