diff --git a/src/server/endpoints/covidcast.py b/src/server/endpoints/covidcast.py index a1df64d83..ff447e085 100644 --- a/src/server/endpoints/covidcast.py +++ b/src/server/endpoints/covidcast.py @@ -21,7 +21,7 @@ parse_single_time_arg, parse_single_geo_arg, ) -from .._query import QueryBuilder, execute_query, run_query, parse_row +from .._query import QueryBuilder, execute_query, run_query, parse_row, filter_fields from .._printer import create_printer, CSVPrinter from .._validate import ( extract_date, @@ -200,7 +200,7 @@ def gen(rows): raise DatabaseErrorException(str(e)) # now use a generator for sending the rows and execute all the other queries - return p(gen(r)) + return p(filter_fields(gen(r))) @bp.route("/correlation", methods=("GET", "POST")) @@ -260,7 +260,7 @@ def gen(): yield cor.asdict() # now use a generator for sending the rows and execute all the other queries - return p(gen()) + return p(filter_fields(gen())) @bp.route("/csv", methods=("GET", "POST")) @@ -405,7 +405,7 @@ def gen(rows): raise DatabaseErrorException(str(e)) # now use a generator for sending the rows and execute all the other queries - return p(gen(r)) + return p(filter_fields(gen(r))) @bp.route("/meta", methods=("GET", "POST"))