Update AssessmentValue global API

hawc/apps/assessment/exports.py

@@ -41,6 +41,9 @@ def get_annotation_map(self, query_prefix: str) -> dict:
             ),
         }
 
+    def prepare_df(self, df):
+        return self.format_time(df)
+
 
 class AssessmentExport(ModelExport):
     def get_value_map(self) -> dict:
@@ -53,6 +56,9 @@ def get_value_map(self) -> dict:
             "last_updated": "last_updated",
         }
 
+    def prepare_df(self, df):
+        return self.format_time(df)
+
 
 class AssessmentDetailExport(ModelExport):
     def get_value_map(self) -> dict:
@@ -70,6 +76,9 @@ def get_value_map(self) -> dict:
             "last_updated": "last_updated",
         }
 
+    def prepare_df(self, df):
+        return self.format_time(df)
+
     def get_annotation_map(self, query_prefix: str) -> dict:
         return {
             "project_status_display": sql_display(

hawc/apps/common/exports.py

@@ -152,7 +152,7 @@ def format_time(self, df: pd.DataFrame) -> pd.DataFrame:
         tz = timezone.get_default_timezone()
         for key in [self.get_column_name("created"), self.get_column_name("last_updated")]:
             if key in df.columns:
-                df.loc[:, key] = df[key].dt.tz_convert(tz).dt.strftime("%Y-%m-%dT%H:%M:%S.%f%z")
+                df[key] = df[key].dt.tz_convert(tz).dt.strftime("%Y-%m-%dT%H:%M:%S.%f%z")
         return df
 
     def get_df(self, qs: QuerySet) -> pd.DataFrame:

tests/hawc/apps/hawc_admin/test_api.py

@@ -80,4 +80,4 @@ def test_assessment_values(self):
         resp = client.get(url)
         assert resp.status_code == 200
         df = pd.read_json(StringIO(resp.content.decode()))
-        assert df.shape == (3, 33)
+        assert df.shape == (3, 41)