trying to fix py2

skillachie · Apr 17, 2017 · 50b14fd · 50b14fd
1 parent 4397e25
commit 50b14fd
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/finsymbols/symbol_helper.py b/finsymbols/symbol_helper.py
@@ -34,8 +34,12 @@ def get_symbol_list(symbol_data, exchange_name):
 
 
 def save_file(file_path, file_data):
-    with open(file_path, "wb") as saved_file:
-        saved_file.write(file_data.encode('utf-8'))
+    if isinstance(file_data, str):
+        with open(file_path, "w") as saved_file:
+            saved_file.write(file_data)
+    elif isinstance(file_data, bytes):
+        with open(file_path, "wb") as saved_file:
+            saved_file.write(file_data.encode('utf-8'))
 
 
 def get_exchange_url(exchange):