Get Links to Episodes

get_episodes_links.py

import re as regex

def get_episodes_links(link: str, driver_service) -> List[str]:
    """Retrieve links to episodes, from series' season main page.
    
    Args:
        link (str): link to season main page.
        driver_service (Service): a Chrome web driver.
        
    Returns:
        links (List[str]): a list of links to the episodes.
    """
    soup = make_soup_with_selenium(link, driver_service)
    website = "https://www.imdb.com"
    
    links = []
    for link in soup.find_all('a', href=True):
        href = link["href"]
        # Episode links end with the following Regex
        if bool(regex.search('=ttep_ep\d$', href)):
            links.append(href) 
            
    links = list(set(links))
    links.sort()
    links = [f"{website}{l}" for l in links]
    
    return links