import html import logging import re from datetime import datetime, timezone from dateutil.parser._parser import ParserError from enum import Enum from urllib.parse import SplitResult, urlsplit, urlunsplit import dateutil.parser from bs4 import BeautifulSoup, SoupStrainer from requests.exceptions import MissingSchema from newspipe.bootstrap import application from newspipe.lib.utils import newspipe_get logger = logging.getLogger(__name__) PROCESSED_DATE_KEYS = {"published", "created", "updated"} def extract_id(entry): """extract a value from an entry that will identify it among the other of that feed""" entry_id = "undefined" try: entry_id = entry.get("entry_id") or entry.get("id") or entry["link"] except: pass return entry_id async def construct_article(entry, feed, fields=None, fetch=True): "Safe method to transform a feedparser entry into an article" now = datetime.utcnow() article = {} def push_in_article(key, value): if not fields or key in fields: article[key] = value push_in_article("feed_id", feed.id) push_in_article("user_id", feed.user_id) push_in_article("entry_id", extract_id(entry)) push_in_article("retrieved_date", now) if not fields or "date" in fields: for date_key in PROCESSED_DATE_KEYS: if entry.get(date_key): try: article["date"] = dateutil.parser.parse(entry[date_key]).astimezone( timezone.utc ) except ParserError: logger.exception( "Error when parsing date: {}".format(entry[date_key]) ) except Exception as e: pass else: break push_in_article("content", get_article_content(entry)) if fields is None or {"link", "title"}.intersection(fields): link, title = await get_article_details(entry, fetch) push_in_article("link", link) push_in_article("title", title) if "content" in article: # push_in_article('content', clean_urls(article['content'], link)) push_in_article("content", article["content"]) push_in_article( "tags", { tag.get("term").strip() for tag in entry.get("tags", []) if tag and tag.get("term", False) }, ) return article def get_article_content(entry): content = "" if entry.get("content"): content = entry["content"][0]["value"] elif entry.get("summary"): content = entry["summary"] return content async def get_article_details(entry, fetch=True): article_link = entry.get("link") article_title = html.unescape(entry.get("title", "")) if ( fetch and application.config["CRAWLER_RESOLVE_ARTICLE_URL"] and article_link or not article_title ): try: # resolves URL behind proxies (like feedproxy.google.com) response = newspipe_get(article_link, timeout=5) except MissingSchema: split, failed = urlsplit(article_link), False for scheme in "https", "http": try: new_link = urlunsplit(SplitResult(scheme, *split[1:])) response = newspipe_get(new_link, timeout=5) except Exception: failed = True continue failed = False article_link = new_link break if failed: return article_link, article_title or "No title" except Exception as error: logger.info( "Unable to get the real URL of %s. Won't fix " "link or title. Error: %s", article_link, error, ) return article_link, article_title or "No title" article_link = response.url if not article_title: bs_parsed = BeautifulSoup( response.content, "html.parser", parse_only=SoupStrainer("head") ) try: article_title = bs_parsed.find_all("title")[0].text except IndexError: # no title pass return article_link, article_title or "No title" class FiltersAction(Enum): READ = "mark as read" LIKED = "mark as favorite" SKIP = "skipped" class FiltersType(Enum): REGEX = "regex" MATCH = "simple match" EXACT_MATCH = "exact match" TAG_MATCH = "tag match" TAG_CONTAINS = "tag contains" class FiltersTrigger(Enum): MATCH = "match" NO_MATCH = "no match" def process_filters(filters, article, only_actions=None): skipped, read, liked = False, None, False filters = filters or [] if only_actions is None: only_actions = set(FiltersAction) for filter_ in filters: match = False try: pattern = filter_.get("pattern", "") filter_type = FiltersType(filter_.get("type")) filter_action = FiltersAction(filter_.get("action")) filter_trigger = FiltersTrigger(filter_.get("action on")) if filter_type is not FiltersType.REGEX: pattern = pattern.lower() except ValueError: continue if filter_action not in only_actions: logger.debug("ignoring filter %r" % filter_) continue if ( filter_action in {FiltersType.REGEX, FiltersType.MATCH, FiltersType.EXACT_MATCH} and "title" not in article ): continue if ( filter_action in {FiltersType.TAG_MATCH, FiltersType.TAG_CONTAINS} and "tags" not in article ): continue title = article.get("title", "").lower() tags = [tag.lower() for tag in article.get("tags", [])] if filter_type is FiltersType.REGEX: match = re.match(pattern, title) elif filter_type is FiltersType.MATCH: match = pattern in title elif filter_type is FiltersType.EXACT_MATCH: match = pattern == title elif filter_type is FiltersType.TAG_MATCH: match = pattern in tags elif filter_type is FiltersType.TAG_CONTAINS: match = any(pattern in tag for tag in tags) take_action = ( match and filter_trigger is FiltersTrigger.MATCH or not match and filter_trigger is FiltersTrigger.NO_MATCH ) if not take_action: continue if filter_action is FiltersAction.READ: read = True elif filter_action is FiltersAction.LIKED: liked = True elif filter_action is FiltersAction.SKIP: skipped = True if skipped or read or liked: logger.info( "%r applied on %r", filter_action.value, article.get("link") or article.get("title"), ) return skipped, read, liked def get_skip_and_ids(entry, feed): entry_ids = construct_article( entry, feed, {"entry_id", "feed_id", "user_id"}, fetch=False ) skipped, _, _ = process_filters( feed.filters, construct_article(entry, feed, {"title", "tags"}, fetch=False), {FiltersAction.SKIP}, ) return skipped, entry_ids