From 50351596acaced2b5b14fd4ad4f3488f9d09b111 Mon Sep 17 00:00:00 2001
From: Lance Edgar <lance@wuttaproject.org>
Date: Fri, 24 Jan 2025 20:18:13 -0600
Subject: [PATCH] fix: add model for lane_trans `LocalTrans`

---
 corepos/db/lane_trans/model.py | 31 ++++++++++++++++++++++++-------
 1 file changed, 24 insertions(+), 7 deletions(-)

diff --git a/corepos/db/lane_trans/model.py b/corepos/db/lane_trans/model.py
index b656d8f..f2245f5 100644
--- a/corepos/db/lane_trans/model.py
+++ b/corepos/db/lane_trans/model.py
@@ -26,6 +26,7 @@ Data model for CORE POS "lane_trans" DB
 
 import sqlalchemy as sa
 from sqlalchemy import orm
+from sqlalchemy.ext.declarative import declared_attr
 
 from corepos.db.common import trans as common
 
@@ -45,18 +46,34 @@ class DTransactionBase(common.TransactionDetailBase):
 
 class DTransaction(DTransactionBase, Base):
     """
-    Represents a record from ``dtransactions`` table.
+    Data model for ``dtransactions`` table.
     """
     __tablename__ = 'dtransactions'
 
 
-class LocalTempTrans(common.TransactionDetailBase, Base):
+class LocalTransBase(common.TransactionDetailBase):
     """
-    Represents a record from ``localtemptrans`` table.
+    Base class for ``localtrans`` and similar models.
     """
-    __tablename__ = 'localtemptrans'
-    __table_args__ = (
-        sa.PrimaryKeyConstraint('trans_id'),
-    )
+
+    @declared_attr
+    def __table_args__(self):
+        return (
+            sa.PrimaryKeyConstraint('trans_id'),
+        )
 
     date_time = sa.Column('datetime', sa.DateTime(), nullable=True)
+
+
+class LocalTrans(LocalTransBase, Base):
+    """
+    Data model for ``localtrans`` table.
+    """
+    __tablename__ = 'localtrans'
+
+
+class LocalTempTrans(LocalTransBase, Base):
+    """
+    Data model for ``localtemptrans`` table.
+    """
+    __tablename__ = 'localtemptrans'