2018-05-21文件下载器-客户端优化版

import socket

def main():

    '''文件下载器客户端优化'''

    # 创建套接字

    cli_soc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 连接服务器

    cli_soc.connect(('127.0.0.1', 8080))

    # 发送文件名

    file_name = input('请输入文件名 ：')

    cli_soc.send(file_name.encode('utf-8'))

    # 接受文件内容

    file_data = cli_soc.recv(1024*1024)

    if file_data:

        # 保存内容到本地文件

        with open(file_name, 'wb') as file:

            file.write(file_data)

    else:

        print('服务器数据为空，文件不存在')

    # 关闭套接字

    cli_soc.close()

if __name__ == '__main__':

    main()

#如有错误之处，还望大家指正，谢谢！