You can easily define your own context manager to do what you want:

    @contextmanager
    def finally_focus(widget):
        try:
            yield
        finally:
            widget.setFocus()

    with finally_focus(my_widget):
        # do what I need to do


--