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