python socket 多线程 文件传输_python单线程文件传输的实例(C/S)

客户端代码：

#-*-encoding:utf-8-*-

import socket

import os

import sys

import math

import time

def progressbar(cur, total):

percent = '{:.2%}'.format(float(cur) / float(total))

sys.stdout.write('\r')

sys.stdout.write("[%-50s] %s" % (

'=' * int(math.floor(cur * 50 / total)),

percent))

sys.stdout.flush()

def getFileSize(file):

file.seek(0, os.SEEK_END)

fileLength = file.tell()

file.seek(0, 0)

return fileLength

def getFileName(fileFullPath):

index = fileFullPath.rindex('\\')

if index == -1:

return fileFullPath

else:

return fileFullPath[index+1:]

def transferFile():

fileFullPath = r"%s" % raw_input("File path: ").strip("\"")

if os.path.exists(fileFullPath):

timeStart = time.clock()

file = open(fileFullPath, 'rb')

fileSize = getFileSize(file)

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

client.connect((targetHost, targetPort))

# send file size

client.send(str(fileSize))

response = client.recv(1024)

# send file name

client.send(getFileName(fileFullPath))

response = client.recv(1024)

# send file content

sentLength = 0

while sentLength < fileSize:

bufLen = 1024

buf = file.read(bufLen)

client.send(buf)

sentLength += len(buf)

process = int(float(sentLength) / float(fileSize) * 100)

progressbar(process, 100)

client.recv(1024)

file.close()

timeEnd = time.clock()

print "\r\nFinished, spent %d seconds" % (timeEnd - timeStart)

else:

print "File doesn't exist"

targetHost = raw_input("Server IP Address: ")

targetPort = int(raw_input("Server port: "))

while True:

transferFile()

服务器端代码：

#-*-encoding:utf-8-*-

import socket

import threading

import os

import sys

import math

bindIp = "0.0.0.0"

bindPort = 9999

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

server.bind((bindIp, bindPort))

server.listen(1)

print "Listening on %s:%d" % (bindIp, bindPort)

def progressbar(cur, total):

percent = '{:.2%}'.format(float(cur) / float(total))

sys.stdout.write('\r')

sys.stdout.write("[%-50s] %s" % (

'=' * int(math.floor(cur * 50 / total)),

percent))

sys.stdout.flush()

def checkFileName(originalFileName):

extensionIndex = originalFileName.rindex(".")

name = originalFileName[:extensionIndex]

extension = originalFileName[extensionIndex+1:]

index = 1

newNameSuffix = "(" + str(index) + ")"

finalFileName = originalFileName

if os.path.exists(finalFileName):

finalFileName = name + " " + newNameSuffix + "." + extension

while os.path.exists(finalFileName):

index += 1

oldSuffix = newNameSuffix

newNameSuffix = "(" + str(index) + ")"

finalFileName = finalFileName.replace(oldSuffix, newNameSuffix)

return finalFileName

def handleClient(clientSocket):

# receive file size

fileSize = int(clientSocket.recv(1024))

# print "[<==] File size received from client: %d" % fileSize

clientSocket.send("Received")

# receive file name

fileName = clientSocket.recv(1024)

# print "[<==] File name received from client: %s" % fileName

clientSocket.send("Received")

fileName = checkFileName(fileName)

file = open(fileName, 'wb')

# receive file content

print "[==>] Saving file to %s" % fileName

receivedLength = 0

while receivedLength < fileSize:

bufLen = 1024

if fileSize - receivedLength < bufLen:

bufLen = fileSize - receivedLength

buf = clientSocket.recv(bufLen)

file.write(buf)

receivedLength += len(buf)

process = int(float(receivedLength) / float(fileSize) * 100)

progressbar(process, 100)

file.close()

print "\r\n[==>] File %s saved." % fileName

clientSocket.send("Received")

while True:

client, addr = server.accept()

print "[*] Accepted connection from: %s:%d" % (addr[0], addr[1])

clientHandler = threading.Thread(target=handleClient, args=(client,))

clientHandler.start()

运行结果示例：

服务器端：

客户端(服务器端做了端口映射：59999->9999)：

以上这篇python单线程文件传输的实例(C/S)就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持我们。

本文标题: python单线程文件传输的实例(C/S)

本文地址: http://www.cppcns.com/jiaoben/python/252112.html