Data arg in write() can be single string (influxdata#492)

baftek · sebito91 · commit 265d14736bb7 · 2017-08-22T01:18:52.000-05:00
* data arg in write() can be single string

data argument in write() was expected to be list of strings, now it can be single string as well

* Fix linter error

* Trailing whitespace removed

* Transformed to isinstance instead of using type()

* Change list(str) to [str]
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -271,6 +271,7 @@ def write(self, data, params=None, expected_response_code=204,
         :param data: the data to be written
         :type data: (if protocol is 'json') dict
                     (if protocol is 'line') sequence of line protocol strings
+                                            or single string
         :param params: additional parameters for the request, defaults to None
         :type params: dict
         :param expected_response_code: the expected response code of the write
@@ -292,6 +293,8 @@ def write(self, data, params=None, expected_response_code=204,
         if protocol == 'json':
             data = make_lines(data, precision).encode('utf-8')
         elif protocol == 'line':
+            if isinstance(data, str):
+                data = [data]
             data = ('\n'.join(data) + '\n').encode('utf-8')
 
         self.request(
