Querying a district_code to state_code lookup service

get_state_for_district.py

import json
import requests

def get_state_for_district(district_code):
    '''
    Queries a hypothetical URL endpoint of the following form:
    /lookup?level=district&parent=state&district=<district_code>
    
    and retrieves a state_code for the given district_code
    '''
    params = {'level': 'district', 'parent': 'state', 'district': district_code}
    res = requests.get('http://localhost:9500/lookup', params=params)
    res = json.loads(res.text)
    return res['state']