def validate_token():
"""Will return an error code if the token is invalidated
Otherwise, return the user information.
"""
transactions = Transaction.filter(user_id=g.user.user_id).order_by(
Transaction.date.desc()).all()
limits = Limit.get_all(user_id=g.user.user_id)
return make_response(
to_json({
'user': g.user.to_json,
'limits': limits,
'transactions': [t.as_dict() for t in transactions]
}))
def transactions():
# Retrieve user transactions
if request.method == 'GET':
transactions = Transaction.filter(user_id=g.user.user_id).order_by(
Transaction.date.desc()).all()
return make_response(to_json(transactions))
elif request.method in ['PUT', 'POST']:
data = request.get_json()
# Allow to insert new transactions
if request.method == 'PUT':
try:
attributes = get_transaction_attributes(data)
transaction = add_to_db(Transaction(), **attributes)
return HttpResponse(transaction.as_dict(), status=201)
except Exception as e:
return HttpErrorResponse(
e,
'Unable to add the transaction. Please try again later.',
status=400)
# Allow to edit a transaction
elif request.method == 'POST':
transaction = Transaction.get(
user_id=g.user.user_id, transaction_id=data['transaction_id'])
if transaction is not None:
try:
from endless import db_session
attributes = get_transaction_attributes(data)
set_attributes(transaction, **attributes)
db_session.commit()
return HttpResponse(transaction.as_dict(), status=201)
except Exception as e:
return HttpErrorResponse(
e,
'Unable to add the transaction. Please try again later.',
status=400)
else:
return HttpResponse(
"Unable to edit the transaction. Please try again later.",
status=500)
def upload_transactions():
file = request.files.get('file')
bank = Bank.get(bank_id=request.form.get('bank_id'))
if bank is None:
return HttpResponse("No bank was given.", status=400)
# check if the post request has the file part
if file is None or file.filename == '':
return HttpResponse("No file was given.", status=400)
if allowed_file(file.filename):
filename = str(uuid.uuid4()) + os.path.splitext(
secure_filename(file.filename))[1]
relative_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
full_path = os.path.abspath(relative_path)
# Saving file with unique name
file.save(full_path)
parsed = StatementParser(bank=bank).parse(full_path)
if len(parsed['transactions']) > 0:
added_transactions = put_transactions(parsed, g.user)
transactions = Transaction.filter(user_id=g.user.user_id).order_by(
Transaction.date.desc()).all()
response = {
"description": "Transactions uploaded.",
"transactions": [t.as_dict() for t in transactions],
"number": added_transactions
}
status = 200
else:
response = "No transactions found."
status = 204
os.remove(full_path)
return HttpResponse(response, status=status)
return HttpResponse("Error: Unauthorized file.", status=400)
