def get_district_id(state, district):
cowin = CoWinAPI()
states = cowin.get_states()
state = [i for i in states["states"] if state in i["state_name"].lower()]
districts = cowin.get_districts(state[0]["state_id"])
district = [
i for i in districts["districts"]
if district in i["district_name"].lower()
]
district_id = str(district[0]["district_id"])
return district_id
from cowin_api import CoWinAPI cowin = CoWinAPI() states = cowin.get_states() print(states) # get the state id from above add it here state_id = '12' cowin = CoWinAPI() districts = cowin.get_districts(state_id) print(districts)
def test_get_states():
cowin = CoWinAPI()
states = cowin.get_states()
assert isinstance(states, dict)
assert states.get('states')[0].get('state_id') == 1