当前位置 主页 > 网站技术 > 代码类 >

    Python socket模块ftp传输文件过程解析

    栏目:代码类 时间:2019-11-05 15:05

    这篇文章主要介绍了Python socket模块ftp传输文件过程解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下

    使用环境:python3,window环境,需要在头部声明# -*- coding:utf-8 -*-

    实现功能:

    将sever端所处文件夹的文件,传输到client端所处的文件夹中。

    并且通过md5检测是否出错。

    客户端命令的形式是: get 文件名

    client处的新文件是 文件名.new

    ftp_sever.py

    import hashlib
    import socket ,os,time
    server = socket.socket()
    server.bind(('localhost',9999))
    server.listen()
    while True:
      print("I am waiting for connection.")
      conn, addr = server.accept()
      print("new conn:",addr)
      while True:
        print("等待新指令")
        data = conn.recv(1024)
        if not data:
          print("客户端已断开")
          break
        cmd,filename = data.decode().split()
        print(filename)
        if os.path.isfile(filename):  #判断是否该文件名为文件
          f = open(filename,"rb")
          m = hashlib.md5() #为md5准备
          file_size = os.stat(filename).st_size #利用os.stat获取文件的大小
          conn.send( str(file_size).encode() ) #send file size
          conn.recv(1024) #等待确认,同时可以防止粘包。
          for line in f: #一行一行发送数据,同时更新md5
           m.update(line)  #不断更新md5
           conn.send(line) #不断发送数据。
          print("file md5", m.hexdigest()) #十六进制的md5
          f.close()
          conn.send(m.hexdigest().encode()) #send md5
        print("send done")
    server.close()

    ftp_client.py

    import socket
    import hashlib
    
    client = socket.socket()
    
    client.connect(('localhost', 9999))
    
    while True:
      cmd = input(">>:").strip() #形式 get filename
      if len(cmd) == 0: continue
      if cmd.startswith("get"):
        client.send(cmd.encode()) #发送命令,形式 get filename
        server_response = client.recv(1024) #接收文件大小信息
        print("servr response:", server_response)
        client.send(b"ready to recv file") #发送确认信息。
        file_total_size = int(server_response.decode()) #将文件大小int化。
        received_size = 0 #初始化接收数据大小,为0
        filename = cmd.split()[1] #获取文件名
        f = open(filename + ".new", "wb") #以二进制形式写入
        m = hashlib.md5() #为md5准备
    
        while received_size != file_total_size:
    '''下面的if判断是用来完整接收文件,从而避免粘包。'''
    if file_total_size - received_size > 1024: # 要收不止一次 24 size = 1024 25 else: # 最后一次了,剩多少收多少 26 size = file_total_size - received_size 27 print("last receive:", size) 28 data = client.recv(size) #data只需要是一小个内存,大小为1k就好 29 received_size += len(data) 30 m.update(data) #不断更新md5 31 f.write(data) #不断写入 32 # print(file_total_size,received_size) 33 else: 34 new_file_md5 = m.hexdigest() #获取十六进制的md5 35 print("file recv done", received_size, file_total_size) 36 f.close() 37 server_file_md5 = client.recv(1024) #接收md5值 38 print("server file md5:", server_file_md5) 39 print("client file md5:", new_file_md5) 40 41 client.close()

    以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持IIS7站长之家。