Merge ccd213d into 866ef4b

simple_history/__init__.py

@@ -19,5 +19,6 @@ def register(model, app=None, manager_name='history'):
         records = models.HistoricalRecords()
         records.manager_name = manager_name
         records.module = app and ("%s.models" % app) or model.__module__
+        records.add_extra_methods(model)
         records.finalize(model)
         models.registered_models[model._meta.db_table] = model

simple_history/models.py

@@ -42,7 +42,9 @@ def contribute_to_class(self, cls, name):
         self.manager_name = name
         self.module = cls.__module__
         models.signals.class_prepared.connect(self.finalize, sender=cls)
+        self.add_extra_methods(cls)
 
+    def add_extra_methods(self, cls):
         def save_without_historical_record(self, *args, **kwargs):
             """
             Save model without saving a historical record

simple_history/tests/tests.py

@@ -112,6 +112,10 @@ def test_save_without_historical_record(self):
             'history_type': "~",
         })
 
+    def test_save_without_historical_record_for_registered_model(self):
+        registered_model = ExternalModel3.objects.create(name='registered model')
+        self.assertTrue(hasattr(registered_model, 'save_without_historical_record'))
+
     def test_save_raises_exception(self):
         anthony = Person(name='Anthony Gillard')
         with self.assertRaises(RuntimeError):