From c9db61d97c417d8d05f780036f9d683e754c0274 Mon Sep 17 00:00:00 2001
From: tm689 <tm689@cam.ac.uk>
Date: Tue, 21 May 2024 11:45:05 +0100
Subject: [PATCH] feat: picking up the extra SurveySite column

---
 ews/coordinator/survey_servers/processor_surveys_wrt.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ews/coordinator/survey_servers/processor_surveys_wrt.py b/ews/coordinator/survey_servers/processor_surveys_wrt.py
index cbf553b..0992e0f 100644
--- a/ews/coordinator/survey_servers/processor_surveys_wrt.py
+++ b/ews/coordinator/survey_servers/processor_surveys_wrt.py
@@ -107,6 +107,7 @@ def nested_to_flattened(df):
                 "CollectionDate",
                 "UpdateDate",
                 "GrowthStage",
+                "SurveySite",
                 "listDisease"]
         for i in RAW_COLUMNS:
             df[i] = ""
@@ -215,6 +216,7 @@ def get_WRT_form_as_csv(form_credentials: dict, jobPath: str, config: dict, stat
         "CollectionDate" : ('parse_date',(('name_out','survey_infromation-survey_date'),('fmt_in','%m/%d/%Y'))),
         "UpdateDate" : 'None',
         "GrowthStage" : 'site_information-growth_stage',
+        "SurveySite" : 'site_information-survey_site',
         "listDisease" : 'None',
         "YR.Severity" : ('parse_cases',(('name_out','yellow_rust-yellowrust_severity'),('cases', cases_severity),('dtype', int))),
         "YR.Incident" : ('parse_cases',(('name_out','yellow_rust-yellowrust_incidence'),('cases', cases_incident))),
-- 
GitLab