server.py
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
import selectorsimport socketimport osimport timeBASE_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 = 0if __name__ == '__main__': selectFtpserver() |
client.py
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
import socketimport os,sysBASE_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() |
以上就是python 基于selectors库实现文件上传与下载的详细内容，更多关于python 上传下载的资料请关注服务器之家其它相关文章！
原文链接：https://www.cnblogs.com/sugh/p/13930873.html
