Skip to content

Commit d14e42a

Browse files
committed
fix: don't return missing columns if they don't exist
1 parent 3730743 commit d14e42a

File tree

1 file changed

+5
-8
lines changed

1 file changed

+5
-8
lines changed

src/server/endpoints/covidcast.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,11 @@ def handle():
136136

137137
fields_string = ["geo_value", "signal"]
138138
fields_int = ["time_value", "direction", "issue", "lag"]
139+
140+
missing_fields = ["missing_value", "missing_stderr", "missing_sample_size"]
141+
if sql_table_has_columns("covidcast", missing_fields):
142+
fields_int.extend(missing_fields)
143+
139144
fields_float = ["value", "stderr", "sample_size"]
140145
if is_compatibility_mode():
141146
q.set_order("signal", "time_value", "geo_value", "issue")
@@ -145,14 +150,6 @@ def handle():
145150
q.set_order("source", "signal", "time_type", "time_value", "geo_type", "geo_value", "issue")
146151
q.set_fields(fields_string, fields_int, fields_float)
147152

148-
missing_fields = ["missing_value", "missing_stderr", "missing_sample_size"]
149-
fields_int.extend(missing_fields)
150-
if sql_table_has_columns("covidcast", missing_fields):
151-
# real fields
152-
q.fields.extend([f"{q.alias}.{field}" for field in missing_fields])
153-
else:
154-
# fake fields
155-
q.fields.extend([f"0 as {field}" for field in missing_fields])
156153

157154
# basic query info
158155
# data type of each field

0 commit comments

Comments
 (0)