From 9a3a2cb975ab171c4f23c5bf5e938db940c707d4 Mon Sep 17 00:00:00 2001
From: JF Bastien <jfb@chromium.org>
Date: Wed, 13 Jan 2016 13:59:28 -0800
Subject: update.py: support json output

I'm updating the waterfall separately.
---
 update.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'update.py')

diff --git a/update.py b/update.py
index 99c142d79..dc30e6cae 100755
--- a/update.py
+++ b/update.py
@@ -16,6 +16,7 @@
 
 import filecmp
 import glob
+import json
 import os
 import shutil
 import subprocess
@@ -33,7 +34,15 @@ TORTURE_DIR = os.path.join(BASE_DIR, 'torture-s')
 
 def download_revision(force_latest):
   name = 'latest' if force_latest else 'lkgr'
-  return urllib2.urlopen(STORAGE_BASE + name).read().strip()
+  downloaded = urllib2.urlopen(STORAGE_BASE + name).read().strip()
+  # TODO: for now try opening as JSON, if that doesn't work then the content is
+  #       just a hash. The waterfall is in the process of migrating to JSON.
+  info = None
+  try:
+    info = json.loads(downloaded)
+  except:
+    pass
+  return info['build'] if type(info) == dict else downloaded
 
 
 def write_revision(revision):
-- 
cgit v1.2.3