From 8c7acdad78e28a14a451c241d2babf85bdaa9167 Mon Sep 17 00:00:00 2001
From: Thaddee Tyl <thaddee.tyl@gmail.com>
Date: Tue, 16 Jun 2015 08:51:14 +0200
Subject: [PATCH] PyPI: format badge

Part of #465
---
 server.js | 27 +++++++++++++++++++++++++++
 try.html  |  4 ++++
 2 files changed, 31 insertions(+)

diff --git a/server.js b/server.js
index fabee5f4a0778..c35fa3a53631c 100644
--- a/server.js
+++ b/server.js
@@ -1544,6 +1544,33 @@ cache(function(data, match, sendBadge, request) {
         badgeData.text[1] = hasWheel ? 'yes' : 'no';
         badgeData.colorscheme = hasWheel ? 'brightgreen' : 'red';
         sendBadge(format, badgeData);
+      } else if (info === 'format') {
+        var releases = data.releases[data.info.version];
+        var hasWheel = false;
+        var hasEgg = false;
+        for (var i = 0; i < releases.length; i++) {
+          if (releases[i].packagetype === 'wheel' ||
+              releases[i].packagetype === 'bdist_wheel') {
+            hasWheel = true;
+            break;
+          }
+          if (releases[i].packagetype === 'egg' ||
+              releases[i].packagetype === 'bdist_egg') {
+            hasEgg = true;
+          }
+        }
+        badgeData.text[0] = 'format';
+        if (hasWheel) {
+          badgeData.text[1] = 'wheel';
+          badgeData.colorscheme = 'brightgreen';
+        } else if (hasEgg) {
+          badgeData.text[1] = 'egg';
+          badgeData.colorscheme = 'red';
+        } else {
+          badgeData.text[1] = 'source';
+          badgeData.colorscheme = 'yellow';
+        }
+        sendBadge(format, badgeData);
       } else if (info === 'pyversions') {
         var versions = [];
         var pattern = /^Programming Language \:\: Python \:\: (\d\.\d)$/;
diff --git a/try.html b/try.html
index de50dfe9779c1..e475f7983c29e 100644
--- a/try.html
+++ b/try.html
@@ -534,6 +534,10 @@ <h3 id="miscellaneous"> Miscellaneous </h3>
     <td><img src='/pypi/wheel/Django.svg' alt=''/></td>
     <td><code>https://img.shields.io/pypi/wheel/Django.svg</code></td>
   </tr>
+  <tr><th data-keywords='python'> PyPI: </th>
+    <td><img src='/pypi/format/Django.svg' alt=''/></td>
+    <td><code>https://img.shields.io/pypi/format/Django.svg</code></td>
+  </tr>
   <tr><th data-keywords='python'> PyPI: </th>
     <td><img src='/pypi/pyversions/Django.svg' alt=''/></td>
     <td><code>https://img.shields.io/pypi/pyversions/Django.svg</code></td>