[Bug] Key for S3Stream should not have leading slash (#10)

* S3 key should not have leading slash * Update s3_test.py
e2fyi · Oct 28, 2020 · 98d725e · 98d725e
1 parent 45399d0
commit 98d725e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/iotoolz/extensions/s3.py b/iotoolz/extensions/s3.py
@@ -103,6 +103,8 @@ def __init__(
             use_threads=use_threads or self._default_transfer_config.use_threads,
         )
         self._scheme, self.bucket, self.key, query, _ = urllib.parse.urlsplit(uri)
+        if self.key.startswith("/"):
+            self.key = self.key[1:]
         # keep last query value
         query_str = {
             key: value[-1]

diff --git a/iotoolz/extensions/s3_test.py b/iotoolz/extensions/s3_test.py
@@ -41,7 +41,7 @@ def test_s3stream(s3):
         assert stream.info.extras["Metadata"] == {"meta-key": "meta-value"}
         assert stream.info.extras["StorageClass"] == "REDUCED_REDUNDANCY"
 
-    resp = s3.get_object_tagging(Bucket="somebucket", Key="/foo/bar.txt")
+    resp = s3.get_object_tagging(Bucket="somebucket", Key="foo/bar.txt")
     assert resp["TagSet"] == [
         {"Key": "key1", "Value": "value1"},
         {"Key": "key2", "Value": "value2"},