CODEWITHSUNDEEP
pythonjsonselenium
drock
drock

Reputation: 35

KeyError: 0 in Python When Trying to Get JSON Data

I am using selenium to check a value in a json request and I'm getting an error. What I am doing is first logging in as I can't get a response unless I am logged in. Then I am trying to grab a balance from the json for the user and print it so I can see that it actually worked. It is looking like a big nope.

The json looks like this:

{
  "account": {
     "balance": "28590",
     "status": "0"
  }
}

And here is my code:

from selenium import webdriver
import openpyxl
import json
import requests

driver = webdriver.Firefox(executable_path= r"geckodriver path")
baseURL = 'url'

#Get username/password from excel sheet
workbook = openpyxl.load_workbook(r"xlsx")
sheet = workbook.get_sheet_by_name('Sheet1')
username = str(sheet['A1'].value)
password = str(sheet['B1'].value)

#Open browser
driver.get(baseURL)
driver.maximize_window()

#Login
driver.find_element_by_xpath('xpath').click()
driver.implicitly_wait(500)
driver.find_element_by_id('email').send_keys(username)
driver.find_element_by_id('password').send_keys(password)
driver.find_element_by_id('login').click()

#Check balance
balance_api = 'balance/json'
json_data = requests.get(balance_api).json()
balance = json_data['account'][0]['balance']
print(balance)

Which gives me this error message:

Traceback (most recent call last):
File "pointsCheck.py", line 31, in <module>
balance = json_data['account'][0]['balance']
KeyError: 0

I'm not having any issues with the login. It's all when I try to grab the balance. I'm not sure if this is even the correct way to do this. I appreciate any help.

Upvotes: 1

Views: 5234

Answers (1)

Abid Hasan
Abid Hasan

Reputation: 658

You should use json_data['account']['balance']

It's not an array that you need to have the 0 index.

Upvotes: 2

Related Questions