Add badge endpoint.

Author: ericholscher

readthedocs/core/views.py

@@ -88,6 +88,16 @@ def queue_info(request):
 
     return HttpResponse(resp)
 
+def badge(request, project_slug, version_slug):
+    version = get_object_or_404(Version, project__slug=project_slug,
+                                slug=version_slug)
+    last_build = version.builds.filter(type='html').order_by('-date')[0]
+
+    color = 'green'
+    if not last_build.success:
+        color = 'red'
+    return HttpResponseRedirect('http://img.shields.io/badge/Docs-%s-%s.svg' % (version.slug, color))
+
 def live_builds(request):
     builds = Build.objects.filter(state='building')[:5]
     WEBSOCKET_HOST = getattr(settings, 'WEBSOCKET_HOST', 'localhost:8088')

readthedocs/urls.py

@@ -95,6 +95,9 @@
     url(r'^wipe/(?P<project_slug>[-\w]+)/(?P<version_slug>[\w]{1}[-\w\.]+)/$',
         'core.views.wipe_version',
         name='wipe_version'),
+    url(r'^badge/(?P<project_slug>[-\w]+)/(?P<version_slug>[\w]{1}[-\w\.]+)/$',
+        'core.views.badge',
+        name='badge'),
     url(r'^profiles/create/', 'profiles.views.create_profile',
         {'form_class': UserProfileForm},
         name='profiles_profile_create'),