mercore starter

app/models/transaction.py

@@ -0,0 +1,57 @@
+from datetime import datetime, timezone
+from app.extensions import db
+from app.models import account
+from sqlalchemy.exc import IntegrityError
+from sqlalchemy import and_, or_, not_
+from sqlalchemy.sql import func
+
+class Transaction(db.Model):
+    __tablename__ = 'transactions'
+    id = db.Column(
+        db.Integer,
+        primary_key=True,
+        autoincrement=True,
+    )
+    transaction_id = db.Column(db.String(50), nullable=False)
+    account_id = db.Column(db.String(50), nullable=True)
+    customer_id = db.Column(db.String(50), nullable=True)
+    type = db.Column(db.String(50), nullable=False)
+    channel = db.Column(db.String(50), nullable=False)
+    phone_number = db.Column(db.String(50), nullable=True)
+    created_at = db.Column(db.DateTime(timezone=True), server_default=func.now())
+    updated_at = db.Column(db.DateTime(timezone=True), server_default=func.now(), onupdate=func.now())
+    
+    def __repr__(self):
+        return f'<Transaction {self.id}>'
+
+    @classmethod
+    def create_transaction(cls, transaction_id, account_id, customer_id, type, channel):
+
+        # if cls.query.filter_by(transaction_id=transaction_id).first():
+        #     raise ValueError("Duplicate Transaction")
+
+        if cls.query.filter( and_( cls.transaction_id ==transaction_id, cls.type==type) ).first():
+            raise ValueError("Duplicate Transaction")
+
+
+
+        transaction = cls(
+            transaction_id = transaction_id,
+            customer_id = customer_id,
+            account_id = account_id,
+            type = type,
+            channel = channel,
+            created_at=datetime.now(timezone.utc),
+            updated_at=datetime.now(timezone.utc)
+        )
+
+        try:
+            db.session.add(transaction)
+        except IntegrityError as err:
+            raise ValueError(f"Database integrity error: {err}")
+
+        return transaction
+
+    @classmethod
+    def get_transaction_by_id(cls, transaction_id):
+        return cls.query.get(transaction_id)