Merge pull request timkpaine#348 from timkpaine/tkp/contenttype

Determine content type locally

pyEX/common/__init__.py

@@ -32,6 +32,7 @@
     _timeseriesWrapper,
     _toDatetime,
     json_normalize,
+    _interpolateDatatype,
 )
 from .exception import *
 from .timing import _EST, _PYEX_CACHE_FOLDER, _UTC, _expire, _interval

pyEX/common/checks.py

@@ -12,6 +12,7 @@
 
 import re
 import pandas as pd
+import json
 from six import string_types
 
 from .exception import PyEXception
@@ -535,6 +536,25 @@ def _overrideFormat(kwargs):
     kwargs["format"] = "json"
 
 
+def _interpolateDatatype(data):
+    """Attempt to determine the data type from the data"""
+    if isinstance(data, str):
+        # check if json or else default to csv
+        try:
+            if data.strip().startswith("[{") or data.strip().startswith("{"):
+                return data, {"content-type": "application/json"}
+            return data, {"content-type": "text/csv"}
+
+        except ValueError:
+            return data, {"content-type": "text/csv"}
+
+    if isinstance(data, list):
+        if data and isinstance(data[0], dict):
+            # json
+            return json.dumps(data), {"content-type": "application/json"}
+    raise PyEXception("Could not determine data type")
+
+
 try:
     if pd.__version__ > "1.":
         json_normalize = pd.json_normalize