Move back to bz2

salopensource · Aug 18, 2015 · e7c687f · e7c687f
1 parent 90ab6a3
commit e7c687f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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: