From 5a77d14a261dc770a4f13fc3e632065ca8ec48b1 Mon Sep 17 00:00:00 2001
From: Lance Edgar <lance@edbob.org>
Date: Sun, 18 Jun 2023 11:38:14 -0500
Subject: [PATCH] Add models for StockPurchase and EquityLiveBalance

---
 corepos/db/office_trans/model.py | 35 ++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/corepos/db/office_trans/model.py b/corepos/db/office_trans/model.py
index 20c244c..d484b4b 100644
--- a/corepos/db/office_trans/model.py
+++ b/corepos/db/office_trans/model.py
@@ -31,6 +31,41 @@ from sqlalchemy import orm
 Base = orm.declarative_base()
 
 
+# TODO: not sure what primary key should be for this?  am trying a
+# composite one so far, we'll see...
+class StockPurchase(Base):
+    """
+    Represents a member equity payment.
+    """
+    __tablename__ = 'stockpurchases'
+
+    card_number = sa.Column('card_no', sa.Integer(), nullable=False, primary_key=True, autoincrement=False)
+
+    amount = sa.Column('stockPurchase', sa.Numeric(precision=10, scale=2), nullable=True)
+
+    datetime = sa.Column('tdate', sa.DateTime(), nullable=True, primary_key=True)
+
+    transaction_number = sa.Column('trans_num', sa.String(length=50), nullable=True, primary_key=True)
+
+    transaction_id = sa.Column('trans_id', sa.Integer(), nullable=True)
+
+    department_number = sa.Column('dept', sa.Integer(), nullable=True)
+
+    def __str__(self):
+        return f"#{self.card_number} for ${self.amount}"
+
+
+class EquityLiveBalance(Base):
+
+    __tablename__ = 'equity_live_balance'
+
+    member_number = sa.Column('memnum', sa.Integer(), nullable=False, primary_key=True, autoincrement=False)
+
+    payments = sa.Column(sa.Numeric(precision=10, scale=2), nullable=True)
+
+    start_date = sa.Column('startdate', sa.DateTime(), nullable=True)
+
+
 class TransactionDetailBase(object):
     """
     Represents a POS transaction detail record.