From 64564e228a7263313e16701e5415a24c9f1c98ee Mon Sep 17 00:00:00 2001
From: indivar <indimishra@gmail.com>
Date: Sun, 11 Sep 2022 11:28:08 +0530
Subject: [PATCH] add #286: add parent_table in utils

---
 sqlalchemy_continuum/utils.py    | 24 +++++++++++++++++++
 tests/utils/test_parent_table.py | 41 ++++++++++++++++++++++++++++++++
 2 files changed, 65 insertions(+)
 create mode 100644 tests/utils/test_parent_table.py

diff --git a/sqlalchemy_continuum/utils.py b/sqlalchemy_continuum/utils.py
index a87f58ab..96ec85a2 100644
--- a/sqlalchemy_continuum/utils.py
+++ b/sqlalchemy_continuum/utils.py
@@ -90,6 +90,30 @@ def parent_class(version_cls):
     return get_versioning_manager(version_cls).parent_class_map[version_cls]
 
 
+def parent_table(table):
+    versioning_manager = get_versioning_manager(table)
+    if table in versioning_manager.association_version_tables:
+        parent_table = next(
+            iter(
+                t
+                for t in versioning_manager.association_tables
+                if versioning_manager.options['table_name'] % t.name == table.name
+            ),
+            None,
+        )
+        return parent_table
+
+    parent_table = next(
+        iter(
+            parent_class(m).__table__
+            for m in versioning_manager.parent_class_map
+            if m.__table__ == table
+        ),
+        None,
+    )
+    return parent_table
+
+
 def transaction_class(cls):
     """
     Return the associated transaction class for given versioned SQLAlchemy
diff --git a/tests/utils/test_parent_table.py b/tests/utils/test_parent_table.py
new file mode 100644
index 00000000..de78a89d
--- /dev/null
+++ b/tests/utils/test_parent_table.py
@@ -0,0 +1,41 @@
+import copy
+import datetime
+import sqlalchemy as sa
+from sqlalchemy_continuum.utils import parent_table, version_table
+
+from tests import TestCase
+
+
+class TestParentTable(TestCase):
+
+    def create_models(self):
+        super().create_models()
+
+        article_author_table = sa.Table(
+            'article_author',
+            self.Model.metadata,
+            sa.Column('article_id', sa.Integer, sa.ForeignKey('article.id'), primary_key=True, nullable=False),
+            sa.Column('author_id', sa.Integer, sa.ForeignKey('author.id'), primary_key=True, nullable=False),
+            sa.Column('created_date', sa.DateTime, nullable=False, server_default=sa.func.current_timestamp(), default=datetime.datetime.utcnow),
+        )
+       
+        class Author(self.Model):
+            __tablename__ = 'author'
+            __versioned__ = {
+                'baseclass': (self.Model, )
+            }
+            id = sa.Column(sa.Integer, autoincrement=True, primary_key=True)
+            name = sa.Column(sa.Unicode(255))
+            articles = sa.orm.relationship('Article', secondary=article_author_table, backref='author')
+
+
+        self.Author = Author
+        self.article_author_table = article_author_table
+
+    def test_parent_table_from_version_table(self):
+        author_version_table = version_table(self.Author.__table__)
+        assert parent_table(author_version_table) == self.Author.__table__
+
+    def test_parent_table_from_association_table(self):
+        versioned_article_author_table = version_table(self.article_author_table)
+        assert parent_table(versioned_article_author_table) == self.article_author_table
\ No newline at end of file