Exemple #1
0
 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)
Exemple #2
0
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']))