xmpp.py: Extend try block in await_notifications

There were cases where the xmpp connection was not closed when
it probably should have been. Extend the try block so that the
connection is explicitly closed on errors in check_for_notifications
or select.