Python3代码-selctors实现文件上传与下载
server.py
import selectors import socket import os import time BASE_DIR =os.path.abspath(os.path.dirname(__file__)) class selectFtpserver: def __init__(self): self.dic = {} # 创建空字典 self.hasReceived = 0 self.hasSend=0 self.sel = selectors.DefaultSelector() # 生成一个select对象 self.create_socket() #create_socket()是创建socket对象函数完成绑定功能 self.hanle() #handle()函数完成循环监听 def create_socket(self): sock = socket.socket() sock.bind(('127.0.0.1', 8899)) sock.listen() sock.setblocking(False) self.sel.register(sock, selectors.EVENT_READ, self.accept) # 把刚生成的sock连接对象注册到select连接列表中,并交给accept函数处理 print("服务端已打开,请连接客户端") def hanle(self): while True: events = self.sel.select() # 默认是阻塞,有活动连接就返回活动的连接列表 # 这里看起来是select,其实有可能会使用epoll,如果你的系统支持epoll,那么默认就是epoll # print("event==",events) for key, mask in events: callback = key.data # 去调accept函数 callback(key.fileobj, mask) # key.fileobj就是readable中的一个socket连接对象 def accept(self,sock, mask): conn, addr = sock.accept() # Should be ready print('accepted', conn, 'from', addr) conn.setblocking(False) # 设定非阻塞 self.sel.register(conn, selectors.EVENT_READ, self.read) # 新连接注册read回调函数 self.dic[conn] = {} # 在空字典里进行了conn赋值,self.dic={conn:{},} def read(self, conn, mask): # 接收了conn和mask try: # 加异常防止客户端突然断开 if not self.dic[conn]: # 判断self.dic[conn]里面是否是空字典,如果是空字典,代表第一次进来 print('====第一次进来') data = conn.recv(1024) # conn接收了客户端发来的数据 print("data==",str(data, encoding='utf-8')) cmd, filename,filesize = str(data, encoding='utf-8').split('|') # 把接收到客户端发来的包解开拿到cmd,filename,filesize个信息 self.dic = {conn: {"cmd": cmd, "filename": filename, "filesize": int(filesize)}} # 把拿到的cmd,filename,filesize信息放到self.dic字典里去后程序返回到handle()函数里的events继续监听 print(self.dic) if cmd == 'put': # 如果接收的信息是put conn.send(bytes("OK", encoding='utf8')) # 给客户端返回一条数据 if self.dic[conn]['cmd'] == 'get': file = os.path.join(BASE_DIR, "upload", filename) if os.path.exists(file): print("文件存在的情况,返回YES给客户端") filesize = os.path.getsize(file) self.dic[conn]['filesize'] = filesize print("self.dic",self.dic) send_info = '%s|%s' % ('YES', filesize) conn.send(bytes(send_info, encoding='utf8')) else: print("文件不存在情况下") send_info = '%s|%s' % ('NO', 0) conn.send(bytes(send_info, encoding='utf8')) self.dic[conn] = {} #文件不存在的情况下,要将清空字典 else: # 如果不是空字典代表不是第一次进来 print('不是第一次来的') print(self.dic) if self.dic[conn].get('cmd', None): # 对接收的命令进行分发判断是put还是get cmd = self.dic[conn].get('cmd') if hasattr(self, cmd): # 如果cmd=put调用put函数,如果是cmd=get函数调用get函数 func = getattr(self, cmd) func(conn) else: print("error cmd!") conn.close() else: print("error cmd!") conn.close() except Exception as e: print('断开的客户端信息是:', conn) self.sel.unregister(conn) # 如果没有接收到数据做一个关闭解除 conn.close() # put上传函数 def put(self, conn): fileName = self.dic[conn]['filename'] fileSize = self.dic[conn]['filesize'] # print("BASE_DIR",BASE_DIR) path = os.path.join(BASE_DIR, "upload", fileName) # 拿到要接收的信息 # print(fileName,fileSize,path) recv_data = conn.recv(1024) # 接收客户端上传的数据1024字节 self.hasReceived += len(recv_data) # 把接收的数据累加到变量self.hasReceived with open(path, 'ab') as f: # 打开文件 f.write(recv_data) # 把接收的数据写到文件里去 if fileSize == self.hasReceived: # 判断文件大小跟接收大小是否一样 if conn in self.dic.keys(): # 如果文件大小跟接收大小一样清空字典 self.dic[conn] = {} self.hasReceived = 0 #S上传结束之后,需要将self.hasReceived 重置成功 print("%s 上传完毕!" % fileName) def get(self,conn): fileName = self.dic[conn]['filename'] file = os.path.join(BASE_DIR, "upload", fileName) # fileSize = os.path.getsize(file) fileSize=self.dic[conn]['filesize'] data = conn.recv(1024) # conn接收了客户端发来的数据 dataOK = str(data, encoding='utf-8') if dataOK == 'OK': with open(file, 'rb') as f: # 打开文件 while fileSize > self.hasSend: # 循环的发送文件给客户端 contant = f.read(1024) recv_size = len(contant) conn.send(contant) self.hasSend += recv_size s = str(int(self.hasSend / fileSize * 100)) + "%" print("正在下载文件: " + fileName + " 已经下载:" + s) if fileSize == self.hasSend: # 判断文件大小跟接收大小是否一样 if conn in self.dic.keys(): # 如果文件大小跟接收大小一样清空字典 self.dic[conn] = {} print("%s 下载完毕!" % fileName) self.hasSend = 0 if __name__ == '__main__': selectFtpserver()
client.py
import socket import os,sys BASE_DIR=os.path.dirname(os.path.abspath(__file__)) class selectFtpClient: def __init__(self): self.args=sys.argv #sys.argv在命令行输入的参数,第一个参数默认文件名,第二个参数跟IP地址和端口 if len(self.args)>1: #如果大于1把第二个参数俩个值赋值给port self.port=(self.args[1],int(self.args[2])) else: self.port=("127.0.0.1",8899) #如果没有第二个参数默认取这个 self.create_socket() # self.command_fanout() #进行命令分发 self.mainPath = os.path.join(BASE_DIR, 'filename') # 获取该客户端下的filename路径 #create_socket函数创建socket对象连接服务端 def create_socket(self): try: self.sk = socket.socket() self.sk.connect(self.port) print('连接FTP服务器成功!') except Exception as e: print("eroor:",e) #command_fanout()函数进行命令分发 def command_fanout(self): while True: try: print("----------------welcome to ftp client-------------------") self.help_info() cmd_info = input('>>>请输入操作命令:').strip() # put 12.png images if not cmd_info: continue cmd,file = cmd_info.split() ##按照空格分隔 # print("命令是什么", cmds) if cmd == "quit": break if hasattr(self, cmd): func = getattr(self, cmd) func(cmd,file) Tag = input("是否继续进入ftp clinet,请选择Y/N:").strip() if Tag.upper() == 'Y': continue else: break else: print('No such command ,please try again') except Exception as e: # server关闭了 print('%s' % e) break def help_info(self): print (''' get + (文件名) 表示下载文件 put + (文件名) 表示上传文件 quit 表示退出登录 ''') #put()上传函数 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) #给文件名字大小打包成fileInf self.sk.send(bytes(fileInfo, encoding='utf8')) #调用send方法把fileInf发给服务端 recvStatus = self.sk.recv(1024) #接收服务端返回的OK内容 print('recvStatus' , recvStatus) hasSend = 0 if str(recvStatus, encoding='utf8') == "OK": #如果接收到服务端返回的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('要上传的文件不存在') #get()下载函数 def get(self,cmd,fileName): path = os.path.join(BASE_DIR, "download", fileName) # 拿到要接收的信息 fileSize=0 fileInfo = '%s|%s|%s' % (cmd, fileName, fileSize) # 给文件名字大小打包成fileInf print(fileInfo) self.sk.send(bytes(fileInfo, encoding='utf8')) # 调用send方法把fileInfo发给服务端 recvdata = self.sk.recv(1024) # 接收服务端返回的是否存在文件内容 recvStatus, fileSize = str(recvdata, encoding='utf-8').split('|') print("recvStatus==",recvStatus,fileSize) fileSize = int(fileSize) hasReceived = 0 if recvStatus == "YES": # 如果接收到服务端返回的YES self.sk.send(bytes('OK', encoding='utf8')) # 通知服务端可以正常下载了 while fileSize > hasReceived: # 循环的发送文件给客户端 recv_data = self.sk.recv(1024) # 接收客户端上传的数据1024字节 hasReceived += len(recv_data) # 把接收的数据累加到变量self.hasReceived print("hasReceived",hasReceived) with open(path, 'ab') as f: # 打开文件 f.write(recv_data) # 把接收的数据写到文件里去 if fileSize == hasReceived: # 判断文件大小跟接收大小是否一样 print("%s 下载完毕!" % fileName) recvStatus = 'YESS' else: print('要下载的文件不存在') if __name__=='__main__': selectFtpClient()