Merge branch 'kyheo-feature/module-as-class-name'

sqlalchemy_continuum/manager.py

@@ -87,7 +87,8 @@ def __init__(
             'transaction_column_name': 'transaction_id',
             'end_transaction_column_name': 'end_transaction_id',
             'operation_type_column_name': 'operation_type',
-            'strategy': 'validity'
+            'strategy': 'validity',
+            'use_module_name': False
         }
         if plugins is None:
             self.plugins = []

sqlalchemy_continuum/model_builder.py

@@ -252,11 +252,13 @@ def mapper_args(cls):
 
         args.update(self.get_inherited_denormalized_columns(table))
 
-        return type(
-            '%sVersion' % self.model.__name__,
-            self.base_classes(),
-            args
-        )
+        if self.manager.options.get('use_module_name', True):
+            name = '%s%sVersion' % (
+                self.model.__module__.title().replace('.', ''),
+                self.model.__name__)
+        else:
+            name = '%sVersion' % (self.model.__name__,)
+        return type(name, self.base_classes(), args)
 
     def __call__(self, table, tx_class):
         """

tests/utils/test_version_class.py

@@ -1,5 +1,7 @@
 from pytest import raises
 from sqlalchemy_continuum import ClassNotVersioned, version_class
+from sqlalchemy_continuum.manager import VersioningManager
+from sqlalchemy_continuum.model_builder import ModelBuilder
 
 from tests import TestCase
 
@@ -12,3 +14,10 @@ def test_version_class_for_versioned_class(self):
     def test_throws_error_for_non_versioned_class(self):
         with raises(ClassNotVersioned):
             version_class('invalid')
+
+    def test_module_name_in_class_name(self):
+        options = {'use_module_name': True}
+        vm = VersioningManager(options=options)
+        mb = ModelBuilder(vm, self.Article)
+        ArticleVersion = mb.build_model(self.Article.__table__)
+        assert ArticleVersion.__name__ == 'TestsArticleVersion'