Pygame set focus.

Pygame set focus .