encode DOQL response in "UTF-8"

device42 · Jan 24, 2019 · ae76dcc · ae76dcc
1 parent c70d0e3
commit ae76dcc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib.py b/lib.py
@@ -81,7 +81,7 @@ def doql(self):
 
     @staticmethod
     def get_list_from_csv(text):
-        f = StringIO(text.decode("utf-8"))
+        f = StringIO(text.encode("utf-8", "replace"))
         list_ = []
         dict_reader = csv.DictReader(f, quotechar='"', delimiter=',', quoting=csv.QUOTE_ALL, skipinitialspace=True, dialect='excel')
         for item in dict_reader:
@@ -121,7 +121,7 @@ def write_inventory_file(groups):
         for group in groups:
             f.write('[' + group + ']\n')
             for device in groups[group]:
-                f.write(device.encode('ascii', 'replace') + '\n')
+                f.write(device + '\n')
             f.write('\n')
 
         f.close()