在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解码,否则会报错