def test_DataTransform_otp_credit(self): test_data_result = {} bank = 'otp_credit' file_path = 'test/data/otp_credit.csv' 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'] 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: test_data_result.update({'date': '2020-04-30'}) test_data_result.update({'payee': 'TEST PAYEE'}) if row['transaction']['payee'] == 'TEST PAYEE NOTES': test_data_result.update({'payee': 'TEST PAYEE NOTES'}) test_data_result.update({'amount': row['transaction']['amount']}) test_data_result.update( {'currency': row['transaction']['currency']}) test_data_result.update({'notes': 'TEST NOTES'}) if row['transaction']['payee'] == 'TEST PAYEE NOTES': test_data_result.update({'notes': 'TEST PAYEE NOTES'}) self.assertDictEqual(row['transaction'], test_data_result)
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']))