Выше по коду с помощью fcntl дескриптору сокета выставляется неблокирующий режим. Так делается в стиле C. То есть здесь другой способ сделать, то о чем ты говоришь. Но более правильнее использовать питонячий стиль я думаю. Я просто не стал уже переделывать.  Еще замечание по поводу recv - аргумент должен соответствовать степени двойки для лучшей производительности.