fix patch

1 files changed, 22 insertions, 1 deletions

diff --git a/scraper/s2-geocode-server.py b/scraper/s2-geocode-server.py
index ad3efd5d..4be59253 100644
--- a/scraper/s2-geocode-server.py
+++ b/scraper/s2-geocode-server.py
@@ -137,7 +137,28 @@ def find_verification(dataset, sha256):
 def add_verification():
   form = request.get_json()
   print(form)
-  update_or_append_worksheet('verifications', form)
+  # update_or_append_worksheet('verifications', form)
+
+  worksheet = fetch_worksheet('verifications')
+  keys = worksheet.row_values(1)
+  row = [ form[key] if key in form else '' for key in keys ]
+  try:
+    cells = worksheet.findall(form['paper_id'])
+  except:
+    cells = []
+  if len(cells):
+    for cell in cells:
+      values_list = worksheet.row_values(cell.row)
+      lookup = {}
+      for key, value in zip(keys, values_list):
+        lookup[key] = value
+      if lookup['dataset'] == dataset:
+        for i, item in enumerate(row):
+          worksheet.update_cell(cell.row, i+1, item)
+        break
+  else:
+    worksheet.append_row(row)
+
   if form['isUnknown']:
     update_or_append_worksheet('paper_locations', form)
   return jsonify({