Compare commits
25 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a8c7e6a972 | ||
|
|
b30c375c5b | ||
|
|
cec5acf58f | ||
|
|
24d0f2d00a | ||
|
|
94007b0b92 | ||
|
|
1a5548f171 | ||
|
|
53c49b9121 | ||
|
|
bd50872043 | ||
|
|
d37c4fd551 | ||
|
|
d9ed7cc6d1 | ||
|
|
0a0808c0a8 | ||
|
|
16354e3528 | ||
|
|
1018e88ec1 | ||
|
|
2916351517 | ||
|
|
46d4b02c81 | ||
|
|
58eac88a10 | ||
|
|
2ef3d8dfbe | ||
|
|
7cf4c29340 | ||
|
|
cdd84563dd | ||
|
|
8138ea9a60 | ||
|
|
c171f9a875 | ||
|
|
3f97a8b8ff | ||
|
|
1147767ff3 | ||
|
|
afff08b017 | ||
|
|
238a5de5cc |
20
.build/Dockerfile
Normal file
20
.build/Dockerfile
Normal file
@@ -0,0 +1,20 @@
|
||||
FROM python:3.9-slim-buster AS deps
|
||||
|
||||
RUN apt-get update && \
|
||||
apt-get install -y build-essential git curl python3-dev libatlas3-base libatlas-base-dev liblapack-dev libxml2 libxml2-dev libffi6 libffi-dev musl-dev libxslt-dev
|
||||
|
||||
FROM deps AS testable
|
||||
ARG REPORT_PATH
|
||||
|
||||
VOLUME ["/var/mailgun", "/etc/mailgun/ssl", ${REPORT_PATH}]
|
||||
|
||||
ADD . /app
|
||||
WORKDIR /app
|
||||
COPY wheel/* /wheel/
|
||||
|
||||
RUN mkdir -p ${REPORT_PATH}
|
||||
|
||||
RUN python ./setup.py build bdist_wheel -d /wheel && \
|
||||
pip install --no-deps /wheel/*
|
||||
|
||||
ENTRYPOINT ["/bin/sh", "/app/run_tests.sh"]
|
||||
3
.gitignore
vendored
3
.gitignore
vendored
@@ -54,3 +54,6 @@ _trial_temp
|
||||
|
||||
# OSX
|
||||
.DS_Store
|
||||
|
||||
# vim-backup
|
||||
*.bak
|
||||
|
||||
@@ -5,3 +5,10 @@ include classifier
|
||||
include LICENSE
|
||||
include MANIFEST.in
|
||||
include README.rst
|
||||
include talon/signature/data/train.data
|
||||
include talon/signature/data/classifier
|
||||
include talon/signature/data/classifier_01.npy
|
||||
include talon/signature/data/classifier_02.npy
|
||||
include talon/signature/data/classifier_03.npy
|
||||
include talon/signature/data/classifier_04.npy
|
||||
include talon/signature/data/classifier_05.npy
|
||||
|
||||
11
requirements.txt
Normal file
11
requirements.txt
Normal file
@@ -0,0 +1,11 @@
|
||||
chardet>=1.0.1
|
||||
cchardet>=0.3.5
|
||||
cssselect
|
||||
html5lib
|
||||
joblib
|
||||
lxml>=2.3.3
|
||||
numpy
|
||||
regex>=1
|
||||
scikit-learn>=1.0.0
|
||||
scipy
|
||||
six>=1.10.0
|
||||
4
run_tests.sh
Executable file
4
run_tests.sh
Executable file
@@ -0,0 +1,4 @@
|
||||
#!/usr/bin/env bash
|
||||
set -ex
|
||||
REPORT_PATH="${REPORT_PATH:-./}"
|
||||
nosetests --with-xunit --with-coverage --cover-xml --cover-xml-file $REPORT_PATH/coverage.xml --xunit-file=$REPORT_PATH/nosetests.xml --cover-package=talon .
|
||||
31
setup.py
31
setup.py
@@ -19,17 +19,17 @@ class InstallCommand(install):
|
||||
if self.no_ml:
|
||||
dist = self.distribution
|
||||
dist.packages=find_packages(exclude=[
|
||||
'tests',
|
||||
'tests.*',
|
||||
'talon.signature',
|
||||
'talon.signature.*',
|
||||
"tests",
|
||||
"tests.*",
|
||||
"talon.signature",
|
||||
"talon.signature.*",
|
||||
])
|
||||
for not_required in ['numpy', 'scipy', 'scikit-learn==0.16.1']:
|
||||
for not_required in ["numpy", "scipy", "scikit-learn==0.24.1"]:
|
||||
dist.install_requires.remove(not_required)
|
||||
|
||||
|
||||
setup(name='talon',
|
||||
version='1.4.5',
|
||||
version='1.5.0',
|
||||
description=("Mailgun library "
|
||||
"to extract message quotations and signatures."),
|
||||
long_description=open("README.rst").read(),
|
||||
@@ -44,20 +44,21 @@ setup(name='talon',
|
||||
include_package_data=True,
|
||||
zip_safe=True,
|
||||
install_requires=[
|
||||
"lxml>=2.3.3",
|
||||
"regex>=1",
|
||||
"lxml",
|
||||
"regex",
|
||||
"numpy",
|
||||
"scipy",
|
||||
"scikit-learn>=0.16.1", # pickled versions of classifier, else rebuild
|
||||
'chardet>=1.0.1',
|
||||
'cchardet>=0.3.5',
|
||||
'cssselect',
|
||||
'six>=1.10.0',
|
||||
'html5lib'
|
||||
"scikit-learn>=1.0.0",
|
||||
"chardet",
|
||||
"cchardet",
|
||||
"cssselect",
|
||||
"six",
|
||||
"html5lib",
|
||||
"joblib",
|
||||
],
|
||||
tests_require=[
|
||||
"mock",
|
||||
"nose>=1.2.1",
|
||||
"nose",
|
||||
"coverage"
|
||||
]
|
||||
)
|
||||
|
||||
@@ -87,23 +87,24 @@ def cut_gmail_quote(html_message):
|
||||
|
||||
def cut_microsoft_quote(html_message):
|
||||
''' Cuts splitter block and all following blocks. '''
|
||||
#use EXSLT extensions to have a regex match() function with lxml
|
||||
ns = {"re": "http://exslt.org/regular-expressions"}
|
||||
|
||||
#general pattern: @style='border:none;border-top:solid <color> 1.0pt;padding:3.0pt 0<unit> 0<unit> 0<unit>'
|
||||
#outlook 2007, 2010 (international) <color=#B5C4DF> <unit=cm>
|
||||
#outlook 2007, 2010 (american) <color=#B5C4DF> <unit=pt>
|
||||
#outlook 2013 (international) <color=#E1E1E1> <unit=cm>
|
||||
#outlook 2013 (american) <color=#E1E1E1> <unit=pt>
|
||||
#also handles a variant with a space after the semicolon
|
||||
splitter = html_message.xpath(
|
||||
#outlook 2007, 2010 (international)
|
||||
"//div[@style='border:none;border-top:solid #B5C4DF 1.0pt;"
|
||||
"padding:3.0pt 0cm 0cm 0cm']|"
|
||||
#outlook 2007, 2010 (american)
|
||||
"//div[@style='border:none;border-top:solid #B5C4DF 1.0pt;"
|
||||
"padding:3.0pt 0in 0in 0in']|"
|
||||
#outlook 2013 (international)
|
||||
"//div[@style='border:none;border-top:solid #E1E1E1 1.0pt;"
|
||||
"padding:3.0pt 0cm 0cm 0cm']|"
|
||||
#outlook 2013 (american)
|
||||
"//div[@style='border:none;border-top:solid #E1E1E1 1.0pt;"
|
||||
"padding:3.0pt 0in 0in 0in']|"
|
||||
#outlook 2007, 2010, 2013 (international, american)
|
||||
"//div[@style[re:match(., 'border:none; ?border-top:solid #(E1E1E1|B5C4DF) 1.0pt; ?"
|
||||
"padding:3.0pt 0(in|cm) 0(in|cm) 0(in|cm)')]]|"
|
||||
#windows mail
|
||||
"//div[@style='padding-top: 5px; "
|
||||
"border-top-color: rgb(229, 229, 229); "
|
||||
"border-top-width: 1px; border-top-style: solid;']"
|
||||
, namespaces=ns
|
||||
)
|
||||
|
||||
if splitter:
|
||||
|
||||
@@ -22,7 +22,7 @@ import six
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
|
||||
RE_FWD = re.compile("^[-]+[ ]*Forwarded message[ ]*[-]+$", re.I | re.M)
|
||||
RE_FWD = re.compile("^[-]+[ ]*Forwarded message[ ]*[-]+\s*$", re.I | re.M)
|
||||
|
||||
RE_ON_DATE_SMB_WROTE = re.compile(
|
||||
u'(-*[>]?[ ]?({0})[ ].*({1})(.*\n){{0,2}}.*({2}):?-*)'.format(
|
||||
@@ -139,13 +139,17 @@ RE_ORIGINAL_MESSAGE = re.compile(u'[\s]*[-]+[ ]*({})[ ]*[-]+'.format(
|
||||
'Oprindelig meddelelse',
|
||||
))), re.I)
|
||||
|
||||
RE_FROM_COLON_OR_DATE_COLON = re.compile(u'(_+\r?\n)?[\s]*(:?[*]?{})[\s]?:[*]?.*'.format(
|
||||
RE_FROM_COLON_OR_DATE_COLON = re.compile(u'((_+\r?\n)?[\s]*:?[*]?({})[\s]?:([^\n$]+\n){{1,2}}){{2,}}'.format(
|
||||
u'|'.join((
|
||||
# "From" in different languages.
|
||||
'From', 'Van', 'De', 'Von', 'Fra', u'Från',
|
||||
# "Date" in different languages.
|
||||
'Date', 'Datum', u'Envoyé', 'Skickat', 'Sendt',
|
||||
))), re.I)
|
||||
'Date', '[S]ent', 'Datum', u'Envoyé', 'Skickat', 'Sendt', 'Gesendet',
|
||||
# "Subject" in different languages.
|
||||
'Subject', 'Betreff', 'Objet', 'Emne', u'Ämne',
|
||||
# "To" in different languages.
|
||||
'To', 'An', 'Til', u'À', 'Till'
|
||||
))), re.I | re.M)
|
||||
|
||||
# ---- John Smith wrote ----
|
||||
RE_ANDROID_WROTE = re.compile(u'[\s]*[-]+.*({})[ ]*[-]+'.format(
|
||||
@@ -189,9 +193,6 @@ RE_PARENTHESIS_LINK = re.compile("\(https?://")
|
||||
|
||||
SPLITTER_MAX_LINES = 6
|
||||
MAX_LINES_COUNT = 1000
|
||||
# an extensive research shows that exceeding this limit
|
||||
# leads to excessive processing time
|
||||
MAX_HTML_LEN = 2794202
|
||||
|
||||
QUOT_PATTERN = re.compile('^>+ ?')
|
||||
NO_QUOT_LINE = re.compile('^[^>].*[\S].*')
|
||||
@@ -417,25 +418,31 @@ def extract_from_html(msg_body):
|
||||
|
||||
Returns a unicode string.
|
||||
"""
|
||||
msg_body_bytes = msg_body
|
||||
if isinstance(msg_body, six.text_type):
|
||||
msg_body = msg_body.encode('utf8')
|
||||
elif not isinstance(msg_body, bytes):
|
||||
msg_body = msg_body.encode('ascii')
|
||||
msg_body_bytes = msg_body.encode('utf8')
|
||||
|
||||
result = _extract_from_html(msg_body)
|
||||
if isinstance(result, bytes):
|
||||
result = result.decode('utf8')
|
||||
if msg_body_bytes.strip() == b'':
|
||||
return msg_body
|
||||
|
||||
msg_body_bytes = msg_body_bytes.replace(b'\r\n', b'\n')
|
||||
# Cut out xml and doctype tags to avoid conflict with unicode decoding.
|
||||
msg_body_bytes = re.sub(br"\<\?xml.+\?\>|\<\!DOCTYPE.+]\>", b"", msg_body_bytes)
|
||||
html_tree = html_document_fromstring(msg_body_bytes)
|
||||
if html_tree is None:
|
||||
return msg_body
|
||||
|
||||
result = extract_from_html_tree(html_tree)
|
||||
if not result:
|
||||
return msg_body
|
||||
|
||||
return result
|
||||
|
||||
|
||||
def _extract_from_html(msg_body):
|
||||
def extract_from_html_tree(html_tree):
|
||||
"""
|
||||
Extract not quoted message from provided html message body
|
||||
using tags and plain text algorithm.
|
||||
|
||||
Cut out first some encoding html tags such as xml and doctype
|
||||
for avoiding conflict with unicode decoding
|
||||
Extract not quoted message from provided parsed html tree using tags and
|
||||
plain text algorithm.
|
||||
|
||||
Cut out the 'blockquote', 'gmail_quote' tags.
|
||||
Cut Microsoft quotations.
|
||||
@@ -448,18 +455,6 @@ def _extract_from_html(msg_body):
|
||||
then checking deleted checkpoints,
|
||||
then deleting necessary tags.
|
||||
"""
|
||||
if msg_body.strip() == b'':
|
||||
return msg_body
|
||||
|
||||
msg_body = msg_body.replace(b'\r\n', b'\n')
|
||||
|
||||
msg_body = re.sub(r"\<\?xml.+\?\>|\<\!DOCTYPE.+]\>", "", msg_body)
|
||||
|
||||
html_tree = html_document_fromstring(msg_body)
|
||||
|
||||
if html_tree is None:
|
||||
return msg_body
|
||||
|
||||
cut_quotations = (html_quotations.cut_gmail_quote(html_tree) or
|
||||
html_quotations.cut_zimbra_quote(html_tree) or
|
||||
html_quotations.cut_blockquote(html_tree) or
|
||||
@@ -477,7 +472,7 @@ def _extract_from_html(msg_body):
|
||||
|
||||
# Don't process too long messages
|
||||
if len(lines) > MAX_LINES_COUNT:
|
||||
return msg_body
|
||||
return None
|
||||
|
||||
# Collect checkpoints on each line
|
||||
line_checkpoints = [
|
||||
@@ -496,7 +491,7 @@ def _extract_from_html(msg_body):
|
||||
lines_were_deleted, first_deleted, last_deleted = return_flags
|
||||
|
||||
if not lines_were_deleted and not cut_quotations:
|
||||
return msg_body
|
||||
return None
|
||||
|
||||
if lines_were_deleted:
|
||||
#collect checkpoints from deleted lines
|
||||
@@ -510,9 +505,73 @@ def _extract_from_html(msg_body):
|
||||
)
|
||||
|
||||
if _readable_text_empty(html_tree_copy):
|
||||
return msg_body
|
||||
return None
|
||||
|
||||
return html.tostring(html_tree_copy)
|
||||
# NOTE: We remove_namespaces() because we are using an HTML5 Parser, HTML
|
||||
# parsers do not recognize namespaces in HTML tags. As such the rendered
|
||||
# HTML tags are no longer recognizable HTML tags. Example: <o:p> becomes
|
||||
# <oU0003Ap>. When we port this to golang we should look into using an
|
||||
# XML Parser NOT and HTML5 Parser since we do not know what input a
|
||||
# customer will send us. Switching to a common XML parser in python
|
||||
# opens us up to a host of vulnerabilities.
|
||||
# See https://docs.python.org/3/library/xml.html#xml-vulnerabilities
|
||||
#
|
||||
# The down sides to removing the namespaces is that customers might
|
||||
# judge the XML namespaces important. If that is the case then support
|
||||
# should encourage customers to preform XML parsing of the un-stripped
|
||||
# body to get the full unmodified XML payload.
|
||||
#
|
||||
# Alternatives to this approach are
|
||||
# 1. Ignore the U0003A in tag names and let the customer deal with it.
|
||||
# This is not ideal, as most customers use stripped-html for viewing
|
||||
# emails sent from a recipient, as such they cannot control the HTML
|
||||
# provided by a recipient.
|
||||
# 2. Preform a string replace of 'U0003A' to ':' on the rendered HTML
|
||||
# string. While this would solve the issue simply, it runs the risk
|
||||
# of replacing data outside the <tag> which might be essential to
|
||||
# the customer.
|
||||
remove_namespaces(html_tree_copy)
|
||||
s = html.tostring(html_tree_copy)
|
||||
if not s:
|
||||
return None
|
||||
|
||||
return s.decode('utf-8')
|
||||
|
||||
|
||||
def remove_namespaces(root):
|
||||
"""
|
||||
Given the root of an HTML document iterate through all the elements
|
||||
and remove any namespaces that might have been provided and remove
|
||||
any attributes that contain a namespace
|
||||
|
||||
<html xmlns:o="urn:schemas-microsoft-com:office:office">
|
||||
becomes
|
||||
<html>
|
||||
|
||||
<o:p>Hi</o:p>
|
||||
becomes
|
||||
<p>Hi</p>
|
||||
|
||||
Start tags do NOT have a namespace; COLON characters have no special meaning.
|
||||
if we don't remove the namespace the parser translates the tag name into a
|
||||
unicode representation. For example <o:p> becomes <oU0003Ap>
|
||||
|
||||
See https://www.w3.org/TR/2011/WD-html5-20110525/syntax.html#start-tags
|
||||
|
||||
|
||||
"""
|
||||
for child in root.iter():
|
||||
for key, value in child.attrib.items():
|
||||
# If the attribute includes a colon
|
||||
if key.rfind("U0003A") != -1:
|
||||
child.attrib.pop(key)
|
||||
|
||||
# If the tag includes a colon
|
||||
idx = child.tag.rfind("U0003A")
|
||||
if idx != -1:
|
||||
child.tag = child.tag[idx+6:]
|
||||
|
||||
return root
|
||||
|
||||
|
||||
def split_emails(msg):
|
||||
@@ -567,7 +626,6 @@ def _correct_splitlines_in_headers(markers, lines):
|
||||
updated_markers = ""
|
||||
i = 0
|
||||
in_header_block = False
|
||||
|
||||
for m in markers:
|
||||
# Only set in_header_block flag when we hit an 's' and line is a header
|
||||
if m == 's':
|
||||
|
||||
@@ -23,17 +23,14 @@ trained against, don't forget to regenerate:
|
||||
from __future__ import absolute_import
|
||||
import os
|
||||
|
||||
from . import extraction
|
||||
from . extraction import extract #noqa
|
||||
from . learning import classifier
|
||||
|
||||
|
||||
DATA_DIR = os.path.join(os.path.dirname(__file__), 'data')
|
||||
|
||||
EXTRACTOR_FILENAME = os.path.join(DATA_DIR, 'classifier')
|
||||
EXTRACTOR_DATA = os.path.join(DATA_DIR, 'train.data')
|
||||
from talon.signature import extraction
|
||||
from talon.signature.extraction import extract
|
||||
from talon.signature.learning import classifier
|
||||
|
||||
|
||||
def initialize():
|
||||
extraction.EXTRACTOR = classifier.load(EXTRACTOR_FILENAME,
|
||||
EXTRACTOR_DATA)
|
||||
data_dir = os.path.join(os.path.dirname(__file__), 'data')
|
||||
extractor_filename = os.path.join(data_dir, 'classifier')
|
||||
extractor_data_filename = os.path.join(data_dir, 'train.data')
|
||||
extraction.EXTRACTOR = classifier.load(extractor_filename,
|
||||
extractor_data_filename)
|
||||
|
||||
1
talon/signature/data/__init__.py
Normal file
1
talon/signature/data/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
|
||||
Binary file not shown.
File diff suppressed because it is too large
Load Diff
@@ -8,7 +8,7 @@ body belongs to the signature.
|
||||
from __future__ import absolute_import
|
||||
|
||||
from numpy import genfromtxt
|
||||
from sklearn.externals import joblib
|
||||
import joblib
|
||||
from sklearn.svm import LinearSVC
|
||||
|
||||
|
||||
|
||||
@@ -102,7 +102,7 @@ def flatten_list(list_to_flatten):
|
||||
|
||||
|
||||
def contains_sender_names(sender):
|
||||
'''Returns a functions to search sender\'s name or it\'s part.
|
||||
"""Returns a functions to search sender\'s name or it\'s part.
|
||||
|
||||
>>> feature = contains_sender_names("Sergey N. Obukhov <xxx@example.com>")
|
||||
>>> feature("Sergey Obukhov")
|
||||
@@ -115,7 +115,7 @@ def contains_sender_names(sender):
|
||||
1
|
||||
>>> contains_sender_names("<serobnic@mail.ru>")("serobnic")
|
||||
1
|
||||
'''
|
||||
"""
|
||||
names = '( |$)|'.join(flatten_list([[e, e.capitalize()]
|
||||
for e in extract_names(sender)]))
|
||||
names = names or sender
|
||||
@@ -140,10 +140,16 @@ def extract_names(sender):
|
||||
sender = "".join([char if char.isalpha() else ' ' for char in sender])
|
||||
# Remove too short words and words from "black" list i.e.
|
||||
# words like `ru`, `gmail`, `com`, `org`, etc.
|
||||
sender = [word for word in sender.split() if len(word) > 1 and
|
||||
not word in BAD_SENDER_NAMES]
|
||||
# Remove duplicates
|
||||
names = list(set(sender))
|
||||
names = list()
|
||||
for word in sender.split():
|
||||
if len(word) < 2:
|
||||
continue
|
||||
if word in BAD_SENDER_NAMES:
|
||||
continue
|
||||
if word in names:
|
||||
continue
|
||||
names.append(word)
|
||||
|
||||
return names
|
||||
|
||||
|
||||
@@ -208,20 +214,26 @@ def many_capitalized_words(s):
|
||||
|
||||
|
||||
def has_signature(body, sender):
|
||||
'''Checks if the body has signature. Returns True or False.'''
|
||||
"""Checks if the body has signature. Returns True or False."""
|
||||
non_empty = [line for line in body.splitlines() if line.strip()]
|
||||
candidate = non_empty[-SIGNATURE_MAX_LINES:]
|
||||
upvotes = 0
|
||||
sender_check = contains_sender_names(sender)
|
||||
for line in candidate:
|
||||
# we check lines for sender's name, phone, email and url,
|
||||
# those signature lines don't take more then 27 lines
|
||||
if len(line.strip()) > 27:
|
||||
continue
|
||||
elif contains_sender_names(sender)(line):
|
||||
|
||||
if sender_check(line):
|
||||
return True
|
||||
elif (binary_regex_search(RE_RELAX_PHONE)(line) +
|
||||
binary_regex_search(RE_EMAIL)(line) +
|
||||
binary_regex_search(RE_URL)(line) == 1):
|
||||
|
||||
if (binary_regex_search(RE_RELAX_PHONE)(line) +
|
||||
binary_regex_search(RE_EMAIL)(line) +
|
||||
binary_regex_search(RE_URL)(line) == 1):
|
||||
upvotes += 1
|
||||
|
||||
if upvotes > 1:
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
@@ -131,7 +131,7 @@ def html_tree_to_text(tree):
|
||||
for el in tree.iter():
|
||||
el_text = (el.text or '') + (el.tail or '')
|
||||
if len(el_text) > 1:
|
||||
if el.tag in _BLOCKTAGS:
|
||||
if el.tag in _BLOCKTAGS + _HARDBREAKS:
|
||||
text += "\n"
|
||||
if el.tag == 'li':
|
||||
text += " * "
|
||||
@@ -142,7 +142,8 @@ def html_tree_to_text(tree):
|
||||
if href:
|
||||
text += "(%s) " % href
|
||||
|
||||
if el.tag in _HARDBREAKS and text and not text.endswith("\n"):
|
||||
if (el.tag in _HARDBREAKS and text and
|
||||
not text.endswith("\n") and not el_text):
|
||||
text += "\n"
|
||||
|
||||
retval = _rm_excessive_newlines(text)
|
||||
@@ -179,9 +180,6 @@ def html_fromstring(s):
|
||||
if isinstance(s, six.text_type):
|
||||
s = s.encode('utf8')
|
||||
try:
|
||||
if html_too_big(s):
|
||||
return None
|
||||
|
||||
return html5parser.fromstring(s, parser=_html5lib_parser())
|
||||
except Exception:
|
||||
pass
|
||||
@@ -193,9 +191,6 @@ def html_document_fromstring(s):
|
||||
if isinstance(s, six.text_type):
|
||||
s = s.encode('utf8')
|
||||
try:
|
||||
if html_too_big(s):
|
||||
return None
|
||||
|
||||
return html5parser.document_fromstring(s, parser=_html5lib_parser())
|
||||
except Exception:
|
||||
pass
|
||||
@@ -205,12 +200,6 @@ def cssselect(expr, tree):
|
||||
return CSSSelector(expr)(tree)
|
||||
|
||||
|
||||
def html_too_big(s):
|
||||
if isinstance(s, six.text_type):
|
||||
s = s.encode('utf8')
|
||||
return s.count(b'<') > _MAX_TAGS_COUNT
|
||||
|
||||
|
||||
def _contains_charset_spec(s):
|
||||
"""Return True if the first 4KB contain charset spec
|
||||
"""
|
||||
@@ -257,7 +246,3 @@ _BLOCKTAGS = ['div', 'p', 'ul', 'li', 'h1', 'h2', 'h3']
|
||||
_HARDBREAKS = ['br', 'hr', 'tr']
|
||||
|
||||
_RE_EXCESSIVE_NEWLINES = re.compile("\n{2,10}")
|
||||
|
||||
# an extensive research shows that exceeding this limit
|
||||
# might lead to excessive processing time
|
||||
_MAX_TAGS_COUNT = 419
|
||||
|
||||
3
test-requirements.txt
Normal file
3
test-requirements.txt
Normal file
@@ -0,0 +1,3 @@
|
||||
coverage
|
||||
mock
|
||||
nose>=1.2.1
|
||||
@@ -8,6 +8,7 @@ import re
|
||||
from talon import quotations, utils as u
|
||||
from . import *
|
||||
from .fixtures import *
|
||||
from lxml import html
|
||||
|
||||
RE_WHITESPACE = re.compile("\s")
|
||||
RE_DOUBLE_WHITESPACE = re.compile("\s")
|
||||
@@ -390,18 +391,6 @@ def test_gmail_forwarded_msg():
|
||||
eq_(RE_WHITESPACE.sub('', msg_body), RE_WHITESPACE.sub('', extracted))
|
||||
|
||||
|
||||
@patch.object(u, '_MAX_TAGS_COUNT', 4)
|
||||
def test_too_large_html():
|
||||
msg_body = 'Reply' \
|
||||
'<div class="gmail_quote">' \
|
||||
'<div class="gmail_quote">On 11-Apr-2011, at 6:54 PM, Bob <bob@example.com> wrote:' \
|
||||
'<div>Test</div>' \
|
||||
'</div>' \
|
||||
'</div>'
|
||||
eq_(RE_WHITESPACE.sub('', msg_body),
|
||||
RE_WHITESPACE.sub('', quotations.extract_from_html(msg_body)))
|
||||
|
||||
|
||||
def test_readable_html_empty():
|
||||
msg_body = """
|
||||
<blockquote>
|
||||
@@ -424,3 +413,23 @@ def test_readable_html_empty():
|
||||
def test_bad_html():
|
||||
bad_html = "<html></html>"
|
||||
eq_(bad_html, quotations.extract_from_html(bad_html))
|
||||
|
||||
|
||||
def test_remove_namespaces():
|
||||
msg_body = """
|
||||
<html xmlns:o="urn:schemas-microsoft-com:office:office" xmlns="http://www.w3.org/TR/REC-html40">
|
||||
<body>
|
||||
<o:p>Dear Sir,</o:p>
|
||||
<o:p>Thank you for the email.</o:p>
|
||||
<blockquote>thing</blockquote>
|
||||
</body>
|
||||
</html>
|
||||
"""
|
||||
|
||||
rendered = quotations.extract_from_html(msg_body)
|
||||
|
||||
assert_true("<p>" in rendered)
|
||||
assert_true("xmlns" in rendered)
|
||||
|
||||
assert_true("<o:p>" not in rendered)
|
||||
assert_true("<xmlns:o>" not in rendered)
|
||||
|
||||
@@ -453,7 +453,8 @@ def test_link_closed_with_quotation_marker_on_new_line():
|
||||
msg_body = '''8.45am-1pm
|
||||
|
||||
From: somebody@example.com
|
||||
|
||||
Date: Wed, 16 May 2012 00:15:02 -0600
|
||||
|
||||
<http://email.example.com/c/dHJhY2tpbmdfY29kZT1mMDdjYzBmNzM1ZjYzMGIxNT
|
||||
> <bob@example.com <mailto:bob@example.com> >
|
||||
|
||||
@@ -494,7 +495,9 @@ def test_from_block_starts_with_date():
|
||||
msg_body = """Blah
|
||||
|
||||
Date: Wed, 16 May 2012 00:15:02 -0600
|
||||
To: klizhentas@example.com"""
|
||||
To: klizhentas@example.com
|
||||
|
||||
"""
|
||||
eq_('Blah', quotations.extract_from_plain(msg_body))
|
||||
|
||||
|
||||
@@ -564,11 +567,12 @@ def test_mark_message_lines():
|
||||
# next line should be marked as splitter
|
||||
'_____________',
|
||||
'From: foo@bar.com',
|
||||
'Date: Wed, 16 May 2012 00:15:02 -0600',
|
||||
'',
|
||||
'> Hi',
|
||||
'',
|
||||
'Signature']
|
||||
eq_('tessemet', quotations.mark_message_lines(lines))
|
||||
eq_('tesssemet', quotations.mark_message_lines(lines))
|
||||
|
||||
lines = ['Just testing the email reply',
|
||||
'',
|
||||
@@ -807,7 +811,7 @@ def test_split_email():
|
||||
>
|
||||
>
|
||||
"""
|
||||
expected_markers = "stttttsttttetesetesmmmmmmssmmmmmmsmmmmmmmm"
|
||||
expected_markers = "stttttsttttetesetesmmmmmmsmmmmmmmmmmmmmmmm"
|
||||
markers = quotations.split_emails(msg)
|
||||
eq_(markers, expected_markers)
|
||||
|
||||
@@ -822,4 +826,16 @@ The user experience was unparallelled. Please continue production. I'm sending p
|
||||
that this line is intact."""
|
||||
|
||||
parsed = quotations.extract_from_plain(msg_body)
|
||||
eq_(msg_body, parsed.decode('utf8'))
|
||||
eq_(msg_body, parsed)
|
||||
|
||||
|
||||
def test_appointment_2():
|
||||
msg_body = """Invitation for an interview:
|
||||
|
||||
Date: Wednesday 3, October 2011
|
||||
Time: 7 : 00am
|
||||
Address: 130 Fox St
|
||||
|
||||
Please bring in your ID."""
|
||||
parsed = quotations.extract_from_plain(msg_body)
|
||||
eq_(msg_body, parsed)
|
||||
|
||||
@@ -125,39 +125,13 @@ def test_html_fromstring_exception():
|
||||
eq_(None, u.html_fromstring("<html></html>"))
|
||||
|
||||
|
||||
@patch.object(u, 'html_too_big', Mock())
|
||||
@patch.object(u.html5parser, 'fromstring')
|
||||
def test_html_fromstring_too_big(fromstring):
|
||||
eq_(None, u.html_fromstring("<html></html>"))
|
||||
assert_false(fromstring.called)
|
||||
|
||||
|
||||
@patch.object(u.html5parser, 'document_fromstring')
|
||||
def test_html_document_fromstring_exception(document_fromstring):
|
||||
document_fromstring.side_effect = Exception()
|
||||
eq_(None, u.html_document_fromstring("<html></html>"))
|
||||
|
||||
|
||||
@patch.object(u, 'html_too_big', Mock())
|
||||
@patch.object(u.html5parser, 'document_fromstring')
|
||||
def test_html_document_fromstring_too_big(document_fromstring):
|
||||
eq_(None, u.html_document_fromstring("<html></html>"))
|
||||
assert_false(document_fromstring.called)
|
||||
|
||||
|
||||
@patch.object(u, 'html_fromstring', Mock(return_value=None))
|
||||
def test_bad_html_to_text():
|
||||
bad_html = "one<br>two<br>three"
|
||||
eq_(None, u.html_to_text(bad_html))
|
||||
|
||||
|
||||
@patch.object(u, '_MAX_TAGS_COUNT', 3)
|
||||
def test_html_too_big():
|
||||
eq_(False, u.html_too_big("<div></div>"))
|
||||
eq_(True, u.html_too_big("<div><span>Hi</span></div>"))
|
||||
|
||||
|
||||
@patch.object(u, '_MAX_TAGS_COUNT', 3)
|
||||
def test_html_to_text():
|
||||
eq_(b"Hello", u.html_to_text("<div>Hello</div>"))
|
||||
eq_(None, u.html_to_text("<div><span>Hi</span></div>"))
|
||||
|
||||
Reference in New Issue
Block a user