Merge pull request #1166 from andrewsg/images-blobstore

Add get_serving_url sample for images/blobstore

Author: andrewsg

appengine/standard/images/api/blobstore.py

@@ -48,5 +48,26 @@ def get(self):
 # [END thumbnailer]
 
 
-app = webapp2.WSGIApplication([('/img', Thumbnailer)], debug=True)
+class ServingUrlRedirect(webapp2.RequestHandler):
+    def get(self):
+        blob_key = self.request.get("blob_key")
+
+        if blob_key:
+            blob_info = blobstore.get(blob_key)
+
+            if blob_info:
+                # [START get_serving_url]
+                url = images.get_serving_url(
+                    blob_key, size=150, crop=True, secure_url=True)
+                # [END get_serving_url]
+                return webapp2.redirect(url)
+
+        # Either "blob_key" wasn't provided, or there was no value with that ID
+        # in the Blobstore.
+        self.error(404)
+
+
+app = webapp2.WSGIApplication(
+    [('/img', Thumbnailer),
+     ('/redirect', ServingUrlRedirect)], debug=True)
 # [END all]

appengine/standard/images/api/blobstore_test.py

@@ -44,3 +44,24 @@ def test_img_missing(app):
 def test_no_img_id(app):
     # No blob_key, should get error
     app.get('/img', status=404)
+
+
+def test_url_redirect(app):
+    with mock.patch('blobstore.images') as mock_images:
+        with mock.patch('blobstore.blobstore') as mock_blobstore:
+            mock_blobstore.get.return_value = b'123'
+            mock_images.get_serving_url.return_value = 'http://lh3.ggpht.com/X'
+
+            response = app.get('/redirect?blob_key=123')
+
+            assert response.status_int == 302
+
+
+def test_url_redirect_missing(app):
+    # Bogus blob_key, should get error
+    app.get('/redirect?blob_key=123', status=404)
+
+
+def test_url_redirect_no_key(app):
+    # No blob_key, should get error
+    app.get('/redirect', status=404)