def delete(self):
parser = reqparse.RequestParser()
id = parser.add_argument('id', required=True,
type=int).parse_args()['id']
balance = Balance.find_by_id(id)
if balance.system is False:
Balance.delete(id)
def get(self):
request = reqparse.RequestParser() \
.add_argument('id', type=int) \
.add_argument('balances') \
.parse_args()
user_id = get_jwt_identity()
id = request['id']
if id is not None:
if id < 0:
return resp_util.invalid_request()
t = Transaction.find_by_user_id(user_id, id)
return ser_util.serialize_transaction(
t, Balance.actual_balance(
t.balance_id)) if t else resp_util.not_found()
try:
balances_string = request['balances']
except KeyError:
balances_string = ''
balances_stripped_string = balances_string.strip()
transactions = Transaction.find_by_user_id(user_id) if not balances_stripped_string \
else Transaction.find_by_balance_ids(balances_stripped_string.split(','))
return ser_util.serialize_transactions(
transactions, calculate_actual_balances(transactions))
def get(self):
id = reqparse.RequestParser() \
.add_argument('id', type=int) \
.parse_args()['id']
if id is None:
response = []
for goal in User.find_by_id(get_jwt_identity()).goals:
response.append(
ser_util.serialize_goal(
goal, Balance.actual_balance(goal.balance_id)))
return response
if id <= 0:
return resp_util.invalid_request('Given id isn\'t valid')
goal = Goal.find_by_user_id(get_jwt_identity(), id)
if not goal:
return resp_util.not_found()
return ser_util.serialize_goal(goal,
Balance.actual_balance(goal.balance_id))
def calculate_actual_balances(transactions):
actual_balances_dict = {}
actual_balances = []
for t in transactions:
try:
actual_balance = actual_balances_dict[t.balance_id]
except KeyError:
actual_balance = Balance.actual_balance(t.balance_id)
actual_balances_dict[t.balance_id] = actual_balance
actual_balances.append(actual_balance)
return actual_balances
def post(self):
parser = reqparse.RequestParser()
parser.add_argument('name', required=True)
parser.add_argument('currency', required=True)
data = parser.parse_args()
if not validate_balance_request(data):
return resp_util.invalid_request()
Balance(user_id=get_jwt_identity(),
name=data['name'],
annual_income_percentage=0,
system=False,
currency=data['currency']).save_to_db()
def post(self):
request = reqparse.RequestParser() \
.add_argument('name', required=True) \
.add_argument('initialAmount', required=True, type=int) \
.add_argument('currency', required=True) \
.parse_args()
if not validate_budget_request(request):
return resp_util.invalid_request()
balance = Balance(user_id=get_jwt_identity(),
name='System balance for ' + request['name'] +
' goal',
annual_income_percentage=0,
system=True,
currency=request['currency'])
balance_id = balance.save_to_db()
Goal(user_id=get_jwt_identity(),
initial_amount=request['initialAmount'],
name=request['name'],
currency=request['currency'],
balance_id=balance_id).save_to_db()
def put(self):
parser = reqparse.RequestParser()
parser.add_argument('name', required=True)
parser.add_argument('id', required=True)
data = parser.parse_args()
if not data['name'].strip():
return resp_util.invalid_request()
balance = Balance.find_by_id(data['id'])
if balance is None or balance.system is True:
return resp_util.not_found()
balance.name = data['name']
try:
balance.update()
except:
return {'message': 'Something went wrong'}, 500
def get(self):
request = reqparse.RequestParser() \
.add_argument('id', type=int) \
.add_argument('currency') \
.parse_args()
if request['id'] is None:
balances = list(
filter(lambda b: b.system is False,
User.find_by_id(get_jwt_identity()).balances))
return ser_util.serialize_balances(balances) if not request['currency'] \
else ser_util.serialize_balances(list(filter(lambda b: b.currency == request['currency'], balances)))
if request['id'] <= 0:
return resp_util.invalid_request('Given id isn\'t valid')
balance = Balance.find_by_user_id(get_jwt_identity(), request['id'])
if not balance or balance.system is True:
return resp_util.not_found()
return ser_util.serialize_balance(balance)
def post(self):
request = reqparse.RequestParser() \
.add_argument('balanceId', type=int, required=True) \
.add_argument('amount', type=float, required=True) \
.add_argument('goalId', type=int, required=True) \
.parse_args()
user_id = get_jwt_identity()
if request['goalId'] not in list(
map(lambda b: b.id,
User.find_by_id(user_id).goals)):
return resp_util.access_denied()
if request['balanceId'] not in list(
map(lambda b: b.id,
User.find_by_id(user_id).balances)):
return resp_util.access_denied()
if request['amount'] <= 0:
return resp_util.invalid_request()
goal = Goal.find_by_id(request['goalId'])
balance = Balance.find_by_id(request['balanceId'])
if goal.currency != balance.currency:
return {'message': 'Currencies doesn\'t match'}, 500
t1 = Transaction(balance_id=request['balanceId'],
user_id=user_id,
amount=-request['amount'],
datetime=datetime.now(),
description='filling the goal: ' + str(goal.id))
t2 = Transaction(balance_id=goal.balance_id,
user_id=user_id,
amount=request['amount'],
datetime=datetime.now(),
description='filling the goal: ' + str(goal.id))
Transaction.save_to_db_in_session(t1, t2)
def create(user_id, currency, name, income_percentage):
balance = Balance(user_id=user_id, currency=currency, name=name, annual_income_percentage=income_percentage)
db.session.add(balance)
db.session.commit()