简单ftp—文件上传

一、客户端

 1 # Author：David Liu 
 2 # Be Happy!
 3 
 4 import socket
 5 import hashlib,os,json
 6 
 7 class FtpClient(object):
 8     def __init(self,client):
 9         self.client=socket.socket()
10     def help(self):
11         print('''
12         请按照以下格式操作：
13         put
14         get
15         …
16         '''
17         )
18     def connect(self,ip,port):
19         self.client = socket.socket()
20         self.client.connect((ip, port))
21     def interactive(self):
22         #self.authenticate()
23         while True:
24             cmd=input(">>:").strip()
25             if len(cmd)==0:continue
26             cmd_str=cmd.split()[0]
27             if hasattr(self,'cmd_%s'%cmd_str):
28                 func=getattr(self,'cmd_%s'%cmd_str)
29                 func(cmd)
30             else:
31                 self.help()
32     def cmd_put(self,*args):
33         cmd_split=args[0].split()
34         if len(cmd_split)>1:
35             filename=cmd_split[1]
36             if os.path.isfile(filename):
37                 filesize=os.stat(filename).st_size
38                 msg_dic={
39                     'action':'put',
40                     'filename':filename,
41                     'size':filesize
42                 }
43                 self.client.send(json.dumps(msg_dic).encode())
44                 #为了防止粘包，等服务器确认
45                 server_response=self.client.recv(1024)
46                 f=open(filename,'rb')
47                 for line in f:
48                     self.client.send(line)
49                 else:
50                     print(filename,'upload succeed!')
51                     f.close()
52             else:
53                 print('[%s] is not exit'%filename)
54     def cmd_get(self,*args):#下载还没写
55         pass
56 ftp=FtpClient()
57 ftp.connect('localhost',9999)
58 ftp.interactive()

二、服务器端

 1 # Author：David Liu 
 2 # Be Happy!
 3 import socketserver,json,os
 4 
 5 class MyTCPHandler(socketserver.BaseRequestHandler):
 6     def put(self,*args):
 7         '''接收客户端文件'''
 8         cmd_dic=args[0]
 9         filename=cmd_dic['filename']
10         filesize=cmd_dic['size']
11         if os.path.isfile(filename):
12             f=open(filename+'new','wb')
13         else:
14             f = open(filename, 'wb')
15 
16         self.request.send(b"200 ok")
17         received_size=0
18         while received_size<filesize:
19             data=self.request.recv(1024)
20             f.write(data)
21             received_size+=len(data)
22         else:
23             print('file [%s] has uploaded'%filename)
24 
25     def handle(self):
26         #和客户端所有的交互都在这里面写
27         while True:
28             try:
29                 self.data=self.request.recv(1024).strip()
30                 print("{} wrote:".format(self.client_address[0]))
31                 print(self.data.decode())
32                 cmd_dic=json.loads(self.data.decode())
33                 action=cmd_dic['action']
34                 if hasattr(self,action):
35                     func=getattr(self,action)
36                     func(cmd_dic)
37 
38                 if not self.data:
39                     print(self.client_address,"客户端已断开")
40                     break
41                 self.request.send(self.data.upper())
42             except ConnectionResetError as e:
43                 print("客户端已断开>>--",e)
44                 break
45 
46 if __name__=="__main__":
47     HOST,PORT="localhost",9999
48     server=socketserver.ThreadingTCPServer((HOST,PORT),MyTCPHandler)
49     server.serve_forever()

注：用于下载的get方法还没有写

2019年5月24日