Python tcp socket 传送文件
#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).lower() not in ExtSet:#排除非指定后缀的文件
continue
pathfile = os.path.join(parent, filename)
relPath = pathfile.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))
页:
[1]