通过非IO阻塞模型实现ftp并发的小代码

import os
import time
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
import socket
import selectors

class selectFtpServer:#服务端创建一个类

    def __init__(self):#对象初始化的定义
        self.dic = {}#定义一个空字典
        self.hasReceived=0#定义一个接受值的大小
        self.sel = selectors.DefaultSelector()#实例化一个selectors对象
        self.create_socket()#调用建立连接函数
        self.handle()#调用handle函数。

    def create_socket(self):
        server = socket.socket()
        server.bind(("127.0.0.1",8885))
        server.listen(5)
        server.setblocking(False)#非IO阻塞模型
        self.sel.register(server, selectors.EVENT_READ, self.accept)
        #注册一个监听对象，如果server变化执行accept函数。
        print("服务端已开启，等待用户连接...")

    def handle(self):
        while True:
            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("from %s %s connected"%addr)
        self.sel.register(conn, selectors.EVENT_READ, self.read)
        #注册conn，如果发生变化执行read函数
        self.dic[conn] = {}#将conn传入字典，将一个空字典作为值

    def read(self, conn, mask):#read函数具体执行上传和下载的功能
        try:
            if not self.dic[conn] :#如果这个conn不在字典里，说明这是第一次接收到该客户端消息
                data = conn.recv(1024)#接收1024的内容
                cmd,filename,filesize = str(data, encoding='utf-8').split('|')#将内容以管道符分割，分别赋值
                self.dic={conn:{"cmd": cmd, "filename": filename,"filesize": int(filesize)}}#将取到的内容放到字典里
                if cmd == 'put':#如果cmd为put
                    conn.send(bytes("OK",encoding='utf8'))#回一个返回值
                if self.dic[conn]['cmd'] == 'get':#如果传来的命令是get
                    file = os.path.join(BASE_DIR,"download",filename)#拼接文件路径
                    if os.path.exists(file):#如果文件存在
                        fileSize = os.path.getsize(file)#文件大小等于filesize
                        send_info = '%s|%s'%('YES',fileSize)#将文件大小等放入事先定制好的格式当中。
                        conn.send(bytes(send_info, encoding='utf8'))#发送这个文件信息
                    else:
                        send_info = '%s|%s'%('NO',0)#如果文件不存在，将消息赋值
                        conn.send(bytes(send_info, encoding='utf8'))#发送
            else:
                if self.dic[conn].get('cmd',None):#如果conn在字典里，取出cmd，如果没有则返回None
                    cmd=self.dic[conn].get('cmd')#将get到的内容赋值给cmd
                    if hasattr(self, cmd):#如果cmd这个函数存在
                        func = getattr(self,cmd)#获取这个函数并赋值
                        func(conn)#加上参数，调用执行
                    else:
                        print("error cmd!")#如果这个函数不存在，打印error
                        conn.close()#关闭连接
                else:
                    print("error cmd!")
                    conn.close()

        except Exception as e:#异常处理
            print('error', e)
            self.sel.unregister(conn)#解除注册
            conn.close()

    def put(self, conn):#上传

        fileName = self.dic[conn]['filename']#字典的文件名赋值
        fileSize = self.dic[conn]['filesize']#字典的文件大小赋值
        path = os.path.join(BASE_DIR,"upload",fileName)#文件拼接
        recv_data = conn.recv(1024)#收1024
        self.hasReceived += len(recv_data)#已经收到的

        with open(path, 'ab') as f:#做收文件内容的操作
            f.write(recv_data)
        if fileSize == self.hasReceived:#如果收到的大小等于实际大小
            if conn in self.dic.keys():#如果这个conn在字典中存在则清空
                self.dic[conn] = {}
            print("%s上传完毕！"%fileName)

    def get(self,conn):#下载
        filename = self.dic[conn]['filename']#取文件名
        path = os.path.join(BASE_DIR,"download",filename)#拼接路径
        if str(conn.recv(1024), 'utf-8') == "second_active":#收文件
            with open(path, 'rb') as f:
                for line in f:
                    conn.send(line)
            self.dic[conn] = {}#请空字典
            print('文件下载完毕!')


if __name__ == '__main__':

    selectFtpServer()

上面是server端代码，主要就是通过socket和selectors来实现，可以多加练习。

import socket
import os,sys
BASE_DIR = os.path.dirname(os.path.abspath(__file__))

class selectFtpClient:

    def __init__(self):
        self.args=sys.argv
        if len(self.args)>1:
            self.port=(self.args[1],int(self.args[2]))#如果运行的时候后面有参数
        else:
            self.port=("127.0.0.1",8885)#默认
        self.create_socket()#创建连接函数
        self.command_fanout()#通过反射获取函数的函数

    def create_socket(self):
        try:
            self.sk = socket.socket()
            self.sk.connect(self.port)
            print('连接FTP服务器成功!')
        except Exception as e:
            print("error: ",e)

    def command_fanout(self):
        while True:
            cmd = input('>>>').strip()
            if cmd == 'exit()':
                break
            cmd,file = cmd.split()
            if hasattr(self, cmd):
                func = getattr(self, cmd)
                func(cmd,file)
            else:
                print('调用错误!')

    def put(self,cmd,file):

        if os.path.isfile(file):
            fileName= os.path.basename(file)
            fileSize = os.path.getsize(file)
            fileInfo ='%s|%s|%s'%(cmd,fileName,fileSize)
            self.sk.send(bytes(fileInfo, encoding='utf8'))
            recvStatus = self.sk.recv(1024)
            print('recvStatus', recvStatus)
            hasSend = 0
            if str(recvStatus, encoding='utf8') == "OK":
                with open(file, 'rb') as f:
                    while fileSize > hasSend :
                        contant = f.read(1024)
                        recv_size = len(contant)
                        self.sk.send(contant)
                        hasSend += recv_size
                        s=str(int(hasSend/fileSize*100))+"%"
                        print("正在上传文件："+fileName+"   已经上传："+s)
                print('%s文件上传完毕' % (fileName,))
        else:
            print('文件不存在')


if __name__ == '__main__':

    selectFtpClient()

上面为客户端代码