由于学习写的心得的文件来回上传比较麻烦,所以直接写了一个服务器和客户端来进行文件同步
当服务器收到客户端的
download命令时,将服务器的文件夹压缩后发送到客户端,然后由客户端解压缩后放到本地当服务器收到客户端的
upload命令时,将客户机本地文件夹压缩后上传到服务器,由服务器解压后放入本地需要用到的模块:
socketserver,shutil,time,os
服务端
import socket,shutil,time,os
# 服务器上文件所在的文件夹
file = r'C:\Users\wudi.HAPMMAGNA\Desktop\Python学习'
# 将文件夹中的文件全部压缩,并转化为二进制文件,给二进制文件添加nowisend结束后缀
def zip_data():
shutil.make_archive('ZipData','zip',file)
with open('ZipData.zip','rb')as f:
data = f.read()
return data + b'nowisend'
# 将从客户端上传的二进制文件先写入为压缩包,再将压缩包解压到服务器文件夹,如果文件夹中存在文件,则先删除后添加
def unzip_data(data):
with open('UnzipData.zip','wb')as f:
f.write(data)
list = os.listdir(file)
if list != None:
for i in list:
os.remove(os.path.join(file,i))
shutil.unpack_archive('UnzipData.zip',file)
# 主函数
def main():
# 建立TCP服务器,用IP4地址
addr = ('XXXXXXX',8888)
server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
# 服务器绑定地址
server.bind(addr)
# 服务器开始监听
server.listen()
# 第一死循环,让服务器不断循环等待客户端
while True:
print('等待客户端')
con,addr = server.accept()
print('客户端已连接,{0}'.format(addr))
#第二个死循环,当连接后,一直接受客户端的相应命令
while True:
# 指令
cmd = con.recv(1024).decode()
# 当指令为download时,发送压缩包二进制文件
if cmd == 'download':
data = zip_data()
con.sendall(data)
print('{0}发送文件,大小为 {1}'.format(time.ctime(), len(data)))
# 当指令为upload时,接受二进制文件,并将其解压到相应文件夹
if cmd == 'upload':
data = b''
while True:
temp = con.recv(1024)
if b'nowisend' in temp:
temp = temp.replace(b'nowisend', b'')
data += temp
break
data += temp
unzip_data(data)
print('{0}上传文件,大小为{1}'.format(time.ctime(), len(data)))
#当指令为exit时,跳出接受指令的死循环循环
if cmd == 'exit':
break
con.close()
print('客户端已关闭')
# 主函数调用
if __name__ == '__main__':
main()
注意:
服务端有两个死循环,一个是用来循环接收客户端的连接,另一个死循环是用来循环接收客户端传递的消息,最开始我只写了一个死循环,每次执行完一个指令后就会阻塞在accept处,无法接收下一个指令
-
对于二进制数据的传递有三种方式,如果想要接收完整数据,就需要采用以下方式,否则一旦
recv()中指定的值小于文件大小,就会造成阻塞短连接法,每次连接后关闭连接,数据就会传递出来,不会造成阻塞
末尾标识法,在末尾添加哨兵值,本文采取此种方式
负载长度法,传递长度值,小于长度值一直接收,否则跳出循环
客户端
import socket,shutil,os
file = r'/home/wu/桌面/Python'
def unzip_file(data):
with open('UnzipData.zip','wb')as f:
f.write(data)
list = os.listdir(file)
if list != None:
for i in list:
os.remove(os.path.join(file,i))
shutil.unpack_archive('UnzipData.zip',file)
def zip_file():
shutil.make_archive('ZipData','zip',file)
with open('ZipData.zip','rb')as f:
data = f.read()
return data + b'nowisend'
def main():
addr = ('XXXXXX',8888)
client = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
client.connect(addr)
while True:
st = input('请输入命令: ')
cmd = st.encode()
client.sendall(cmd)
if st == 'download':
data = b''
while True:
temp = client.recv(1024)
if b'nowisend' in temp:
temp = temp.replace(b'nowisend',b'')
data += temp
break
data += temp
unzip_file(data)
print('文件已下载到本机')
if st == 'upload':
data = zip_file()
client.sendall(data)
print('文件已经上传')
if st == 'exit':
client.close()
break
if __name__ == '__main__':
main()
