Python基於stuck實現scoket檔案傳輸

阿新 • • 發佈：2020-04-03

使用socket中的struck來實現客戶端傳送

服務端：

　　客戶端：

# -*- coding: UTF-8 -*-
import socket,time,socketserver,struct,os,_thread
 
host = '127.0.0.1'
port = 12307
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM) # 定義socket型別
s.bind((host,port)) # 繫結需要監聽的Ip和埠號，tuple格式
s.listen(1)
 
 
def conn_thread(connection,address):
  while True:
    try:
      connection.settimeout(600)
      fileinfo_size = struct.calcsize('12sl')#12s表示12個字元，l表示一個長整型數
      buf = connection.recv(fileinfo_size)
      if buf: # 如果不加這個if，第一個檔案傳輸完成後會自動走到下一句，需要拿到檔案大小資訊才可以繼續執行
        filename,filesize = struct.unpack('12sl',buf)
        filename_f = filename.decode("utf-8").strip('\00') # C語言中'\0'是一個ASCII碼為0的字元,在python中表示佔一個位置得空字元
        filenewname = os.path.join('e:\\',os.path.basename(filename_f))
        print(u'檔名稱:%s,檔案大小: %s' % (filenewname,filesize))
        recvd_size = 0 # 定義接收了的檔案大小
        file = open(filenewname,'wb')
        print(u"開始傳輸檔案內容")
        while not recvd_size == filesize:
          if filesize - recvd_size > 1024:
            rdata = connection.recv(1024)
            recvd_size += len(rdata)
          else:
            rdata = connection.recv(filesize - recvd_size)
            recvd_size = filesize
          file.write(rdata)
        file.close()
        print('receive done')
        # connection.close()
    except socket.timeout:
      connection.close()
 
while True:
  print(u"開始進入監聽狀態")
  connection,address = s.accept()
  print('Connected by ',address)
  # thread = threading.Thread(target=conn_thread,args=(connection,address)) #使用threading也可以
  # thread.start()
  _thread.start_new_thread(conn_thread,(connection,address))
s.close()

　　服務端效果：

# -*- coding: UTF-8 -*-
import socket,struct
 
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.connect(('127.0.0.1',12307))
while True:
  filepath = input('請輸入要傳輸的檔案絕對路徑:\r\n')
  print(type(filepath))
  print(len(filepath.encode("utf-8")))
  if os.path.isfile(filepath):
    #fileinfo_size = struct.calcsize('20sl') # 定義打包規則
    # 定義檔案頭資訊，包含檔名和檔案大小
    fhead = struct.pack('12sl',filepath.encode("utf-8"),os.stat(filepath).st_size)
    print(os.stat(filepath).st_size)
    s.send(fhead)
    print (u'檔案路徑: ',filepath)
    # with open(filepath,'rb') as fo: 這樣傳送檔案有問題，傳送完成後還會發一些東西過去
    fo = open(filepath,'rb')
    while True:
      filedata = fo.read(1024)
      if not filedata:
        break
      s.send(filedata)
    fo.close()
    print (u'傳輸成功')
    # s.close()

Python基於stuck實現scoket檔案傳輸