From e7c687f6eb524f1b293552b8033e9b2a8f19cd50 Mon Sep 17 00:00:00 2001
From: Graham Gilbert <graham@grahamgilbert.com>
Date: Tue, 18 Aug 2015 09:38:36 +0100
Subject: [PATCH] Move back to bz2

---
 server/models.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/server/models.py b/server/models.py
index 40a3ccd0..db7d89c5 100644
--- a/server/models.py
+++ b/server/models.py
@@ -6,7 +6,6 @@
 from xml.parsers.expat import ExpatError
 import base64
 import bz2
-import zlib
 from datetime import datetime
 
 
@@ -97,7 +96,7 @@ class Machine(models.Model):
 
     def encode(self, plist):
         string = plistlib.writePlistToString(plist)
-        bz2data = zlib.compress(string)
+        bz2data = bz2.compress(string)
         b64data = base64.b64encode(bz2data)
         return b64data
 
@@ -120,7 +119,7 @@ def decode(self, data):
     def b64bz_decode(self, data):
         try:
             bz2data = base64.b64decode(data)
-            string = zlib.decompress(bz2data)
+            string = bz2.decompress(bz2data)
             plist = plistlib.readPlistFromString(string)
             return plist
         except Exception: