getMovieTitles

getMovieTitles.py

import requests

def firstQuery(substr):
    response = requests.get(f"https://jsonmock.hackerrank.com/api/movies/search/?Title={substr}")
    respDict = response.json()
    rtn = {
        "per_page": respDict["per_page"], "total": respDict["total"], "total_pages": respDict["total_pages"]
    }
    return rtn
    
def pageQuery(substr, p):
    response = requests.get(f"https://jsonmock.hackerrank.com/api/movies/search/?Title={substr}&page={p}")
    respDict = response.json()
    
    return list(map(lambda d: d["Title"], respDict["data"]))
    
def getMovieTitles(substr):    
    info = firstQuery(substr)
    rtn = []
    
    for p in range(1, info["total_pages"] + 1):
        rtn += pageQuery(substr, p)
    rtn.sort()
    return rtn

getMovieTitles("waterworld")