From 45e12215678001da731f002c4004cd803ff2b7d7 Mon Sep 17 00:00:00 2001
From: Cameron Yick <cameron.yick@yale.edu>
Date: Sun, 22 Oct 2017 18:47:17 -0400
Subject: [PATCH] Add Objects to Prices endpoint, Refactor with List
 Comprehension

---
 tiingo/api.py | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/tiingo/api.py b/tiingo/api.py
index 8285f98..2a63e0c 100644
--- a/tiingo/api.py
+++ b/tiingo/api.py
@@ -123,7 +123,7 @@ class TiingoClient(RestClient):
         """
         url = "tiingo/daily/{}/prices".format(ticker)
         params = {
-            'format': fmt,
+            'format': fmt if fmt != "object" else 'json',  # conversion local
             'frequency': frequency
         }
 
@@ -137,6 +137,9 @@ class TiingoClient(RestClient):
         response = self._request('GET', url, params=params)
         if fmt == "json":
             return response.json()
+        elif fmt == "object":
+            data = response.json()
+            return [dict_to_object(item, "TickerPrice") for item in data]
         else:
             return response.content.decode("utf-8")
 
@@ -175,12 +178,7 @@ class TiingoClient(RestClient):
         if fmt == 'json':
             return data
         elif fmt == 'object':
-            obj_arr = []
-            for el in data:
-                # inspired by https://stackoverflow.com/a/15882054
-                arr_el = dict_to_object(el, "NewsArticle")
-                obj_arr.append(arr_el)
-            return obj_arr
+            return [dict_to_object(item, "NewsArticle") for item in data]
 
     def get_bulk_news(self, file_id=None, fmt='json'):
         """Only available to institutional clients.