selectors 模块完成文件上传下载功能

############    server端

import selectors,socket,time,pickle,os,struct
ip_port = ('192.168.43.182',8080)
buffer_size = 1024
back_log = 5
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

class SelectorsServer:
    def __init__(self):
        self.conn_dict = {}
        self.has_put = 0
        self.has_get = 0
        self.sel = selectors.DefaultSelector()      ##  创建selectors 对象
        self.creat_socket()
        self.handle()       ##  监听

    def creat_socket(self):     ##  创建socket 对象
        sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
        sock.bind(ip_port)
        sock.listen(back_log)
        sock.setblocking(False)
        self.sel.register(sock,selectors.EVENT_READ,self.accept)

    def handle(self):      ##  监听事件
         while 1:           ##  循环监听
            events = self.sel.select()
            for key,mask in events:
                callback = key.data
                callback(key.fileobj,mask)


    def accept(self,sock,mask):     ###     连接
        conn,addr = sock.accept()
        # print(conn)
        print('连接成功。。。')
        conn.setblocking(False)
        self.sel.register(conn,selectors.EVENT_READ,self.read)     ##  给conn注册
        self.conn_dict[conn] = {}   ##  标识conn连接完成

    def read(self,conn,mask):     ##  通信循环
        if not self.conn_dict[conn]:    ##  第一次进来
            res = conn.recv(buffer_size)
            res = pickle.loads(res)
            action = res['action']
            filename = res['filename']
            filename = os.path.join(BASE_DIR,'files',filename)
            print('ok')
            conn.sendall(b'ready')        ##  取到动作和文件名
            self.conn_dict[conn] = {'action':action,'filename':filename}
            if hasattr(self, action):
                func = getattr(self,action)
                func(conn)
        else:       ##  不是第一次进来
            if self.conn_dict[conn]['action'] == 'put':     ##  上传
                self.put_append(conn)       ##  追加

    def put_append(self,conn):
        f = open(self.conn_dict[conn]['filename'],'ab')
        if self.has_put < self.conn_dict[conn]['file_size']:
            data = conn.recv(buffer_size)       ##  接收正文
            f.write(data)
            self.has_put += len(data)
        f.close()

    def put(self,conn):      ##  上传
        print('能运行到put',conn)

        file_size = conn.recv(4)
        file_size = struct.unpack('i',file_size)
        self.conn_dict[conn]['file_size'] = file_size
        f = open(self.conn_dict[conn]['filename'],'wb')
        if self.has_put < file_size:
            data = conn.recv(buffer_size)       ##  接收正文
            f.write(data)
            self.has_put += len(data)
        f.close()


    def get(self):      ##  下载
        pass





if __name__ == '__main__':
    SelectorsServer()

############        client端
import socket,os,pickle,struct
ip_port = ('192.168.43.182',8080)
buffer_size = 1024
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
print(BASE_DIR)
class SelectorsClient:
    def __init__(self):
        self.creat_socket()
        self.handle()

    def handle(self):
        while 1:
            cmd_res = input('>>>:')
            action,filename = cmd_res.split()
            send_dict = {
                'action':action,
                'filename':filename
            }
            self.client.sendall(pickle.dumps(send_dict))        ##  传一个字典
            recv_data = self.client.recv(buffer_size)
            print(recv_data)
            if hasattr(self,action):
                func = getattr(self,action)
                func(filename)


    ##  put 123.png
    def put(self,filename):         ##  上传操作
        with open(filename,'rb') as f:
            data = f.read()
        length = len(data)
        file_size = struct.pack('i',length)
        self.client.sendall(file_size)     ##  发送文件大小
        self.client.sendall(data)       ##  发送正文



    ##  get 123.png
    def get(self):          ##  下载操作
        pass

    def creat_socket(self):
        self.client = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
        self.client.connect(ip_port)

if __name__ == '__main__':
    SelectorsClient()