1 import os
2 import time
3 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
4 import socket
5 import selectors
6
7 class selectFtpServer:#服务端创建一个类
8
9 def __init__(self):#对象初始化的定义
10 self.dic = {}#定义一个空字典
11 self.hasReceived=0#定义一个接受值的大小
12 self.sel = selectors.DefaultSelector()#实例化一个selectors对象
13 self.create_socket()#调用建立连接函数
14 self.handle()#调用handle函数。
15
16 def create_socket(self):
17 server = socket.socket()
18 server.bind(("127.0.0.1",8885))
19 server.listen(5)
20 server.setblocking(False)#非IO阻塞模型
21 self.sel.register(server, selectors.EVENT_READ, self.accept)
22 #注册一个监听对象，如果server变化执行accept函数。
23 print("服务端已开启，等待用户连接...")
24
25 def handle(self):
26 while True:
27 events = self.sel.select()#监听，将变化的内容赋值
28 for key, mask in events:#循环发生变化的对象
29 callback = key.data#将该对象的方法赋值
30 callback(key.fileobj, mask)#执行这个方法并传参
31
32 def accept(self,sock, mask):
33 conn, addr = sock.accept()
34 print("from %s %s connected"%addr)
35 self.sel.register(conn, selectors.EVENT_READ, self.read)
36 #注册conn，如果发生变化执行read函数
37 self.dic[conn] = {}#将conn传入字典，将一个空字典作为值
38
39 def read(self, conn, mask):#read函数具体执行上传和下载的功能
40 try:
41 if not self.dic[conn] :#如果这个conn不在字典里，说明这是第一次接收到该客户端消息
42 data = conn.recv(1024)#接收1024的内容
43 cmd,filename,filesize = str(data, encoding='utf-8').split('|')#将内容以管道符分割，分别赋值
44 self.dic={conn:{"cmd": cmd, "filename": filename,"filesize": int(filesize)}}#将取到的内容放到字典里
45 if cmd == 'put':#如果cmd为put
46 conn.send(bytes("OK",encoding='utf8'))#回一个返回值
47 if self.dic[conn]['cmd'] == 'get':#如果传来的命令是get
48 file = os.path.join(BASE_DIR,"download",filename)#拼接文件路径
49 if os.path.exists(file):#如果文件存在
50 fileSize = os.path.getsize(file)#文件大小等于filesize
51 send_info = '%s|%s'%('YES',fileSize)#将文件大小等放入事先定制好的格式当中。
52 conn.send(bytes(send_info, encoding='utf8'))#发送这个文件信息
53 else:
54 send_info = '%s|%s'%('NO',0)#如果文件不存在，将消息赋值
55 conn.send(bytes(send_info, encoding='utf8'))#发送
56 else:
57 if self.dic[conn].get('cmd',None):#如果conn在字典里，取出cmd，如果没有则返回None
58 cmd=self.dic[conn].get('cmd')#将get到的内容赋值给cmd
59 if hasattr(self, cmd):#如果cmd这个函数存在
60 func = getattr(self,cmd)#获取这个函数并赋值
61 func(conn)#加上参数，调用执行
62 else:
63 print("error cmd!")#如果这个函数不存在，打印error
64 conn.close()#关闭连接
65 else:
66 print("error cmd!")
67 conn.close()
68
69 except Exception as e:#异常处理
70 print('error', e)
71 self.sel.unregister(conn)#解除注册
72 conn.close()
73
74 def put(self, conn):#上传
75
76 fileName = self.dic[conn]['filename']#字典的文件名赋值
77 fileSize = self.dic[conn]['filesize']#字典的文件大小赋值
78 path = os.path.join(BASE_DIR,"upload",fileName)#文件拼接
79 recv_data = conn.recv(1024)#收1024
80 self.hasReceived += len(recv_data)#已经收到的
81
82 with open(path, 'ab') as f:#做收文件内容的操作
83 f.write(recv_data)
84 if fileSize == self.hasReceived:#如果收到的大小等于实际大小
85 if conn in self.dic.keys():#如果这个conn在字典中存在则清空
86 self.dic[conn] = {}
87 print("%s上传完毕！"%fileName)
88
89 def get(self,conn):#下载
90 filename = self.dic[conn]['filename']#取文件名
91 path = os.path.join(BASE_DIR,"download",filename)#拼接路径
92 if str(conn.recv(1024), 'utf-8') == "second_active":#收文件
93 with open(path, 'rb') as f:
94 for line in f:
95 conn.send(line)
96 self.dic[conn] = {}#请空字典
97 print('文件下载完毕!')
98
99
100 if __name__ == '__main__':
101
102 selectFtpServer()
上面是server端代码，主要就是通过socket和selectors来实现，可以多加练习。
1 import socket
2 import os,sys
3 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
4
5 class selectFtpClient:
6
7 def __init__(self):
8 self.args=sys.argv
9 if len(self.args)>1:
10 self.port=(self.args[1],int(self.args[2]))#如果运行的时候后面有参数
11 else:
12 self.port=("127.0.0.1",8885)#默认
13 self.create_socket()#创建连接函数
14 self.command_fanout()#通过反射获取函数的函数
15
16 def create_socket(self):
17 try:
18 self.sk = socket.socket()
19 self.sk.connect(self.port)
20 print('连接FTP服务器成功!')
21 except Exception as e:
22 print("error: ",e)
23
24 def command_fanout(self):
25 while True:
26 cmd = input('>>>').strip()
27 if cmd == 'exit()':
28 break
29 cmd,file = cmd.split()
30 if hasattr(self, cmd):
31 func = getattr(self, cmd)
32 func(cmd,file)
33 else:
34 print('调用错误!')
35
36 def put(self,cmd,file):
37
38 if os.path.isfile(file):
39 fileName= os.path.basename(file)
40 fileSize = os.path.getsize(file)
41 fileInfo ='%s|%s|%s'%(cmd,fileName,fileSize)
42 self.sk.send(bytes(fileInfo, encoding='utf8'))
43 recvStatus = self.sk.recv(1024)
44 print('recvStatus', recvStatus)
45 hasSend = 0
46 if str(recvStatus, encoding='utf8') == "OK":
47 with open(file, 'rb') as f:
48 while fileSize > hasSend :
49 contant = f.read(1024)
50 recv_size = len(contant)
51 self.sk.send(contant)
52 hasSend += recv_size
53 s=str(int(hasSend/fileSize*100))+"%"
54 print("正在上传文件："+fileName+" 已经上传："+s)
55 print('%s文件上传完毕' % (fileName,))
56 else:
57 print('文件不存在')
58
59
60 if __name__ == '__main__':
61
62 selectFtpClient()
上面为客户端代码