update rows properly

1 files changed, 18 insertions, 2 deletions

diff --git a/scraper/s2-geocode-server.py b/scraper/s2-geocode-server.py
index 9d824443..ed7ab4e7 100644
--- a/scraper/s2-geocode-server.py
+++ b/scraper/s2-geocode-server.py
@@ -90,6 +90,12 @@ def add_address():
     'status': 'ok'
   })
 
+@app.route('/api/verifications', methods=['GET'])
+def list_verifications():
+  return jsonify({
+    'verifications': fetch_google_lookup('verifications', item_key='paper_id'),
+  })
+
 @app.route('/api/verify/<sha256>', methods=['GET'])
 def find_verification(sha256):
   worksheet = fetch_worksheet('verifications')
@@ -114,12 +120,22 @@ def find_verification(sha256):
     })
 
 @app.route('/api/verify/add', methods=['POST'])
-def add_verifications():
+def add_verification():
   form = request.get_json()
   print(form)
   keys = verifications_worksheet.row_values(1)
   row = [ form[key] if key in form else '' for key in keys ]
-  verifications_worksheet.append_row(row)
+  try:
+    cell = worksheet.find(form['paper_id'])
+  except:
+    cell = None
+
+  if cell:
+    for item, i in enumerate(row):
+      verifications_worksheet.update_cell(cell.row, i+1, item)
+  else:
+    verifications_worksheet.append_row(row)
+
   return jsonify({
     'status': 'ok'
   })