Manage sockets with a context manager

Since PyMongo requires Python 2.6+, we can use Python with-statements context managers to avoid socket leaks in the face of exceptions.