From 3d35a3e989a180b3785b4f259a03c09eed32a14d Mon Sep 17 00:00:00 2001
From: Clay Gerrard <clay.gerrard@gmail.com>
Date: Mon, 24 Feb 2014 22:41:45 -0800
Subject: Add LengthWrapper in put_object to honor content_length param

Closes-Bug:#1284360

Change-Id: Iec63a3fde77bb8195bfe46c764403b367999ff43
---
 swiftclient/utils.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'swiftclient/utils.py')

diff --git a/swiftclient/utils.py b/swiftclient/utils.py
index a038dcc..5095f9d 100644
--- a/swiftclient/utils.py
+++ b/swiftclient/utils.py
@@ -55,3 +55,21 @@ def prt_bytes(bytes, human_flag):
         bytes = '%12s' % bytes
 
     return(bytes)
+
+
+class LengthWrapper(object):
+
+    def __init__(self, readable, length):
+        self._length = self._remaining = length
+        self._readable = readable
+
+    def __len__(self):
+        return self._length
+
+    def read(self, *args, **kwargs):
+        if self._remaining <= 0:
+            return ''
+        chunk = self._readable.read(
+            *args, **kwargs)[:self._remaining]
+        self._remaining -= len(chunk)
+        return chunk
-- 
cgit v1.2.1