SUSTableView-Proper.swift

class TransactionTable: CustomTableView {
    override init(viewModel: TransactionTableViewModel) {
        super.init()
        
        viewModel.transactionViewModels.forEach { transactionViewModel in
            switch transactionViewModel {
            case .info(let viewModel):
                let cell = TransactionInfoCell(viewModel: viewModel)
                
                self.addCell(cell)
            case .infoPendingAccept(let viewModel):
                let cell = TransactionInfoWAcceptCell(viewModel: viewModel)
                
                self.addCell(cell)
            }
        }
    }
}