def test_transaction_create_ok(session, data_transaction): count1 = session.query(Transaction).count() users = User.get_users(db_session=session) data = TransactionPost( reference=data_transaction['reference'], account=data_transaction['account'], date=data_transaction['date'], amount=data_transaction['amount'], type=data_transaction['type'], category=data_transaction['category'], ) Transaction.create(db_session=session, user_id=users[0].id, data=data) count2 = session.query(Transaction).count() assert count1 + 1 == count2
def new_transaction(session, new_user, data_transaction): data = TransactionPost( reference=data_transaction['reference'], account=data_transaction['account'], date=data_transaction['date'], amount=data_transaction['amount'], type=data_transaction['type'], category=data_transaction['category'], ) return Transaction.create( db_session=session, user_id=new_user.id, data=data, )
def data_import(bank_type, file_path): bank = str(bank_type) config = Config(template_type=bank).template() csv_read_type = config['parsing_settings']['csv_read_type'] csv_column_config = config['csv_column_config'] delimiter = config['parsing_settings']['delimiter'] date_format = config['parsing_settings']['date_format'] file_path = str(file_path) raw_data = Csv(csv_read_type=csv_read_type, file_path=file_path, delimiter=delimiter).reader() data = dataTransform().transform(bank_name=bank, raw_data=raw_data, csv_column_config=csv_column_config, date_format=date_format) for row in data: random_id = genUuid() print( Transaction.create(id=random_id, data=row['transaction'], asset_id=row['asset_id']))