點擊上方藍字關注我！

圖片來源 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-----------------------? ?

掃碼關注 獲取更多