TCP模式的文件同步

• 由于学习写的心得的文件来回上传比较麻烦,所以直接写了一个服务器和客户端来进行文件同步

• 当服务器收到客户端的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()

客户端的代码与服务器端的差不多,只是相对倒着写而已