functions/helper_functions.py

'''Helper functions for MCP tools.'''

import logging
from types import GeneratorType

import feedparser
from findfeed import search as feed_search
from googlesearch import search as google_search

def get_url(company_name: str) -> str:
    '''Finds the website associated with the name of a company or
    publication.

    Args:
        company_name: the name of the company, publication or site to find
        the URL for

    Returns:
        The URL for the company, publication or website.
    '''

    logger = logging.getLogger(__name__ + '.get_url')

    query = f'{company_name} official website'

    for url in google_search(query, num_results=5):
        if 'facebook' not in url and 'linkedin' not in url:
            return url

    return None


def get_feed(website_url: str) -> str:
    '''Finds the RSS feed URI for a website given the website's url.
    
    Args:
        website_url: The url for the website to find the RSS feed for
        
    Returns:
        The website's RSS feed URI as a string
    '''

    logger = logging.getLogger(__name__ + '.get_content')
    logger.info('Getting feed URI for: %s', website_url)

    feeds = feed_search(website_url)

    logger.info('Feeds search result is: %s', type(feeds))
    logger.info('Feeds search results: %s', len(feeds))
    logger.info('Feeds results: %s', list(feeds))

    if len(feeds) > 0:
        return str(feeds[0].url)

    else:
        return f'No feed found for {website_url}'


def parse_feed(feed_uri: str) -> list:
    '''Gets content from a remote RSS feed URI.
    
    Args:
        feed_uri: The RSS feed to get content from

    Returns:
        List of titles for the 10 most recent entries in the RSS feed.
    '''

    logger = logging.getLogger(__name__ + '.parse_feed')

    feed = feedparser.parse(feed_uri)
    logger.info('%s yieled %s entries', feed_uri, len(feed.entries))

    titles = []

    for entry in feed.entries:

        logger.debug('Entry attributes: %s', list(entry.keys()))

        if 'title' in entry:
            titles.append(entry.title)

        if len(titles) >= 10:
            break

    return titles