Get terraform resource based on its address

search.py

import sys
import json

state = json.loads(sys.stdin.read())
address = sys.argv[1]


def get_by_address(state, address):
    if type(state) is list:
        for i in state:
            found = get_by_address(i, address)
            if found:
                return found
    elif type(state) is dict:
        if state.get('address', '') == address:
            return state
        else:
            for i in state:
                found = get_by_address(state[i], address)
                if found:
                    return found


print(json.dumps(get_by_address(state, address)))