def test_currency_to_number(self):
self.assertStrictEqual(util.currency_to_float('123.45'), 123.45)
self.assertStrictEqual(util.currency_to_float('$123.45'), 123.45)
self.assertStrictEqual(util.currency_to_float('123.45$'), 123.45)
self.assertStrictEqual(util.currency_to_float('123.45$USD'), 123.45)
self.assertStrictEqual(util.currency_to_float('1,230.45'), 1230.45)
self.assertStrictEqual(util.currency_to_float('123'), 123.0)
self.assertStrictEqual(util.currency_to_float('$123'), 123.0)
def get_cash_balance(self):
"""
Returns the cash balance available to invest
"""
cash = -1
try:
response = util.get_url('/browse/cashBalanceAj.action')
json = response.json()
if json['result'] == 'success':
self.logger.debug('Cash available: {0}'.format(json['cashBalance']))
cash = util.currency_to_float(json['cashBalance'])
else:
self.logger.error('Could not get cash balance: {0}'.format(response.text))
except Exception as e:
self.logger.error('Could not get the cash balance on the account: {0}\nJSON: {1}'.format(str(e), response.text))
return cash
