Back end Engineer Challenge

#!/usr/bin/python3

import requests

URL = "http://www.letsrevolutionizetesting.com/challenge.json"


def follow(url):
	response = requests.get(url).json()
	while 'follow' in response:
		yield response
		follow_url = response['follow']
		_id = follow_url.split('=')[-1]
		response = requests.get(URL, params={'id': _id}).json()


def main():
	for r in follow(URL):
		print(r)


if __name__ == '__main__':
	main()