美文网首页
采用TCP模式的服务器与客户端通过命令实现文件同步

采用TCP模式的服务器与客户端通过命令实现文件同步

作者: FlyingLittlePG | 来源:发表于2018-08-22 10:32 被阅读0次
    • 由于学习写的心得的文件来回上传比较麻烦,所以直接写了一个服务器和客户端来进行文件同步

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

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

    相关文章

      网友评论

          本文标题:采用TCP模式的服务器与客户端通过命令实现文件同步

          本文链接:https://www.haomeiwen.com/subject/tjyeiftx.html