Also support db_column attribute

Author: arski

simple_history/models.py

@@ -148,6 +148,8 @@ def copy_fields(self, model):
                     field_arguments['to_field'] = old_field.to_fields[0]
                 elif django.get_version() < "1.6" and old_field.rel.field_name != 'id':
                     field_arguments['to_field'] = old_field.rel.field_name
+                if getattr(old_field, 'db_column', None):
+                    field_arguments['db_column'] = old_field.db_column
                 field = FieldType(
                     old_field.rel.to,
                     related_name='+',

simple_history/tests/models.py

@@ -259,3 +259,8 @@ class Country(models.Model):
 class Province(models.Model):
     country = models.ForeignKey(Country, to_field='code')
     history = HistoricalRecords()
+
+
+class City(models.Model):
+    country = models.ForeignKey(Country, db_column='countryCode')
+    history = HistoricalRecords()

simple_history/tests/tests/test_models.py

@@ -18,7 +18,8 @@
     AbstractBase, ConcreteAttr, ConcreteUtil, SelfFK, Temperature, WaterLevel,
     ExternalModel1, ExternalModel3, UnicodeVerboseName, HistoricalChoice,
     HistoricalState, HistoricalCustomFKError, Series, SeriesWork, PollInfo,
-    UserAccessorDefault, UserAccessorOverride, Employee, Country, Province
+    UserAccessorDefault, UserAccessorOverride, Employee, Country, Province,
+    City
 )
 from ..external.models import ExternalModel2, ExternalModel4
 
@@ -266,6 +267,12 @@ def test_to_field_foreign_key_save(self):
         self.assertEqual([c.country_id for c in province.history.all()],
                          [country2.code, country.code])
 
+    def test_db_column_foreign_key_save(self):
+        country = Country.objects.create(code='US')
+        city = City.objects.create(country=country)
+        country_field = City._meta.get_field('country')
+        self.assertTrue(getattr(country_field, 'db_column') in str(city.history.all().query))
+
     def test_raw_save(self):
         document = Document()
         document.save_base(raw=True)