summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorClaude Paroz <claude@2xlibre.net>2015-02-27 18:19:56 +0100
committerClaude Paroz <claude@2xlibre.net>2015-02-27 20:23:07 +0100
commit5cf96b49e43daea6d4a0ba1c80c45e90c74f4e47 (patch)
treed5c92d3a8eb9247905a67ae04eb1ec5195218049
parent2b19b3a031e7d5b5264bc9e880d0192301289bc2 (diff)
Fixed #24418 -- Prevented crash in refresh_from_db with null fk
Thanks Johannes Lerch for the report, Tim Graham for the test case, and Simon Charette for the review.
-rw-r--r--django/db/models/base.py2
-rw-r--r--tests/basic/tests.py7
2 files changed, 8 insertions, 1 deletions
diff --git a/django/db/models/base.py b/django/db/models/base.py
index a505efa87d..cb1d37f9df 100644
--- a/django/db/models/base.py
+++ b/django/db/models/base.py
@@ -596,7 +596,7 @@ class Model(six.with_metaclass(ModelBase)):
if field.rel and field.get_cache_name() in self.__dict__:
rel_instance = getattr(self, field.get_cache_name())
local_val = getattr(db_instance, field.attname)
- related_val = getattr(rel_instance, field.related_field.attname)
+ related_val = None if rel_instance is None else getattr(rel_instance, field.related_field.attname)
if local_val != related_val:
del self.__dict__[field.get_cache_name()]
self._state.db = db_instance._state.db
diff --git a/tests/basic/tests.py b/tests/basic/tests.py
index d851d858de..934c5587f4 100644
--- a/tests/basic/tests.py
+++ b/tests/basic/tests.py
@@ -735,6 +735,13 @@ class ModelRefreshTests(TestCase):
self.assertFalse(hasattr(s3_copy.selfref, 'touched'))
self.assertEqual(s3_copy.selfref, s2)
+ def test_refresh_null_fk(self):
+ s1 = SelfRef.objects.create()
+ s2 = SelfRef.objects.create(selfref=s1)
+ s2.selfref = None
+ s2.refresh_from_db()
+ self.assertEqual(s2.selfref, s1)
+
def test_refresh_unsaved(self):
pub_date = self._truncate_ms(datetime.now())
a = Article.objects.create(pub_date=pub_date)