Merge branch 'master' of github.com:adamhrv/megapixels_dev

1 files changed, 14 insertions, 14 deletions

diff --git a/scraper/s2-geocode-server.py b/scraper/s2-geocode-server.py
index 67da62d7..3aeda881 100644
--- a/scraper/s2-geocode-server.py
+++ b/scraper/s2-geocode-server.py
@@ -95,6 +95,17 @@ def list_verifications():
     'verifications': fetch_google_lookup('verifications', item_key='paper_id'),
   })
 
+@app.route('/api/verifications/<dataset>', methods=['GET'])
+def list_dataset_verifications(dataset):
+  rows = fetch_google_sheet_objects('verifications')
+  verifications = {}
+  for row in rows:
+    if row['dataset'] == dataset:
+      verifications[row['paper_id']] = row
+  return jsonify({
+    dataset: verifications,
+  })
+
 @app.route('/api/verify/<sha256>', methods=['GET'])
 def find_verification(sha256):
   worksheet = fetch_worksheet('verifications')
@@ -120,22 +131,11 @@ def find_verification(sha256):
 
 @app.route('/api/verify/add', methods=['POST'])
 def add_verification():
-  worksheet = fetch_worksheet('verifications')
   form = request.get_json()
   print(form)
-  keys = worksheet.row_values(1)
-  row = [ form[key] if key in form else '' for key in keys ]
-  try:
-    cell = worksheet.find(form['paper_id'])
-  except:
-    cell = None
-
-  if cell:
-    for item, i in enumerate(row):
-      worksheet.update_cell(cell.row, i+1, item)
-  else:
-    worksheet.append_row(row)
-
+  update_or_append_worksheet('verifications', form)
+  if form['isUnknown']:
+    update_or_append_worksheet('paper_locations', form)
   return jsonify({
     'status': 'ok'
   })