models: versioning support

* Uses SQLAlchemy-Continuum to provide full versioning support.

Signed-off-by: Jiri Kuncar <jiri.kuncar@cern.ch>

invenio_records/models.py

@@ -24,10 +24,22 @@
 
 """Record models."""
 
+import pkg_resources
+
 from invenio_db import db
+from sqlalchemy_continuum import make_versioned
 from sqlalchemy_utils.models import Timestamp
 from sqlalchemy_utils.types import JSONType
 
+try:
+    pkg_resources.get_distribution('invenio_accounts')
+except pkg_resources.DistributionNotFound:
+    user_cls = None
+else:
+    from invenio_accounts.models import User as user_cls
+
+make_versioned(user_cls=user_cls)
+
 
 class Record(db.Model, Timestamp):
     """Represent a record metadata inside the SQL database.
@@ -36,6 +48,9 @@ class Record(db.Model, Timestamp):
     with automatically managed timestamps.
     """
 
+    # Enables SQLAlchemy-Continuum versioning
+    __versioned__ = {}
+
     __tablename__ = 'record'
 
     id = db.Column(

setup.py

@@ -37,7 +37,7 @@
 tests_require = [
     'check-manifest>=0.25',
     'coverage>=4.0',
-    'invenio-db[all]>=1.0.0a3',
+    'invenio-db[all]>=1.0.0a4',
     'isort>=4.2.2',
     'pep257>=0.7.0',
     'pytest-cache>=1.0',
@@ -64,9 +64,10 @@
 install_requires = [
     'blinker>=1.4',
     'flask-celeryext>=0.1.0',
-    'invenio-db>=1.0.0a3',
+    'invenio-db>=1.0.0a4',
     'jsonpatch>=1.11',
     'jsonschema>=2.5.1',
+    'sqlalchemy-continuum>=1.2.1',
     'sqlalchemy-utils>=0.31.0',
 ]
 

tests/test_invenio_records.py

@@ -78,7 +78,7 @@ def test_db():
     with app.app_context():
         create_database(db.engine.url)
         db.create_all()
-        assert len(db.metadata.tables) == 1
+        assert len(db.metadata.tables) == 3
 
     data = {'recid': 1, 'title': 'Test'}
     from invenio_records.models import Record as RM
@@ -150,7 +150,7 @@ def test_cli():
     runner = CliRunner()
     script_info = ScriptInfo(create_app=lambda info: app)
 
-    assert len(db.metadata.tables) == 1
+    assert len(db.metadata.tables) == 3
 
     # Test merging a base another file.
     with runner.isolated_filesystem():