| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785 |
- # Copyright 2010-2023 Kurt McKee <contactme@kurtmckee.org>
- # Copyright 2002-2008 Mark Pilgrim
- # All rights reserved.
- #
- # This file is a part of feedparser.
- #
- # Redistribution and use in source and binary forms, with or without
- # modification, are permitted provided that the following conditions are met:
- #
- # * Redistributions of source code must retain the above copyright notice,
- # this list of conditions and the following disclaimer.
- # * Redistributions in binary form must reproduce the above copyright notice,
- # this list of conditions and the following disclaimer in the documentation
- # and/or other materials provided with the distribution.
- #
- # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 'AS IS'
- # AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- # ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
- # LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
- # CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
- # SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
- # INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
- # CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
- # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
- # POSSIBILITY OF SUCH DAMAGE.
- import base64
- import binascii
- import copy
- import html.entities
- import re
- import xml.sax.saxutils
- from .html import _cp1252
- from .namespaces import _base, cc, dc, georss, itunes, mediarss, psc
- from .sanitizer import _sanitize_html, _HTMLSanitizer
- from .util import FeedParserDict
- from .urls import _urljoin, make_safe_absolute_uri, resolve_relative_uris
- class _FeedParserMixin(
- _base.Namespace,
- cc.Namespace,
- dc.Namespace,
- georss.Namespace,
- itunes.Namespace,
- mediarss.Namespace,
- psc.Namespace,
- ):
- namespaces = {
- '': '',
- 'http://backend.userland.com/rss': '',
- 'http://blogs.law.harvard.edu/tech/rss': '',
- 'http://purl.org/rss/1.0/': '',
- 'http://my.netscape.com/rdf/simple/0.9/': '',
- 'http://example.com/newformat#': '',
- 'http://example.com/necho': '',
- 'http://purl.org/echo/': '',
- 'uri/of/echo/namespace#': '',
- 'http://purl.org/pie/': '',
- 'http://purl.org/atom/ns#': '',
- 'http://www.w3.org/2005/Atom': '',
- 'http://purl.org/rss/1.0/modules/rss091#': '',
- 'http://webns.net/mvcb/': 'admin',
- 'http://purl.org/rss/1.0/modules/aggregation/': 'ag',
- 'http://purl.org/rss/1.0/modules/annotate/': 'annotate',
- 'http://media.tangent.org/rss/1.0/': 'audio',
- 'http://backend.userland.com/blogChannelModule': 'blogChannel',
- 'http://creativecommons.org/ns#license': 'cc',
- 'http://web.resource.org/cc/': 'cc',
- 'http://cyber.law.harvard.edu/rss/creativeCommonsRssModule.html': 'creativeCommons',
- 'http://backend.userland.com/creativeCommonsRssModule': 'creativeCommons',
- 'http://purl.org/rss/1.0/modules/company': 'co',
- 'http://purl.org/rss/1.0/modules/content/': 'content',
- 'http://my.theinfo.org/changed/1.0/rss/': 'cp',
- 'http://purl.org/dc/elements/1.1/': 'dc',
- 'http://purl.org/dc/terms/': 'dcterms',
- 'http://purl.org/rss/1.0/modules/email/': 'email',
- 'http://purl.org/rss/1.0/modules/event/': 'ev',
- 'http://rssnamespace.org/feedburner/ext/1.0': 'feedburner',
- 'http://freshmeat.net/rss/fm/': 'fm',
- 'http://xmlns.com/foaf/0.1/': 'foaf',
- 'http://www.w3.org/2003/01/geo/wgs84_pos#': 'geo',
- 'http://www.georss.org/georss': 'georss',
- 'http://www.opengis.net/gml': 'gml',
- 'http://postneo.com/icbm/': 'icbm',
- 'http://purl.org/rss/1.0/modules/image/': 'image',
- 'http://www.itunes.com/DTDs/PodCast-1.0.dtd': 'itunes',
- 'http://example.com/DTDs/PodCast-1.0.dtd': 'itunes',
- 'http://purl.org/rss/1.0/modules/link/': 'l',
- 'http://search.yahoo.com/mrss': 'media',
- # Version 1.1.2 of the Media RSS spec added the trailing slash on the namespace
- 'http://search.yahoo.com/mrss/': 'media',
- 'http://madskills.com/public/xml/rss/module/pingback/': 'pingback',
- 'http://prismstandard.org/namespaces/1.2/basic/': 'prism',
- 'http://www.w3.org/1999/02/22-rdf-syntax-ns#': 'rdf',
- 'http://www.w3.org/2000/01/rdf-schema#': 'rdfs',
- 'http://purl.org/rss/1.0/modules/reference/': 'ref',
- 'http://purl.org/rss/1.0/modules/richequiv/': 'reqv',
- 'http://purl.org/rss/1.0/modules/search/': 'search',
- 'http://purl.org/rss/1.0/modules/slash/': 'slash',
- 'http://schemas.xmlsoap.org/soap/envelope/': 'soap',
- 'http://purl.org/rss/1.0/modules/servicestatus/': 'ss',
- 'http://hacks.benhammersley.com/rss/streaming/': 'str',
- 'http://purl.org/rss/1.0/modules/subscription/': 'sub',
- 'http://purl.org/rss/1.0/modules/syndication/': 'sy',
- 'http://schemas.pocketsoap.com/rss/myDescModule/': 'szf',
- 'http://purl.org/rss/1.0/modules/taxonomy/': 'taxo',
- 'http://purl.org/rss/1.0/modules/threading/': 'thr',
- 'http://purl.org/rss/1.0/modules/textinput/': 'ti',
- 'http://madskills.com/public/xml/rss/module/trackback/': 'trackback',
- 'http://wellformedweb.org/commentAPI/': 'wfw',
- 'http://purl.org/rss/1.0/modules/wiki/': 'wiki',
- 'http://www.w3.org/1999/xhtml': 'xhtml',
- 'http://www.w3.org/1999/xlink': 'xlink',
- 'http://www.w3.org/XML/1998/namespace': 'xml',
- 'http://podlove.org/simple-chapters': 'psc',
- }
- _matchnamespaces = {}
- can_be_relative_uri = {
- 'comments',
- 'docs',
- 'href',
- 'icon',
- 'id',
- 'link',
- 'logo',
- 'url',
- 'wfw_comment',
- 'wfw_commentrss',
- }
- can_contain_relative_uris = {
- 'content',
- 'copyright',
- 'description',
- 'info',
- 'rights',
- 'subtitle',
- 'summary',
- 'tagline',
- 'title',
- }
- can_contain_dangerous_markup = {
- 'content',
- 'copyright',
- 'description',
- 'info',
- 'rights',
- 'subtitle',
- 'summary',
- 'tagline',
- 'title',
- }
- html_types = {
- 'application/xhtml+xml',
- 'text/html',
- }
- def __init__(self):
- if not self._matchnamespaces:
- for k, v in self.namespaces.items():
- self._matchnamespaces[k.lower()] = v
- self.feeddata = FeedParserDict() # feed-level data
- self.entries = [] # list of entry-level data
- self.version = '' # feed type/version, see SUPPORTED_VERSIONS
- self.namespaces_in_use = {} # dictionary of namespaces defined by the feed
- # the following are used internally to track state;
- # this is really out of control and should be refactored
- self.infeed = 0
- self.inentry = 0
- self.incontent = 0
- self.intextinput = 0
- self.inimage = 0
- self.inauthor = 0
- self.incontributor = 0
- self.inpublisher = 0
- self.insource = 0
- self.sourcedata = FeedParserDict()
- self.contentparams = FeedParserDict()
- self._summaryKey = None
- self.namespacemap = {}
- self.elementstack = []
- self.basestack = []
- self.langstack = []
- self.svgOK = 0
- self.title_depth = -1
- self.depth = 0
- self.hasContent = 0
- if self.lang:
- self.feeddata['language'] = self.lang.replace('_', '-')
- # A map of the following form:
- # {
- # object_that_value_is_set_on: {
- # property_name: depth_of_node_property_was_extracted_from,
- # other_property: depth_of_node_property_was_extracted_from,
- # },
- # }
- self.property_depth_map = {}
- super(_FeedParserMixin, self).__init__()
- def _normalize_attributes(self, kv):
- raise NotImplementedError
- def unknown_starttag(self, tag, attrs):
- # increment depth counter
- self.depth += 1
- # normalize attrs
- attrs = [self._normalize_attributes(attr) for attr in attrs]
- # track xml:base and xml:lang
- attrs_d = dict(attrs)
- baseuri = attrs_d.get('xml:base', attrs_d.get('base')) or self.baseuri
- if isinstance(baseuri, bytes):
- baseuri = baseuri.decode(self.encoding, 'ignore')
- # ensure that self.baseuri is always an absolute URI that
- # uses a whitelisted URI scheme (e.g. not `javscript:`)
- if self.baseuri:
- self.baseuri = make_safe_absolute_uri(self.baseuri, baseuri) or self.baseuri
- else:
- self.baseuri = _urljoin(self.baseuri, baseuri)
- lang = attrs_d.get('xml:lang', attrs_d.get('lang'))
- if lang == '':
- # xml:lang could be explicitly set to '', we need to capture that
- lang = None
- elif lang is None:
- # if no xml:lang is specified, use parent lang
- lang = self.lang
- if lang:
- if tag in ('feed', 'rss', 'rdf:RDF'):
- self.feeddata['language'] = lang.replace('_', '-')
- self.lang = lang
- self.basestack.append(self.baseuri)
- self.langstack.append(lang)
- # track namespaces
- for prefix, uri in attrs:
- if prefix.startswith('xmlns:'):
- self.track_namespace(prefix[6:], uri)
- elif prefix == 'xmlns':
- self.track_namespace(None, uri)
- # track inline content
- if self.incontent and not self.contentparams.get('type', 'xml').endswith('xml'):
- if tag in ('xhtml:div', 'div'):
- return # typepad does this 10/2007
- # element declared itself as escaped markup, but it isn't really
- self.contentparams['type'] = 'application/xhtml+xml'
- if self.incontent and self.contentparams.get('type') == 'application/xhtml+xml':
- if tag.find(':') != -1:
- prefix, tag = tag.split(':', 1)
- namespace = self.namespaces_in_use.get(prefix, '')
- if tag == 'math' and namespace == 'http://www.w3.org/1998/Math/MathML':
- attrs.append(('xmlns', namespace))
- if tag == 'svg' and namespace == 'http://www.w3.org/2000/svg':
- attrs.append(('xmlns', namespace))
- if tag == 'svg':
- self.svgOK += 1
- return self.handle_data('<%s%s>' % (tag, self.strattrs(attrs)), escape=0)
- # match namespaces
- if tag.find(':') != -1:
- prefix, suffix = tag.split(':', 1)
- else:
- prefix, suffix = '', tag
- prefix = self.namespacemap.get(prefix, prefix)
- if prefix:
- prefix = prefix + '_'
- # Special hack for better tracking of empty textinput/image elements in
- # illformed feeds.
- if (not prefix) and tag not in ('title', 'link', 'description', 'name'):
- self.intextinput = 0
- if (not prefix) and tag not in ('title', 'link', 'description', 'url', 'href', 'width', 'height'):
- self.inimage = 0
- # call special handler (if defined) or default handler
- methodname = '_start_' + prefix + suffix
- try:
- method = getattr(self, methodname)
- return method(attrs_d)
- except AttributeError:
- # Since there's no handler or something has gone wrong we
- # explicitly add the element and its attributes.
- unknown_tag = prefix + suffix
- if len(attrs_d) == 0:
- # No attributes so merge it into the enclosing dictionary
- return self.push(unknown_tag, 1)
- else:
- # Has attributes so create it in its own dictionary
- context = self._get_context()
- context[unknown_tag] = attrs_d
- def unknown_endtag(self, tag):
- # match namespaces
- if tag.find(':') != -1:
- prefix, suffix = tag.split(':', 1)
- else:
- prefix, suffix = '', tag
- prefix = self.namespacemap.get(prefix, prefix)
- if prefix:
- prefix = prefix + '_'
- if suffix == 'svg' and self.svgOK:
- self.svgOK -= 1
- # call special handler (if defined) or default handler
- methodname = '_end_' + prefix + suffix
- try:
- if self.svgOK:
- raise AttributeError()
- method = getattr(self, methodname)
- method()
- except AttributeError:
- self.pop(prefix + suffix)
- # track inline content
- if self.incontent and not self.contentparams.get('type', 'xml').endswith('xml'):
- # element declared itself as escaped markup, but it isn't really
- if tag in ('xhtml:div', 'div'):
- return # typepad does this 10/2007
- self.contentparams['type'] = 'application/xhtml+xml'
- if self.incontent and self.contentparams.get('type') == 'application/xhtml+xml':
- tag = tag.split(':')[-1]
- self.handle_data('</%s>' % tag, escape=0)
- # track xml:base and xml:lang going out of scope
- if self.basestack:
- self.basestack.pop()
- if self.basestack and self.basestack[-1]:
- self.baseuri = self.basestack[-1]
- if self.langstack:
- self.langstack.pop()
- if self.langstack: # and (self.langstack[-1] is not None):
- self.lang = self.langstack[-1]
- self.depth -= 1
- def handle_charref(self, ref):
- # Called for each character reference, e.g. for ' ', ref is '160'
- if not self.elementstack:
- return
- ref = ref.lower()
- if ref in ('34', '38', '39', '60', '62', 'x22', 'x26', 'x27', 'x3c', 'x3e'):
- text = '&#%s;' % ref
- else:
- if ref[0] == 'x':
- c = int(ref[1:], 16)
- else:
- c = int(ref)
- text = chr(c).encode('utf-8')
- self.elementstack[-1][2].append(text)
- def handle_entityref(self, ref):
- # Called for each entity reference, e.g. for '©', ref is 'copy'
- if not self.elementstack:
- return
- if ref in ('lt', 'gt', 'quot', 'amp', 'apos'):
- text = '&%s;' % ref
- elif ref in self.entities:
- text = self.entities[ref]
- if text.startswith('&#') and text.endswith(';'):
- return self.handle_entityref(text)
- else:
- try:
- html.entities.name2codepoint[ref]
- except KeyError:
- text = '&%s;' % ref
- else:
- text = chr(html.entities.name2codepoint[ref]).encode('utf-8')
- self.elementstack[-1][2].append(text)
- def handle_data(self, text, escape=1):
- # Called for each block of plain text, i.e. outside of any tag and
- # not containing any character or entity references
- if not self.elementstack:
- return
- if escape and self.contentparams.get('type') == 'application/xhtml+xml':
- text = xml.sax.saxutils.escape(text)
- self.elementstack[-1][2].append(text)
- def handle_comment(self, text):
- # Called for each comment, e.g. <!-- insert message here -->
- pass
- def handle_pi(self, text):
- # Called for each processing instruction, e.g. <?instruction>
- pass
- def handle_decl(self, text):
- pass
- def parse_declaration(self, i):
- # Override internal declaration handler to handle CDATA blocks.
- if self.rawdata[i:i+9] == '<![CDATA[':
- k = self.rawdata.find(']]>', i)
- if k == -1:
- # CDATA block began but didn't finish
- k = len(self.rawdata)
- return k
- self.handle_data(xml.sax.saxutils.escape(self.rawdata[i+9:k]), 0)
- return k+3
- else:
- k = self.rawdata.find('>', i)
- if k >= 0:
- return k+1
- else:
- # We have an incomplete CDATA block.
- return k
- @staticmethod
- def map_content_type(content_type):
- content_type = content_type.lower()
- if content_type == 'text' or content_type == 'plain':
- content_type = 'text/plain'
- elif content_type == 'html':
- content_type = 'text/html'
- elif content_type == 'xhtml':
- content_type = 'application/xhtml+xml'
- return content_type
- def track_namespace(self, prefix, uri):
- loweruri = uri.lower()
- if not self.version:
- if (prefix, loweruri) == (None, 'http://my.netscape.com/rdf/simple/0.9/'):
- self.version = 'rss090'
- elif loweruri == 'http://purl.org/rss/1.0/':
- self.version = 'rss10'
- elif loweruri == 'http://www.w3.org/2005/atom':
- self.version = 'atom10'
- if loweruri.find('backend.userland.com/rss') != -1:
- # match any backend.userland.com namespace
- uri = 'http://backend.userland.com/rss'
- loweruri = uri
- if loweruri in self._matchnamespaces:
- self.namespacemap[prefix] = self._matchnamespaces[loweruri]
- self.namespaces_in_use[self._matchnamespaces[loweruri]] = uri
- else:
- self.namespaces_in_use[prefix or ''] = uri
- def resolve_uri(self, uri):
- return _urljoin(self.baseuri or '', uri)
- @staticmethod
- def decode_entities(element, data):
- return data
- @staticmethod
- def strattrs(attrs):
- return ''.join(
- ' %s="%s"' % (t[0], xml.sax.saxutils.escape(t[1], {'"': '"'}))
- for t in attrs
- )
- def push(self, element, expecting_text):
- self.elementstack.append([element, expecting_text, []])
- def pop(self, element, strip_whitespace=1):
- if not self.elementstack:
- return
- if self.elementstack[-1][0] != element:
- return
- element, expecting_text, pieces = self.elementstack.pop()
- # Ensure each piece is a str for Python 3
- for (i, v) in enumerate(pieces):
- if isinstance(v, bytes):
- pieces[i] = v.decode('utf-8')
- if self.version == 'atom10' and self.contentparams.get('type', 'text') == 'application/xhtml+xml':
- # remove enclosing child element, but only if it is a <div> and
- # only if all the remaining content is nested underneath it.
- # This means that the divs would be retained in the following:
- # <div>foo</div><div>bar</div>
- while pieces and len(pieces) > 1 and not pieces[-1].strip():
- del pieces[-1]
- while pieces and len(pieces) > 1 and not pieces[0].strip():
- del pieces[0]
- if pieces and (pieces[0] == '<div>' or pieces[0].startswith('<div ')) and pieces[-1] == '</div>':
- depth = 0
- for piece in pieces[:-1]:
- if piece.startswith('</'):
- depth -= 1
- if depth == 0:
- break
- elif piece.startswith('<') and not piece.endswith('/>'):
- depth += 1
- else:
- pieces = pieces[1:-1]
- output = ''.join(pieces)
- if strip_whitespace:
- output = output.strip()
- if not expecting_text:
- return output
- # decode base64 content
- if base64 and self.contentparams.get('base64', 0):
- try:
- output = base64.decodebytes(output.encode('utf8')).decode('utf8')
- except (binascii.Error, binascii.Incomplete, UnicodeDecodeError):
- pass
- # resolve relative URIs
- if (element in self.can_be_relative_uri) and output:
- # do not resolve guid elements with isPermalink="false"
- if not element == 'id' or self.guidislink:
- output = self.resolve_uri(output)
- # decode entities within embedded markup
- if not self.contentparams.get('base64', 0):
- output = self.decode_entities(element, output)
- # some feed formats require consumers to guess
- # whether the content is html or plain text
- if not self.version.startswith('atom') and self.contentparams.get('type') == 'text/plain':
- if self.looks_like_html(output):
- self.contentparams['type'] = 'text/html'
- # remove temporary cruft from contentparams
- try:
- del self.contentparams['mode']
- except KeyError:
- pass
- try:
- del self.contentparams['base64']
- except KeyError:
- pass
- is_htmlish = self.map_content_type(self.contentparams.get('type', 'text/html')) in self.html_types
- # resolve relative URIs within embedded markup
- if is_htmlish and self.resolve_relative_uris:
- if element in self.can_contain_relative_uris:
- output = resolve_relative_uris(output, self.baseuri, self.encoding, self.contentparams.get('type', 'text/html'))
- # sanitize embedded markup
- if is_htmlish and self.sanitize_html:
- if element in self.can_contain_dangerous_markup:
- output = _sanitize_html(output, self.encoding, self.contentparams.get('type', 'text/html'))
- if self.encoding and isinstance(output, bytes):
- output = output.decode(self.encoding, 'ignore')
- # address common error where people take data that is already
- # utf-8, presume that it is iso-8859-1, and re-encode it.
- if self.encoding in ('utf-8', 'utf-8_INVALID_PYTHON_3') and not isinstance(output, bytes):
- try:
- output = output.encode('iso-8859-1').decode('utf-8')
- except (UnicodeEncodeError, UnicodeDecodeError):
- pass
- # map win-1252 extensions to the proper code points
- if not isinstance(output, bytes):
- output = output.translate(_cp1252)
- # categories/tags/keywords/whatever are handled in _end_category or
- # _end_tags or _end_itunes_keywords
- if element in ('category', 'tags', 'itunes_keywords'):
- return output
- if element == 'title' and -1 < self.title_depth <= self.depth:
- return output
- # store output in appropriate place(s)
- if self.inentry and not self.insource:
- if element == 'content':
- self.entries[-1].setdefault(element, [])
- contentparams = copy.deepcopy(self.contentparams)
- contentparams['value'] = output
- self.entries[-1][element].append(contentparams)
- elif element == 'link':
- if not self.inimage:
- # query variables in urls in link elements are improperly
- # converted from `?a=1&b=2` to `?a=1&b;=2` as if they're
- # unhandled character references. fix this special case.
- output = output.replace('&', '&')
- output = re.sub("&([A-Za-z0-9_]+);", r"&\g<1>", output)
- self.entries[-1][element] = output
- if output:
- self.entries[-1]['links'][-1]['href'] = output
- else:
- if element == 'description':
- element = 'summary'
- old_value_depth = self.property_depth_map.setdefault(self.entries[-1], {}).get(element)
- if old_value_depth is None or self.depth <= old_value_depth:
- self.property_depth_map[self.entries[-1]][element] = self.depth
- self.entries[-1][element] = output
- if self.incontent:
- contentparams = copy.deepcopy(self.contentparams)
- contentparams['value'] = output
- self.entries[-1][element + '_detail'] = contentparams
- elif self.infeed or self.insource: # and (not self.intextinput) and (not self.inimage):
- context = self._get_context()
- if element == 'description':
- element = 'subtitle'
- context[element] = output
- if element == 'link':
- # fix query variables; see above for the explanation
- output = re.sub("&([A-Za-z0-9_]+);", r"&\g<1>", output)
- context[element] = output
- context['links'][-1]['href'] = output
- elif self.incontent:
- contentparams = copy.deepcopy(self.contentparams)
- contentparams['value'] = output
- context[element + '_detail'] = contentparams
- return output
- def push_content(self, tag, attrs_d, default_content_type, expecting_text):
- self.incontent += 1
- if self.lang:
- self.lang = self.lang.replace('_', '-')
- self.contentparams = FeedParserDict({
- 'type': self.map_content_type(attrs_d.get('type', default_content_type)),
- 'language': self.lang,
- 'base': self.baseuri})
- self.contentparams['base64'] = self._is_base64(attrs_d, self.contentparams)
- self.push(tag, expecting_text)
- def pop_content(self, tag):
- value = self.pop(tag)
- self.incontent -= 1
- self.contentparams.clear()
- return value
- # a number of elements in a number of RSS variants are nominally plain
- # text, but this is routinely ignored. This is an attempt to detect
- # the most common cases. As false positives often result in silent
- # data loss, this function errs on the conservative side.
- @staticmethod
- def looks_like_html(s):
- """
- :type s: str
- :rtype: bool
- """
- # must have a close tag or an entity reference to qualify
- if not (re.search(r'</(\w+)>', s) or re.search(r'&#?\w+;', s)):
- return False
- # all tags must be in a restricted subset of valid HTML tags
- if any((t for t in re.findall(r'</?(\w+)', s) if t.lower() not in _HTMLSanitizer.acceptable_elements)):
- return False
- # all entities must have been defined as valid HTML entities
- if any((e for e in re.findall(r'&(\w+);', s) if e not in html.entities.entitydefs)):
- return False
- return True
- def _map_to_standard_prefix(self, name):
- colonpos = name.find(':')
- if colonpos != -1:
- prefix = name[:colonpos]
- suffix = name[colonpos+1:]
- prefix = self.namespacemap.get(prefix, prefix)
- name = prefix + ':' + suffix
- return name
- def _get_attribute(self, attrs_d, name):
- return attrs_d.get(self._map_to_standard_prefix(name))
- def _is_base64(self, attrs_d, contentparams):
- if attrs_d.get('mode', '') == 'base64':
- return 1
- if self.contentparams['type'].startswith('text/'):
- return 0
- if self.contentparams['type'].endswith('+xml'):
- return 0
- if self.contentparams['type'].endswith('/xml'):
- return 0
- return 1
- @staticmethod
- def _enforce_href(attrs_d):
- href = attrs_d.get('url', attrs_d.get('uri', attrs_d.get('href', None)))
- if href:
- try:
- del attrs_d['url']
- except KeyError:
- pass
- try:
- del attrs_d['uri']
- except KeyError:
- pass
- attrs_d['href'] = href
- return attrs_d
- def _save(self, key, value, overwrite=False):
- context = self._get_context()
- if overwrite:
- context[key] = value
- else:
- context.setdefault(key, value)
- def _get_context(self):
- if self.insource:
- context = self.sourcedata
- elif self.inimage and 'image' in self.feeddata:
- context = self.feeddata['image']
- elif self.intextinput:
- context = self.feeddata['textinput']
- elif self.inentry:
- context = self.entries[-1]
- else:
- context = self.feeddata
- return context
- def _save_author(self, key, value, prefix='author'):
- context = self._get_context()
- context.setdefault(prefix + '_detail', FeedParserDict())
- context[prefix + '_detail'][key] = value
- self._sync_author_detail()
- context.setdefault('authors', [FeedParserDict()])
- context['authors'][-1][key] = value
- def _save_contributor(self, key, value):
- context = self._get_context()
- context.setdefault('contributors', [FeedParserDict()])
- context['contributors'][-1][key] = value
- def _sync_author_detail(self, key='author'):
- context = self._get_context()
- detail = context.get('%ss' % key, [FeedParserDict()])[-1]
- if detail:
- name = detail.get('name')
- email = detail.get('email')
- if name and email:
- context[key] = '%s (%s)' % (name, email)
- elif name:
- context[key] = name
- elif email:
- context[key] = email
- else:
- author, email = context.get(key), None
- if not author:
- return
- emailmatch = re.search(r'''(([a-zA-Z0-9\_\-\.\+]+)@((\[[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.)|(([a-zA-Z0-9\-]+\.)+))([a-zA-Z]{2,4}|[0-9]{1,3})(\]?))(\?subject=\S+)?''', author)
- if emailmatch:
- email = emailmatch.group(0)
- # probably a better way to do the following, but it passes
- # all the tests
- author = author.replace(email, '')
- author = author.replace('()', '')
- author = author.replace('<>', '')
- author = author.replace('<>', '')
- author = author.strip()
- if author and (author[0] == '('):
- author = author[1:]
- if author and (author[-1] == ')'):
- author = author[:-1]
- author = author.strip()
- if author or email:
- context.setdefault('%s_detail' % key, detail)
- if author:
- detail['name'] = author
- if email:
- detail['email'] = email
- def _add_tag(self, term, scheme, label):
- context = self._get_context()
- tags = context.setdefault('tags', [])
- if (not term) and (not scheme) and (not label):
- return
- value = FeedParserDict(term=term, scheme=scheme, label=label)
- if value not in tags:
- tags.append(value)
- def _start_tags(self, attrs_d):
- # This is a completely-made up element. Its semantics are determined
- # only by a single feed that precipitated bug report 392 on Google Code.
- # In short, this is junk code.
- self.push('tags', 1)
- def _end_tags(self):
- for term in self.pop('tags').split(','):
- self._add_tag(term.strip(), None, None)
|