socket实现FTP上传下载功能

'''服务器端'''
 1 _author__ = "Dbass"
 import socketserver
 import json,os
 class MyTCPHandler(socketserver.BaseRequestHandler):
     def get(self,*args):
         '''发送客户端请求的文件'''
         cmd_dic= args[0]
         filename=cmd_dic["filename"]
         self.request.send(b"200 ok")
         if os.path.isfile(filename):
             filesize = str(os.stat(filename).st_size)
             print(type(filesize))
             self.request.send(filesize.encode("utf-8"))
             self.request.send(b"size ok")
             f = open(filename, "rb")
             for line in f:
                 self.request.send(line)
         else:
             print(filename, "is not exist")

     def put(self,*args):
         '''接收客户端文件'''
         cmd_dic = args[0]
         filename = cmd_dic["filename"]
         filesize = cmd_dic["size"]
         if os.path.isfile(filename):
             f = open(filename + ".new","wb")
         else:
             f = open(filename , "wb")

         self.request.send(b"200 ok")
         received_size = 0
         while received_size < filesize:
             data = self.request.recv(1024)
             f.write(data)
             received_size += len(data)
         else:
             print("file [%s] has uploaded..." % filename)

     def handle(self):
         while True:
             try:
                 self.data = self.request.recv(1024).strip()
                 print("{} wrote:".format(self.client_address[0]))
                 print(self.data)
                 cmd_dic = json.loads(self.data.decode())
                 action = cmd_dic["action"]
                 if hasattr(self,action):
                     func = getattr(self,action)
                     func(cmd_dic)

             except ConnectionResetError as e:
                 print("err",e)
                 break
 if __name__ == "__main__":
     HOST, PORT = "localhost", 9999
     # Create the server, binding to localhost on port 9999
     server = socketserver.ThreadingTCPServer((HOST, PORT), MyTCPHandler)
     server.serve_forever()

 '''客户端'''
 1 _author__ = "Dbass"
 import socket
 import json
 import os
 #client.connect(('192.168.16.200',9999))
 class FtpClient(object):
     def __init__(self):
         self.client = socket.socket()
     def help(self):
         msg = '''
         ls
         pwd
         cd ../..
         get filename
         put filename
         '''
         print(msg)
     def connect(self,ip,port):
         self.client.connect((ip, port))
     def interactive(self):
         #self.authenticate()
         while True:
             cmd = input(">>").strip()
             if len(cmd) ==0:continue
             cmd_str = cmd.split()[0]
             if hasattr(self,"cmd_%s" % cmd_str):
                 func = getattr(self,"cmd_%s" % cmd_str)
                 func(cmd)
             else:
                 self.help()
     def cmd_put(self,*args):
         cmd_split =  args[0].split()
         if len(cmd_split) >1:
             filename = cmd_split[1]
             if os.path.isfile(filename):
                 filesize = os.stat(filename).st_size
                 msg_dic = {
                     "action": "put",
                     "filename":filename,
                     "size": filesize,
                     "overridden":True
                 }
                 self.client.send( json.dumps(msg_dic).encode("utf-8")  )
                 print("send", json.dumps(msg_dic).encode("utf-8"))
                 # 防止粘包，等服务器确认
                 server_response = self.client.recv(1024)
                 f = open(filename,"rb")
                 for line in f:
                     self.client.send(line)
                 else:
                     print("file upload success...")
                     f.close()
             else:
                 print(filename,"is not exist")
     def cmd_get(self,*args):
         '''get IMG_6245.jpg'''
         cmd_split=args[0].split()
         if len(cmd_split) >1:
             filename=cmd_split[1]
             if os.path.isfile(filename):
                 print("filename has lived")
             else:
                 msg_dic = {
                     "action": "get",
                     "filename": filename,
                     "overridden": True
                 }
                 self.client.send(json.dumps(msg_dic).encode("utf-8"))
                 print("send", json.dumps(msg_dic).encode("utf-8"))
                 server_response = self.client.recv(1024)# 防止粘包，等服务器确认
                 print(server_response)
                 filesize = int(self.client.recv(1024).decode())
                 print(filesize)
                 server_response2 = self.client.recv(1024)# 防止粘包，等服务器确认
                 print(server_response2)
                 received_size = 0
                 f = open(filename, "wb")
                 while received_size < filesize:
                     data = self.client.recv(1024)
                     f.write(data)
                     received_size += len(data)
                 else:
                     print("file [%s] has uploaded..." % filename)

 ftp = FtpClient()
 ftp.connect("localhost",9999)
 ftp.interactive()