Add support for rt, rt+, wt, wt+, at, at+ methods

smart_open/smart_open_lib.py

@@ -341,8 +341,11 @@ def open(
     #
     try:
         binary_mode = {'r': 'rb', 'r+': 'rb+',
+                       'rt': 'rb', 'rt+': 'rb+',
                        'w': 'wb', 'w+': 'wb+',
-                       'a': 'ab', 'a+': 'ab+'}[mode]
+                       'wt': 'wb', "wt+": 'wb+',
+                       'a': 'ab', 'a+': 'ab+',
+                       'at': 'ab', 'at+': 'ab+'}[mode]
     except KeyError:
         binary_mode = mode
     binary, filename = _open_binary_stream(uri, binary_mode, transport_params)

smart_open/tests/test_smart_open.py

@@ -319,6 +319,9 @@ class SmartOpenFileObjTest(unittest.TestCase):
     Test passing raw file objects.
     """
 
+    def setUp(self):
+        self.temp_file = tempfile.NamedTemporaryFile(prefix='test', delete=True).name
+
     def test_read_bytes(self):
         """Can we read bytes from a byte stream?"""
         buffer = make_buffer(initial_value=SAMPLE_BYTES)
@@ -354,6 +357,64 @@ def test_read_str_from_bytes(self):
             data = sf.read()
         self.assertEqual(data, SAMPLE_TEXT)
 
+    def test_read_str_api_rt(self):
+        """Can we read strings from a byte stream?"""
+        buffer = make_buffer(initial_value=SAMPLE_BYTES)
+        with smart_open.smart_open(buffer, 'rt') as sf:
+            data = sf.read()
+        self.assertEqual(data, SAMPLE_TEXT)
+
+    def test_append_str_from_bytes_api_a(self):
+        """Can we write, append, and read strings from a real text file?"""
+        buffer = make_buffer()
+        with smart_open.smart_open(self.temp_file, 'wb') as fout:
+            fout.write(SAMPLE_BYTES)
+        with smart_open.smart_open(self.temp_file, 'ab') as fout:
+            fout.write(SAMPLE_BYTES)
+        with smart_open.smart_open(self.temp_file, 'rb') as fin:
+            data = fin.read()
+        self.assertEqual(data.decode('utf-8'), SAMPLE_TEXT+SAMPLE_TEXT)
+
+    def test_append_str_api_a_plus(self):
+        """Can we read strings from a byte stream?"""
+        buffer = make_buffer()
+        with smart_open.smart_open(self.temp_file, 'wb') as fout:
+            fout.write(SAMPLE_BYTES)
+        with smart_open.smart_open(self.temp_file, 'a+') as fout:
+            fout.write(SAMPLE_TEXT)
+        with smart_open.smart_open(self.temp_file, 'rb') as fin:
+            data = fin.read()
+        self.assertEqual(data.decode('utf-8'), SAMPLE_TEXT+SAMPLE_TEXT)
+
+    def test_append_str_api_at(self):
+        """Can we read strings from a byte stream?"""
+        buffer = make_buffer()
+        with smart_open.smart_open(self.temp_file, 'wb') as fout:
+            fout.write(SAMPLE_BYTES)
+        with smart_open.smart_open(self.temp_file, 'at') as fout:
+            fout.write(SAMPLE_TEXT)
+        with smart_open.smart_open(self.temp_file, 'rb') as fin:
+            data = fin.read()
+        self.assertEqual(data.decode('utf-8'), SAMPLE_TEXT+SAMPLE_TEXT)
+
+    def test_append_str_api_at_plus(self):
+        """Can we read strings from a byte stream?"""
+        buffer = make_buffer()
+        with smart_open.smart_open(self.temp_file, 'wb') as fout:
+            fout.write(SAMPLE_BYTES)
+        with smart_open.smart_open(self.temp_file, 'at+') as fout:
+            fout.write(SAMPLE_TEXT)
+        with smart_open.smart_open(self.temp_file, 'rb') as fin:
+            data = fin.read()
+        self.assertEqual(data.decode('utf-8'), SAMPLE_TEXT+SAMPLE_TEXT)
+
+    def test_read_str_api_rt_plus(self):
+        """Can we read strings from a byte stream?"""
+        buffer = make_buffer(initial_value=SAMPLE_BYTES)
+        with smart_open.smart_open(buffer, 'rt+') as sf:
+            data = sf.read()
+        self.assertEqual(data, SAMPLE_TEXT)
+
     def test_write_str_to_bytes(self):
         """Can we write strings to a byte stream?"""
         buffer = make_buffer()