Skip to content
This repository was archived by the owner on Jan 18, 2023. It is now read-only.

Commit 3b1893a

Browse files
committed
Fix serving not being exported to the CSV
1 parent c3f0247 commit 3b1893a

File tree

1 file changed

+21
-2
lines changed

1 file changed

+21
-2
lines changed

firststreet/api/csv_format.py

Lines changed: 21 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -186,10 +186,29 @@ def format_adaptation_detail(data):
186186
df['adaptationId'] = df['adaptationId'].apply(str)
187187
df['returnPeriod'] = df['returnPeriod'].astype('Int64').apply(str)
188188
df['geometry'] = df['geometry'].apply(get_geom_center)
189+
if "serving" in df and df.serving.isnull().all():
190+
df["serving_property"] = pd.NA
191+
df["serving_neighborhood"] = pd.NA
192+
df["serving_zcta"] = pd.NA
193+
df["serving_tract"] = pd.NA
194+
df["serving_city"] = pd.NA
195+
df["serving_county"] = pd.NA
196+
df["serving_cd"] = pd.NA
197+
df["serving_state"] = pd.NA
198+
else:
199+
df = df.rename(columns={"serving.property": "serving_property",
200+
"serving.neighborhood": "serving_neighborhood",
201+
"serving.zcta": "serving_zcta",
202+
"serving.tract": "serving_tract",
203+
"serving.city": "serving_city",
204+
"serving.county": "serving_county",
205+
"serving.cd": "serving_cd",
206+
"serving.state": "serving_state"})
189207
df = pd.concat([df.drop(['geometry'], axis=1), df['geometry'].apply(pd.Series)], axis=1)
190208

191-
return df[['adaptationId', 'valid_id', 'name', 'type', 'scenario', 'conveyance', 'returnPeriod',
192-
'latitude', 'longitude']]
209+
return df[['adaptationId', 'valid_id', 'name', 'type', 'scenario', 'conveyance', 'returnPeriod', 'serving_property',
210+
'serving_neighborhood', 'serving_zcta', 'serving_tract', 'serving_city', 'serving_county', 'serving_cd',
211+
'serving_state', 'latitude', 'longitude']]
193212

194213

195214
def format_adaptation_summary(data):

0 commit comments

Comments
 (0)