From 31ae5eacd57669126627232a81a0de6bf0d2e662 Mon Sep 17 00:00:00 2001
From: Lance Edgar <lance@edbob.org>
Date: Fri, 2 Nov 2018 21:43:56 -0500
Subject: [PATCH] Tweak status code rendering for upgrades API view

---
 tailbone/api/upgrades.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/tailbone/api/upgrades.py b/tailbone/api/upgrades.py
index 28bea59c..87eda224 100644
--- a/tailbone/api/upgrades.py
+++ b/tailbone/api/upgrades.py
@@ -38,15 +38,19 @@ class UpgradeView(APIMasterView):
     model_class = model.Upgrade
 
     def normalize(self, upgrade):
-        return {
+        data = {
             'created': upgrade.created.isoformat(),
             'description': upgrade.description,
             'enabled': upgrade.enabled,
-            'status_code': self.enum.UPGRADE_STATUS.get(upgrade.status_code,
-                                                        six.text_type(upgrade.status_code)),
             'executed': upgrade.executed.isoformat() if upgrade.executed else None,
             # 'executed_by': 
         }
+        if upgrade.status_code is None:
+            data['status_code'] = None
+        else:
+            data['status_code'] = self.enum.UPGRADE_STATUS.get(upgrade.status_code,
+                                                               six.text_type(upgrade.status_code))
+        return data
 
 
 def includeme(config):