Python tcp socket 传送文件

xq8995209

#encoding: utf-8　　
#author: walker
　　
#date: 2017-12-01
　　
#summary: 用socket连接tcp server以传送文件
　　

　　
import os, time
　　
import socket
　　
import struct
　　

　　
cur_dir_fullpath = os.path.dirname(os.path.abspath(__file__))
　　

　　
Host = '127.0.0.1'
　　
Port = 6789
　　
BufSize = 8196
　　
FmtHead = '256sHL'#L决定单个文件必须小于4G
　　
SrcRoot = os.path.join(cur_dir_fullpath, 'input')#源目录
　　
ExtSet = {'.pdf', '.html', '.htm'}#后缀列表
　　
StartTime = time.time()#开始时间
　　

　　

　　
def ProcAll():
　　cnt = 0
　　fileSizeTotal = 0
　　sockClient = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
　　print('Connect %s:%d ...' % (Host, Port))
　　sockClient.connect((Host, Port))
　　print('local: %s:%d' % (sockClient.getsockname()))
　　for parent, dirnames, filenames in os.walk(SrcRoot):
　　for filename in filenames:
　　if os.path.splitext(filename)[1].lower() not in ExtSet:#排除非指定后缀的文件
　　continue
　　pathfile = os.path.join(parent, filename)
　　relPath = pathfile[len(SrcRoot):].strip(os.path.sep)#相对路径（相对于SrcRoot）
　　fileSize = os.path.getsize(pathfile)
　　#发送数据包头（相对路径，相对路径长度，文件大小）
　　pkgHead = struct.pack(FmtHead, relPath.encode('utf8'), len(relPath), fileSize)
　　sockClient.sendall(pkgHead)
　　with open(pathfile, mode='rb') as f:
　　while True:
　　buf = f.read(BufSize)
　　if len(buf) < 1:
　　break
　　sockClient.sendall(buf)
　　fileSizeTotal += fileSize
　　cnt += 1
　　print('cnt: %d, time cost: %.2fs, %s'% (cnt, time.time()-StartTime, pathfile))
　　#发送文件传送结束消息
　　pkgHead = struct.pack(FmtHead, 'OVER'.encode('utf8'), len('OVER'), 0)
　　sockClient.sendall(pkgHead)
　　sockClient.close()
　　print('cnt: %d, time total: %.2fs, send bytes:%ld B'% (cnt, time.time()-StartTime, fileSizeTotal))
　　

　　
if __name__ == '__main__':
　　

　　

　　
    ProcAll()
　　

　　
    print('Time total:%.2fs' % (time.time() - StartTime))