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

2018-08-22  本文已影响0人  FlyingLittlePG

服务端

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()

注意:

客户端

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()

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

上一篇 下一篇

猜你喜欢

热点阅读