Method fetch_links_helper

sdow/database.py:200–223

Helper function which handles duplicate logic for fetch_outgoing_links() and fetch_incoming_links(). Args: page_ids: A list of page IDs whose links to fetch. outcoming_or_incoming_links: String which indicates whether to fetch outgoing ("source_id") or incoming ("target_

(self, page_ids, outcoming_or_incoming_links)

    return self.fetch_links_helper(page_ids, 'incoming_links')

  def fetch_links_helper(self, page_ids, outcoming_or_incoming_links):
    """Helper function which handles duplicate logic for fetch_outgoing_links() and
    fetch_incoming_links().

    Args:
      page_ids: A list of page IDs whose links to fetch.
      outcoming_or_incoming_links: String which indicates whether to fetch outgoing ("source_id") or
        incoming ("target_id") links.

    Returns:
      list(int, int): A cursor of a lists of integer tuples representing links from the list of
        provided page IDs to other pages.
    """
    # Convert the page IDs into a string surrounded by parentheses for insertion into the query
    # below. The replace() bit is some hackery to handle Python printing a trailing ',' when there
    # is only one key.
    page_ids = str(tuple(page_ids)).replace(',)', ')')

    # There is no need to escape the query parameters here since they are never user-defined.
    query = 'SELECT id, {0} FROM links WHERE id IN {1};'.format(
        outcoming_or_incoming_links, page_ids)
    self.sdow_cursor.execute(query)

    return self.sdow_cursor

  def insert_result(self, search):
    """Inserts a new search result into the searches table.