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']))
