点击上方蓝字关注我！

图片来源 pexels.com

简单实现文件上传、下载

1

Server端

# -*- coding: utf-8 -*-import jsonimport os__author__ = 'sange'# @Time    : 2020/8/17 下午5:26# @Author  : 'sange'# @File    : tcpserver_socket.py# @Software: PyCharmimport socketserverclass MyTCPHandler(socketserver.BaseRequestHandler):    def put(self,*args):        ''' 接收客户端传过来的文件'''        cmd_dict = args[0]        filename= cmd_dict["filename"]        filenameSize = cmd_dict["fileSize"]        if os.path.isfile(filename):            f = open(filename + "-new","wb")        else:            f = open(filename,"wb")        self.request.send(b"200 ok")        recved_size = 0        while recved_size < filenameSize:            recv_data = self.request.recv(1024)            f.write(recv_data)            recved_size+=len(recv_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)                if not self.data: break                cmd = json.loads(self.data.decode())                action= cmd["action"]                if hasattr(self,action):                    doaction=getattr(self,action)                    doaction(cmd)            except ConnectionResetError as e:                print("err",e)                break    def get(self,*args):        fileName = args[0]["fileName"]        if os.path.isfile(fileName):            file_size = os.path.getsize(fileName)            self.request.send(str(file_size).encode("utf-8"))  # 发送给客户端文件大小            self.request.recv(1024)  # wait client send 粘包            f = open(fileName,"rb")            for line in f:                self.request.send(line)            f.close()        print("[" + fileName + "]" + "文件发送完毕。。。。。。。")if __name__ == "__main__":    ip_port = ("localhost", 8000)    s = socketserver.ThreadingTCPServer(ip_port,MyTCPHandler)    s.serve_forever()

2

Client端

# -*- coding: utf-8 -*-import jsonimport random__author__ = 'sange'# @Time    : 2020/8/17 下午5:44# @Author  : 'sange'# @File    : tcpserver_client.py# @Software: PyCharm''''''import socket,os,sysclass FtpClient(object):  def __init__(self):    self.client = socket.socket()  def connect(self,ip,port):    '''    connet server    :param ip:    :param port:    :return:    '''    self.client.connect((ip,port))  def help(self):    msg = '''     ls    pwd    cd     put filename    get filename    '''    print(msg)  def jiaohu(self):    '''    send cmd to server,and recv server return result    '''    while True:      cmd_input = input("输入要执行的操作==>").strip()      if len(cmd_input) == 0:continue      cmd_action = cmd_input.split()[0] #获取命令执行相应的方法 如 get test.py执行下面的get方法,put则执行下面的put方法      if hasattr(self,"cmd_%s" % cmd_action):        func = getattr(self,"cmd_%s" % cmd_action)        func(cmd_input)      else:        self.help()  def cmd_put(self,*args):    #解析命令参数,put filename    cmd_ac = args[0].split()    if len(cmd_ac) > 1:      filename = cmd_ac[1]      if os.path.isfile(filename):        fileSize = os.path.getsize(filename)        put_param = {          "fileSize":fileSize,          "action":"put",          "filename":filename        }        self.client.send(json.dumps(put_param).encode("utf-8"))        #防止粘包,等服务器确认        recv = self.client.recv(1024)        f = open(filename,"rb")        for line in f:          self.client.send(line)        else:          print("%s file upload success" % filename)        f.close()      else:        print("%s file is not exist"%filename)  def cmd_get(self,*args):    ''' get filename'''    cmd_inp = args[0] # get test    fileName = cmd_inp.split()[1]    json_get = {      "fileName":fileName,      "action":"get"    }    self.client.send(json.dumps(json_get).encode("utf-8"))    server_response = self.client.recv(1024)    print("servr response:", server_response)    self.client.send(b"ready to recv file")    file_total_size = int(server_response.decode())    received_size = 0    f = open(fileName+'-'+str(random.randint(100000,900000)), 'wb')    while received_size < file_total_size:      if file_total_size - received_size > 1024:        size = 1024      else:        size = file_total_size - received_size      data = self.client.recv(size)      received_size += len(data)      f.write(data)    else:      print("cmd res receive done...", received_size)      f.close()if __name__ == "__main__":  f = FtpClient()  f.connect("localhost",8000)  f.jiaohu()

------------------------end-----------------------   

扫码关注 获取更多