def import_csv(session, filepath, bankaccount): tx_imported = 0 if bankaccount: with open(filepath, "r") as csvfile: batch = ImportBatch.from_csv(session, csvfile) if batch is not None: batch.bank_account = bankaccount session.add(batch) tx_imported = len(batch.transactions) session.commit() return tx_imported
def test_from_csv_invalid(self): self.assertIsNone(ImportBatch.from_csv(self.session, StringIO.StringIO(""))) self.assertIsNone(ImportBatch.from_csv(self.session, StringIO.StringIO("sdfj, asdfakfvankvdsj")))
def test_from_csv(self): self.session.query.return_value.filter_by.return_value.count.return_value = 0 csvdata = StringIO.StringIO("""06/09/2013,"-400.00",description""") batch = ImportBatch.from_csv(self.session, csvdata) self.assertEqual(1, len(batch.transactions))