服务端代码：程序目录server/server.py 上传文件目录:server/upload
import os import time import socket import selectors #封装了一些相应的操作 BASE_DIR=os.path.dirname(os.path.abspath(__file__)) #第二步： class selectFtpServer: def __init__(self): self.dic = {} #创建空字典 self.hasReceived=0 self.sel = selectors.DefaultSelector() #通过selectors模块下的DefaultSelector这个类拿I/O多路方法拿到实例对象变成实例变量self.sel self.create_socket() #create_socket()是创建socket对象函数完成绑定功能 self.handle() #handle()函数完成循环监听 #create_socket函数是用来创建socket对象完成绑定功能 def create_socket(self): server = socket.socket() #创建socket对象赋值server server.bind(('192.168.1.160', 8080)) #绑定 server.listen(5) #监听 server.setblocking(False) #设置非阻塞 self.sel.register(server, selectors.EVENT_READ,self.accept) #有了server对象通过self.sel实例变量register注册完成绑定功能(server跟self.accept做绑定) print("服务端已启动，等待客户端连接.....") #handle()函数完成循环监听 def handle(self): while True: #第一步：程序启动后走while循环 #第七步：运行完accept函数后到这里 #所有操作围绕一个对象self.sel实例变量核心对象展开的 events = self.sel.select() #第二步：调用self.sel实例变量，监听的内容server封装到events对象里 #第八步：如果客户端发过来此刻监听的内容就有变化有俩个对象server和conn封装到events对象里 for key, mask in events: #第三步：for循环events(可迭代对象)拿到key和mask #第九步：for循环events(可迭代对象)拿到key和mask callback = key.data #第四步：当前key.data是self.accept函数赋值给callback #第十步：当前key.data是read函数赋值给callback #key.fileobj是拿到的监听的对象。 callback(key.fileobj, mask) #第五步：运行callback执行accpt()函数 #第十一步：运行callback执行read()函数里面放到的之前链接相应的文件描述符conn #print('当前监听的对象:', key.fileobj) #第六步：accept函数接收连接服务端的客户端信息 def accept(self,sock,mask): #接收sock和mask conn, addr = sock.accept() #sock.accept接收此时链接服务端的socket对象拿到conn和addr #print('accepted', conn, 'from', addr) print("from %s %s connected" %addr) conn.setblocking(False) #设置成非阻塞 self.sel.register(conn, selectors.EVENT_READ, self.read) #把conn跟read做绑定后程序返回到handle()函数里的events继续监听 #print(conn) self.dic[conn] = {} #在空字典里进行了conn赋值,self.dic={conn:{},} #第十二步：read函数返回给客户端ok def read(self,conn, mask): #接收了conn和mask try: #加异常防止客户端突然断开 if not self.dic[conn]: #判断self.dic[conn]里面是否是空字典，如果是空字典，代表第一次进来 data = conn.recv(1024) #conn接收了客户端发来的数据 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继续监听 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): fileSize = os.path.getsize(file) 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): #对接收的命令进行分发判断是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'] 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] = {} print("%s 上传完毕！" %fileName) if __name__=='__main__': selectFtpServer() #第一步：实例化触发selectFtpServer这个类
客户端代码：目录结果：/client/clenb.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=("192.168.1.160",8080) #如果没有第二个参数默认取这个 self.create_socket() # self.command_fanout() #进行命令分发 #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: cmd = input('>>>').strip() #引导用户输入上传还是下载 if cmd == 'exit()': break cmd,file = cmd.split() #把输入的命令分开进行反射 if hasattr(self,cmd): func = getattr(self,cmd) func(cmd,file) else: print('调用错误！') #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,file): pass if __name__=='__main__': selectFtpClient()
