Fixed #9723 -- Not all Python distributions contain the bz2 module, so we need to allow for that. Based on a patch from AdamG.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@9537 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
Malcolm Tredinnick 2008-12-01 02:20:59 +00:00
parent 8e68fc6cd1
commit 153b5a4a3a
1 changed files with 8 additions and 1 deletions

View File

@ -10,6 +10,12 @@ try:
except NameError:
from sets import Set as set # Python 2.3 fallback
try:
import bz2
has_bz2 = True
except ImportError:
has_bz2 = False
class Command(BaseCommand):
help = 'Installs the named fixture(s) in the database.'
args = "fixture [fixture ...]"
@ -62,10 +68,11 @@ class Command(BaseCommand):
compression_types = {
None: file,
'bz2': bz2.BZ2File,
'gz': gzip.GzipFile,
'zip': SingleZipReader
}
if has_bz2:
compression_types['bz2'] = bz2.BZ2File
app_fixtures = [os.path.join(os.path.dirname(app.__file__), 'fixtures') for app in get_apps()]
for fixture_label in fixture_labels: