summaryrefslogtreecommitdiff
path: root/py-fake-service
diff options
context:
space:
mode:
authorAlexandre Pretto Nunes <anunes@thoughtworks.com>2014-08-18 13:26:32 -0300
committerAlexandre Pretto Nunes <anunes@thoughtworks.com>2014-08-18 13:26:32 -0300
commite2da0313caf715eef495ca3b14bd7f8af9735556 (patch)
tree081e76be01059e507284a53a768e1efd961aac14 /py-fake-service
parent051c2838a3e9f31a25359504d0cfc8aae07affa5 (diff)
Gunzip the mailset archive before untarring it.
Diffstat (limited to 'py-fake-service')
-rw-r--r--py-fake-service/app/pixelated_user_agent.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/py-fake-service/app/pixelated_user_agent.py b/py-fake-service/app/pixelated_user_agent.py
index e86e6760..3378bcca 100644
--- a/py-fake-service/app/pixelated_user_agent.py
+++ b/py-fake-service/app/pixelated_user_agent.py
@@ -113,6 +113,7 @@ def draft_reply_for(mail_id):
def load_mailset(mailset):
import os
from tarfile import TarFile
+ from gzip import GzipFile
mbox_root = os.path.join(os.environ['HOME'], 'mailsets')
if not os.path.isdir(os.path.join(mbox_root)):
os.mkdir(mbox_root)
@@ -125,7 +126,8 @@ def load_mailset(mailset):
mbox_archive = open(mbox_archive_path, 'w')
mbox_archive.write(response.content)
mbox_archive.close()
- tarfile = TarFile(name=mbox_archive_path)
+ gzippedfile = GzipFile(filename=mbox_archive_path)
+ tarfile = TarFile(fileobj=gzippedfile)
tarfile.extractall(path=mbox_root)
mail_service.load_mailset()