Author: russellm
Date: 2008-11-06 05:19:13 -0600 (Thu, 06 Nov 2008)
New Revision: 9357
Modified:
django/trunk/django/core/management/commands/loaddata.py
django/trunk/tests/regressiontests/fixtures_regress/models.py
Log:
Fixed #9011 -- Corrected handling of fixture files that contain errors to
correctly report the broken fixture name. Thanks to [EMAIL PROTECTED] for the
report and initial patch.
Modified: django/trunk/django/core/management/commands/loaddata.py
===================================================================
--- django/trunk/django/core/management/commands/loaddata.py 2008-11-06
10:48:20 UTC (rev 9356)
+++ django/trunk/django/core/management/commands/loaddata.py 2008-11-06
11:19:13 UTC (rev 9357)
@@ -35,7 +35,6 @@
# Keep a count of the installed objects and fixtures
fixture_count = 0
object_count = 0
- objects_per_fixture = []
models = set()
humanize = lambda dirname: dirname and "'%s'" % dirname or 'absolute
path'
@@ -103,17 +102,17 @@
return
else:
fixture_count += 1
- objects_per_fixture.append(0)
+ objects_in_fixture = 0
if verbosity > 0:
print "Installing %s fixture '%s' from %s." % \
(format, fixture_name,
humanize(fixture_dir))
try:
objects = serializers.deserialize(format,
fixture)
for obj in objects:
- object_count += 1
- objects_per_fixture[-1] += 1
+ objects_in_fixture += 1
models.add(obj.object.__class__)
obj.save()
+ object_count += objects_in_fixture
label_found = True
except (SystemExit, KeyboardInterrupt):
raise
@@ -131,22 +130,21 @@
(full_path,
traceback.format_exc())))
return
fixture.close()
+
+ # If the fixture we loaded contains 0 objects,
assume that an
+ # error was encountered during fixture loading.
+ if objects_in_fixture == 0:
+ sys.stderr.write(
+ self.style.ERROR("No fixture data found
for '%s'. (File format may be invalid.)" %
+ (fixture_name)))
+ transaction.rollback()
+ transaction.leave_transaction_management()
+ return
except:
if verbosity > 1:
print "No %s fixture '%s' in %s." % \
(format, fixture_name, humanize(fixture_dir))
-
- # If any of the fixtures we loaded contain 0 objects, assume that an
- # error was encountered during fixture loading.
- if 0 in objects_per_fixture:
- sys.stderr.write(
- self.style.ERROR("No fixture data found for '%s'. (File format
may be invalid.)" %
- (fixture_name)))
- transaction.rollback()
- transaction.leave_transaction_management()
- return
-
# If we found even one object in a fixture, we need to reset the
# database sequences.
if object_count > 0:
Modified: django/trunk/tests/regressiontests/fixtures_regress/models.py
===================================================================
--- django/trunk/tests/regressiontests/fixtures_regress/models.py
2008-11-06 10:48:20 UTC (rev 9356)
+++ django/trunk/tests/regressiontests/fixtures_regress/models.py
2008-11-06 11:19:13 UTC (rev 9357)
@@ -7,7 +7,7 @@
name = models.CharField(max_length=150)
latin_name = models.CharField(max_length=150)
count = models.IntegerField()
-
+
def __unicode__(self):
return self.common_name
@@ -56,7 +56,7 @@
class Article(models.Model):
title = models.CharField(max_length=255)
channels = models.ManyToManyField(Channel)
-
+
class Meta:
ordering = ('id',)
@@ -113,6 +113,15 @@
>>> management.call_command('loaddata', 'bad_fixture2', verbosity=0)
No fixture data found for 'bad_fixture2'. (File format may be invalid.)
+# Loading a fixture file with no data returns an error
+>>> management.call_command('loaddata', 'empty', verbosity=0)
+No fixture data found for 'empty'. (File format may be invalid.)
+
+# If any of the fixtures contain an error, loading is aborted
+# (Regression for #9011 - error message is correct)
+>>> management.call_command('loaddata', 'bad_fixture2', 'animal', verbosity=0)
+No fixture data found for 'bad_fixture2'. (File format may be invalid.)
+
>>> sys.stderr = savestderr
###############################################
@@ -123,7 +132,7 @@
>>> management.call_command('loaddata', 'model-inheritance.json', verbosity=0)
###############################################
-# Test for ticket #7572 -- MySQL has a problem if the same connection is
+# Test for ticket #7572 -- MySQL has a problem if the same connection is
# used to create tables, load data, and then query over that data.
# To compensate, we close the connection after running loaddata.
# This ensures that a new connection is opened when test queries are issued.
--~--~---------~--~----~------------~-------~--~----~
You received this message because you are subscribed to the Google Groups
"Django updates" group.
To post to this group, send email to [email protected]
To unsubscribe from this group, send email to [EMAIL PROTECTED]
For more options, visit this group at
http://groups.google.com/group/django-updates?hl=en
-~----------~----~----~----~------~----~------~--~---
