# Lucas Mathews - Fontys Student ID: 5023572
# Banking System Transaction Class

from sqlalchemy import Column, String, Integer, ForeignKey

from class_base import Base

class Transaction(Base):
    __tablename__ = 'transactions'
    transaction_id = Column("transaction_id", String, primary_key=True)
    transaction_type = Column("transaction_type", String)
    amount = Column("amount", Integer)
    timestamp = Column("timestamp", String)
    description = Column("description", String)
    account_id = Column(String, ForeignKey('accounts.account_id'))
    recipient_account_id = Column(String, ForeignKey('accounts.account_id'))

    def __init__(self, transaction_id, transaction_type, amount, timestamp, description, account_id, recipient_account_id = None):
        """Initialises the Transaction object."""
        self.transaction_id = transaction_id
        self.transaction_type = transaction_type
        self.amount = amount
        self.timestamp = timestamp
        self.description = description
        self.account_id = account_id
        self.recipient_account_id = recipient_account_id

    def to_dict(self):
        """Converts the Transaction object to a dictionary."""
        return {
            "transaction_id": self.transaction_id,
            "transaction_type": self.transaction_type,
            "amount": self.amount,
            "timestamp": self.timestamp,
            "description": self.description,
            "account_id": self.account_id,
            "recipient_account_id": self.recipient_account_id
        }