Fix bug with related

Author: macro1

CHANGES.rst

@@ -5,6 +5,7 @@ tip (unreleased)
 ----------------
 - Extended availability of the ``as_of`` method to models as well as instances.
 - Allow ``history_user`` on historical objects to be set by middleware.
+- Fixed error that occurs when a foreign key is designated using just the name of the model.
 
 1.4.0 (2014-06-29)
 ------------------

simple_history/models.py

@@ -8,6 +8,7 @@
     apps = None
 from django.db import models
 from django.db.models.fields.related import RelatedField
+from django.db.models.related import RelatedObject
 from django.conf import settings
 from django.contrib import admin
 from django.utils import importlib
@@ -184,6 +185,7 @@ def get_instance(self):
             )),
             'history_object': HistoricalObjectDescriptor(model),
             'instance': property(get_instance),
+            'instance_type': model,
             'revert_url': revert_url,
             '__str__': lambda self: '%s as of %s' % (self.history_object,
                                                      self.history_date)
@@ -294,7 +296,8 @@ def get_field(self, other, cls):
 
     def do_related_class(self, other, cls):
         field = self.get_field(other, cls)
-
+        if not hasattr(self, 'related'):
+            self.related = RelatedObject(other, cls.instance_type, self)
         transform_field(field)
         field.rel = None
 

simple_history/tests/tests/test_models.py

@@ -17,7 +17,8 @@
     AdminProfile, Bookcase, MultiOneToOne, Poll, Choice, Restaurant, Person,
     FileModel, Document, Book, HistoricalPoll, Library, State, AbstractBase,
     ConcreteAttr, ConcreteUtil, SelfFK, Temperature, WaterLevel,
-    ExternalModel1, ExternalModel3, UnicodeVerboseName
+    ExternalModel1, ExternalModel3, UnicodeVerboseName, HistoricalChoice,
+    HistoricalState
 )
 from ..external.models import ExternalModel2, ExternalModel4
 
@@ -460,3 +461,11 @@ def test_invalid_bases(self):
         invalid_bases = (AbstractBase, "InvalidBases")
         for bases in invalid_bases:
             self.assertRaises(TypeError, HistoricalRecords, bases=bases)
+
+    def test_import_related(self):
+        field_object = HistoricalChoice._meta.get_field_by_name('poll_id')[0]
+        self.assertEqual(field_object.related.model, Choice)
+
+    def test_string_related(self):
+        field_object = HistoricalState._meta.get_field_by_name('library_id')[0]
+        self.assertEqual(field_object.related.model, State)