python3中實現客戶端與服務端交互發送文件

在python3中實現客戶端與服務端程序交互,從客戶端發送文件到服務端

客戶端代碼:client.py

#!/usr/bin/env python
#_*_ encoding=utf-8 _*_


import socket,sys,os

ip_port = ('127.0.0.1',9999)
sk = socket.socket()
sk.connect(ip_port)

container = {'key':'','data':''}
while True:
    input_data = input('path:')
    cmd,path = input_data.split('|')
    file_name = os.path.basename(path)
    file_size = os.stat(path).st_size
    sk.send((cmd+"|"+file_name+"|"+str(file_size)).encode())
    send_size = 0
    f = open(path,'rb')
    Flag = True
    while Flag:
        if send_size + 1024 > file_size:
            data = f.read(file_size - send_size)
            Flag = False
        else:
            data = f.read(1024)
            send_size += 1024
        sk.send(data)
    f.close()
sk.close()


服務端代碼server.py

#!/usr/bin/env python
# _*_ encoding=utf-8 _*_

import socketserver,os

class MyServer(socketserver.BaseRequestHandler):
    def handle(self):
        base_path = 'E:\\\\temp'
        conn = self.request
        print('connected...')
        while True:
            pre_data = conn.recv(1024).decode()
            #獲取請求方法,文件名,文件大小
            cmd,file_name,file_size = pre_data.split('|')
            #已經接收文件的大小
            recv_size = 0
            #上傳文件路徑拼接
            file_dir = os.path.join(base_path,file_name)
            f = open(file_dir,'wb')
            Flag = True
            while Flag:
                #未上傳完畢
                if int(file_size)>recv_size:
                    data = conn.recv(1024)
                    recv_size += len(data)
                else:
                    recv_size = 0
                    Flag = False
                    continue
                #寫入文件
                f.write(data)
            print('upload successed')
            f.close()
            
instance = socketserver.ThreadingTCPServer(('127.0.0.1',9999),MyServer)            
instance.serve_forever()

先啓動服務端程序,再從命令行中運行client.py

E:\\python\\work\\Pythonlearn1\\src\\test2>python client.py
path:put|E:\\DOWNLOAD\\host.txt
path:put|E:\\DOWNLOAD\\host2.txt
path:Traceback (most recent call last):

如果不取消,客戶端會一進提示輸入,命令中有一上cmd,可以定義發送或者接收,如果put,get,然後根據這上進行判斷做進一步處理

這裏有一個問題困擾很久,在python3中,字符串不能直接發送,需要byte,這裏使用了encode,接收時再使用decode解碼,否則會報錯

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章