python之selectors

# -*- coding: utf-8 -*-
__author__ = 'fc'

import selectors
import socket
import time

def accept(sock:socket.socket, mask):
    conn, addr = sock.accept()
    print('client', addr, 'connected')
    conn.setblocking(False)
    sel.register(conn, selectors.EVENT_READ, oper)

def oper(sock:socket.socket, mask):
    if selectors.EVENT_READ & mask:
        print('ready to read')
        data = sock.recv(1024)
        if not data:
            print('client', sock.getpeername(), 'disconnnected')
            sel.unregister(sock)
            sock.close()
        else:
            print('received: ', data)
            sel.modify(sock, selectors.EVENT_WRITE, oper)
    elif selectors.EVENT_WRITE & mask:
        print('ready to write')
        msg = b'hello, friend'
        sock.send(msg)
        print('send: ', msg)
        sel.modify(sock, selectors.EVENT_READ, oper)


sel = selectors.DefaultSelector()
print(sel)
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0)
sock.bind(('', 60001))
sock.listen(5)
sock.setblocking(False)
sel.register(sock, selectors.EVENT_READ, accept)

while True:
    events = sel.select(10)
    if not events:
        print('timeout', time.strftime('%Y-%m-%d %H:%M:%S'))
    for key, mask in events:
        callback = key.data
        callback(key.fileobj, mask)