From d9a80bb6ed4929fc0d5ec17f6d7796e22efdde25 Mon Sep 17 00:00:00 2001
From: Ford Hurley <ford.hurley@gmail.com>
Date: Mon, 7 Apr 2014 08:06:44 -0400
Subject: [PATCH] Add option to show PyPI downloads from last_week and last_day

In addition to /pypi/dm, users can use /pypi/dw and /pypi/dd for more
detailed download counts.
---
 server.js | 26 +++++++++++++++++++-------
 1 file changed, 19 insertions(+), 7 deletions(-)

diff --git a/server.js b/server.js
index fa05d7f..7146f95 100644
--- a/server.js
+++ b/server.js
@@ -404,17 +404,29 @@ cache(function(data, match, sendBadge) {
     }
     try {
       var data = JSON.parse(buffer);
-      if (info === 'dm') {
+      if (info.charAt(0) === 'd') {
         badgeData.text[0] = getLabel('downloads', data);
-        var monthly = data.info.downloads.last_month;
-        badgeData.text[1] = metric(monthly) + '/month';
-        if (monthly === 0) {
+        switch (info.charAt(1)) {
+          case 'm':
+            var downloads = data.info.downloads.last_month;
+            badgeData.text[1] = metric(downloads) + '/month';
+            break;
+          case 'w':
+            var downloads = data.info.downloads.last_week;
+            badgeData.text[1] = metric(downloads) + '/week';
+            break;
+          case 'd':
+            var downloads = data.info.downloads.last_day;
+            badgeData.text[1] = metric(downloads) + '/day';
+            break;
+        }
+        if (downloads === 0) {
           badgeData.colorscheme = 'red';
-        } else if (monthly < 10) {
+        } else if (downloads < 10) {
           badgeData.colorscheme = 'yellow';
-        } else if (monthly < 100) {
+        } else if (downloads < 100) {
           badgeData.colorscheme = 'yellowgreen';
-        } else if (monthly < 1000) {
+        } else if (downloads < 1000) {
           badgeData.colorscheme = 'green';
         } else {
           badgeData.colorscheme = 'brightgreen';