Actually bump up talon's version up to 1.0.5 to match the tag.

2015-09-09 22:46:18 +02:00
33 changed files with 2758 additions and 3601 deletions
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -1,3 +1,5 @@
+recursive-include tests *
+recursive-include talon *
 recursive-exclude tests *.pyc *~
 recursive-exclude talon *.pyc *~
 include train.data
--- a/README.rst
+++ b/README.rst
@@ -95,7 +95,7 @@ classifiers. The core of machine learning algorithm lays in
 apply to a message (``featurespace.py``), how data sets are built
 (``dataset.py``), classifier’s interface (``classifier.py``).

-Currently the data used for training is taken from our personal email
+The data used for training is taken from our personal email
 conversations and from `ENRON`_ dataset. As a result of applying our set
 of features to the dataset we provide files ``classifier`` and
 ``train.data`` that don’t have any personal information but could be
@@ -116,19 +116,8 @@ or
    from talon.signature.learning.classifier import train, init
    train(init(), EXTRACTOR_DATA, EXTRACTOR_FILENAME)

-Open-source Dataset
-------------------
-
-Recently we started a `forge`_ project to create an open-source, annotated dataset of raw emails. In the project we
-used a subset of `ENRON`_ data, cleansed of private, health and financial information by `EDRM`_. At the moment over 190
-emails are annotated. Any contribution and collaboration on the project are welcome. Once the dataset is ready we plan to
-start using it for talon.
-
 .. _scikit-learn: http://scikit-learn.org
 .. _ENRON: https://www.cs.cmu.edu/~enron/
-.. _EDRM: http://www.edrm.net/resources/data-sets/edrm-enron-email-data-set
-.. _forge: https://github.com/mailgun/forge
-

 Research
 --------
--- a/setup.py
+++ b/setup.py
@@ -1,35 +1,8 @@
-from __future__ import absolute_import
 from setuptools import setup, find_packages
-from setuptools.command.install import install
-
-
-class InstallCommand(install):
-    user_options = install.user_options + [
-        ('no-ml', None, "Don't install without Machine Learning modules."),
-    ]
-
-    boolean_options = install.boolean_options + ['no-ml']
-
-    def initialize_options(self):
-        install.initialize_options(self)
-        self.no_ml = None
-
-    def finalize_options(self):
-        install.finalize_options(self)
-        if self.no_ml:
-            dist = self.distribution
-            dist.packages=find_packages(exclude=[
-                'tests',
-                'tests.*',
-                'talon.signature',
-                'talon.signature.*',
-            ])
-            for not_required in ['numpy', 'scipy', 'scikit-learn==0.16.1']:
-                dist.install_requires.remove(not_required)


 setup(name='talon',
-      version='1.3.7',
+      version='1.0.5',
      description=("Mailgun library "
                   "to extract message quotations and signatures."),
      long_description=open("README.rst").read(),
@@ -37,23 +10,16 @@ setup(name='talon',
      author_email='admin@mailgunhq.com',
      url='https://github.com/mailgun/talon',
      license='APACHE2',
-      cmdclass={
-          'install': InstallCommand,
-      },
-      packages=find_packages(exclude=['tests', 'tests.*']),
+      packages=find_packages(exclude=['tests']),
      include_package_data=True,
      zip_safe=True,
      install_requires=[
-          "lxml>=2.3.3",
+          "lxml==2.3.3",
          "regex>=1",
+          "html2text",
          "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==0.16.1", # pickled versions of classifier, else rebuild
          ],
      tests_require=[
          "mock",
--- a/talon/init.py
+++ b/talon/init.py
@@ -1,13 +1,7 @@
-from __future__ import absolute_import
 from talon.quotations import register_xpath_extensions
-try:
-    from talon import signature
-    ML_ENABLED = True
-except ImportError:
-    ML_ENABLED = False
+from talon import signature


 def init():
    register_xpath_extensions()
-    if ML_ENABLED:
-        signature.initialize()
+    signature.initialize()
--- a/talon/constants.py
+++ b/talon/constants.py
@@ -1,4 +1,3 @@
-from __future__ import absolute_import
 import regex as re


--- a/talon/html_quotations.py
+++ b/talon/html_quotations.py
@@ -3,10 +3,8 @@ The module's functions operate on message bodies trying to extract original
 messages (without quoted messages) from html
 """

-from __future__ import absolute_import
 import regex as re

-from talon.utils import cssselect 

 CHECKPOINT_PREFIX = '#!%!'
 CHECKPOINT_SUFFIX = '!%!#'
@@ -14,7 +12,6 @@ CHECKPOINT_PATTERN = re.compile(CHECKPOINT_PREFIX + '\d+' + CHECKPOINT_SUFFIX)

 # HTML quote indicators (tag ids)
 QUOTE_IDS = ['OLK_SRC_BODY_SECTION']
-RE_FWD = re.compile("^[-]+[ ]*Forwarded message[ ]*[-]+$", re.I | re.M)


 def add_checkpoint(html_note, counter):
@@ -79,8 +76,8 @@ def delete_quotation_tags(html_note, counter, quotation_checkpoints):

 def cut_gmail_quote(html_message):
    ''' Cuts the outermost block element with class gmail_quote. '''
-    gmail_quote = cssselect('div.gmail_quote', html_message)
-    if gmail_quote and (gmail_quote[0].text is None or not RE_FWD.match(gmail_quote[0].text)):
+    gmail_quote = html_message.cssselect('.gmail_quote')
+    if gmail_quote:
        gmail_quote[0].getparent().remove(gmail_quote[0])
        return True

@@ -88,12 +85,9 @@ def cut_gmail_quote(html_message):
 def cut_microsoft_quote(html_message):
    ''' Cuts splitter block and all following blocks. '''
    splitter = html_message.xpath(
-        #outlook 2007, 2010 (international)
+        #outlook 2007, 2010
        "//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']|"
        #windows mail
        "//div[@style='padding-top: 5px; "
        "border-top-color: rgb(229, 229, 229); "
@@ -136,7 +130,7 @@ def cut_microsoft_quote(html_message):
 def cut_by_id(html_message):
    found = False
    for quote_id in QUOTE_IDS:
-        quote = cssselect('#{}'.format(quote_id), html_message)
+        quote = html_message.cssselect('#{}'.format(quote_id))
        if quote:
            found = True
            quote[0].getparent().remove(quote[0])
@@ -144,14 +138,9 @@ def cut_by_id(html_message):


 def cut_blockquote(html_message):
-    ''' Cuts the last non-nested blockquote with wrapping elements.'''
-    quote = html_message.xpath(
-        '(.//blockquote)'
-        '[not(@class="gmail_quote") and not(ancestor::blockquote)]'
-        '[last()]')
-
-    if quote:
-        quote = quote[0]
+    ''' Cuts blockquote with wrapping elements. '''
+    quote = html_message.find('.//blockquote')
+    if quote is not None:
        quote.getparent().remove(quote)
        return True

@@ -165,58 +154,21 @@ def cut_from_block(html_message):

    if block:
        block = block[-1]
-        parent_div = None
        while block.getparent() is not None:
            if block.tag == 'div':
-                parent_div = block
-                break
-            block = block.getparent()
-        if parent_div is not None:
-            maybe_body = parent_div.getparent()
-            # In cases where removing this enclosing div will remove all
-            # content, we should assume the quote is not enclosed in a tag.
-            parent_div_is_all_content = (
-                maybe_body is not None and maybe_body.tag == 'body' and
-                len(maybe_body.getchildren()) == 1)
-
-            if not parent_div_is_all_content:
-                parent = block.getparent()
-                next_sibling = block.getnext()
-
-                # remove all tags after found From block
-                # (From block and quoted message are in separate divs)
-                while next_sibling is not None:
-                    parent.remove(block)
-                    block = next_sibling
-                    next_sibling = block.getnext()
-
-                # remove the last sibling (or the
-                # From block if no siblings)
-                if block is not None:
-                    parent.remove(block)
-
+                block.getparent().remove(block)
                return True
-        else:
-            return False
-
-    # handle the case when From: block goes right after e.g. <hr>
-    # and not enclosed in some tag
-    block = html_message.xpath(
-        ("//*[starts-with(mg:tail(), 'From:')]|"
-         "//*[starts-with(mg:tail(), 'Date:')]"))
-    if block:
-        block = block[0]
-
-        if RE_FWD.match(block.getparent().text or ''):
-            return False
-        
-        while(block.getnext() is not None):
-            block.getparent().remove(block.getnext())
-        block.getparent().remove(block)
-        return True
-
-def cut_zimbra_quote(html_message):
-    zDivider = html_message.xpath('//hr[@data-marker="__DIVIDER__"]')
-    if zDivider:
-        zDivider[0].getparent().remove(zDivider[0])
-        return True
+            else:
+                block = block.getparent()
+    else:
+        # handle the case when From: block goes right after e.g. <hr>
+        # and not enclosed in some tag
+        block = html_message.xpath(
+            ("//*[starts-with(mg:tail(), 'From:')]|"
+             "//*[starts-with(mg:tail(), 'Date:')]"))
+        if block:
+            block = block[0]
+            while(block.getnext() is not None):
+                block.getparent().remove(block.getnext())
+            block.getparent().remove(block)
+            return True
--- a/talon/quotations.py
+++ b/talon/quotations.py
@@ -5,18 +5,15 @@ The module's functions operate on message bodies trying to extract
 original messages (without quoted messages)
 """

-from __future__ import absolute_import
 import regex as re
 import logging
 from copy import deepcopy

 from lxml import html, etree
+import html2text

-from talon.utils import (get_delimiter, html_tree_to_text,
-                         html_document_fromstring)
+from talon.utils import get_delimiter
 from talon import html_quotations
-from six.moves import range
-import six


 log = logging.getLogger(__name__)
@@ -25,7 +22,7 @@ log = logging.getLogger(__name__)
 RE_FWD = re.compile("^[-]+[ ]*Forwarded message[ ]*[-]+$", re.I | re.M)

 RE_ON_DATE_SMB_WROTE = re.compile(
-    u'(-*[>]?[ ]?({0})[ ].*({1})(.*\n){{0,2}}.*({2}):?-*)'.format(
+    u'(-*[ ]?({0})[ ].*({1})(.*\n){{0,2}}.*({2}):?-*)'.format(
        # Beginning of the line
        u'|'.join((
            # English
@@ -35,13 +32,7 @@ RE_ON_DATE_SMB_WROTE = re.compile(
            # Polish
            'W dniu',
            # Dutch
-            'Op',
-            # German
-            'Am',
-            # Norwegian
-            u'På',
-            # Swedish, Danish
-            'Den',
+            'Op'
        )),
        # Date and sender separator
        u'|'.join((
@@ -59,28 +50,18 @@ RE_ON_DATE_SMB_WROTE = re.compile(
            # Polish
            u'napisał',
            # Dutch
-            'schreef','verzond','geschreven',
-            # German
-            'schrieb',
-            # Norwegian, Swedish
-            'skrev',
+            'schreef','verzond','geschreven'
        ))
    ))
 # Special case for languages where text is translated like this: 'on {date} wrote {somebody}:'
 RE_ON_DATE_WROTE_SMB = re.compile(
-    u'(-*[>]?[ ]?({0})[ ].*(.*\n){{0,2}}.*({1})[ ]*.*:)'.format(
+    u'(-*[ ]?({0})[ ].*(.*\n){{0,2}}.*({1})[ ].*:)'.format(
        # Beginning of the line
-        u'|'.join((
        	'Op',
-        	#German
-        	'Am'
-        )),
        # Ending of the line
        u'|'.join((
            # Dutch
-            'schreef','verzond','geschreven',
-            # German
-            'schrieb'
+            'schreef','verzond','geschreven'
        ))
    )
    )
@@ -111,7 +92,7 @@ RE_EMPTY_QUOTATION = re.compile(
    (
        # quotation border: splitter line or a number of quotation marker lines
        (?:
-            (?:se*)+
+            s
            |
            (?:me*){2,}
        )
@@ -131,66 +112,37 @@ 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]?:[*]? .*'.format(
    u'|'.join((
        # "From" in different languages.
-        'From', 'Van', 'De', 'Von', 'Fra', u'Från',
+        'From', 'Van', 'De', 'Von', 'Fra',
        # "Date" in different languages.
-        'Date', 'Datum', u'Envoyé', 'Skickat', 'Sendt',
+        'Date', 'Datum', u'Envoyé'
    ))), re.I)

-# ---- John Smith wrote ----
-RE_ANDROID_WROTE = re.compile(u'[\s]*[-]+.*({})[ ]*[-]+'.format(
-    u'|'.join((
-        # English
-        'wrote'
-    ))), re.I)
-
-# Support polymail.io reply format
-# On Tue, Apr 11, 2017 at 10:07 PM John Smith
-#
-# <
-# mailto:John Smith <johnsmith@gmail.com>
-# > wrote:
-RE_POLYMAIL = re.compile('On.*\s{2}<\smailto:.*\s> wrote:', re.I)
-
 SPLITTER_PATTERNS = [
    RE_ORIGINAL_MESSAGE,
+    # <date> <person>
+    re.compile("(\d+/\d+/\d+|\d+\.\d+\.\d+).*@", re.VERBOSE),
    RE_ON_DATE_SMB_WROTE,
    RE_ON_DATE_WROTE_SMB,
    RE_FROM_COLON_OR_DATE_COLON,
-    # 02.04.2012 14:20 пользователь "bob@example.com" <
-    # bob@xxx.mailgun.org> написал:
-    re.compile("(\d+/\d+/\d+|\d+\.\d+\.\d+).*@", re.S),
-    # 2014-10-17 11:28 GMT+03:00 Bob <
-    # bob@example.com>:
-    re.compile("\d{4}-\d{2}-\d{2}\s+\d{2}:\d{2}\s+GMT.*@", re.S),
-    # Thu, 26 Jun 2014 14:00:51 +0400 Bob <bob@example.com>:
    re.compile('\S{3,10}, \d\d? \S{3,10} 20\d\d,? \d\d?:\d\d(:\d\d)?'
-               '( \S+){3,6}@\S+:'),
-    # Sent from Samsung MobileName <address@example.com> wrote:
-    re.compile('Sent from Samsung .*@.*> wrote'),
-    RE_ANDROID_WROTE,
-    RE_POLYMAIL
+               '( \S+){3,6}@\S+:')
    ]

+
 RE_LINK = re.compile('<(http://[^>]*)>')
 RE_NORMALIZED_LINK = re.compile('@@(http://[^>@]*)@@')

 RE_PARENTHESIS_LINK = re.compile("\(https?://")

-SPLITTER_MAX_LINES = 6
+SPLITTER_MAX_LINES = 4
 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].*')

-# Regular expression to identify if a line is a header.
-RE_HEADER = re.compile(": ")
-

 def extract_from(msg_body, content_type='text/plain'):
    try:
@@ -204,19 +156,6 @@ def extract_from(msg_body, content_type='text/plain'):
    return msg_body


-def remove_initial_spaces_and_mark_message_lines(lines):
-    """
-    Removes the initial spaces in each line before marking message lines.
-
-    This ensures headers can be identified if they are indented with spaces.
-    """
-    i = 0
-    while i < len(lines):
-        lines[i] = lines[i].lstrip(' ')
-        i += 1
-    return mark_message_lines(lines)
-
-
 def mark_message_lines(lines):
    """Mark message lines with markers to distinguish quotation lines.

@@ -230,7 +169,7 @@ def mark_message_lines(lines):
    >>> mark_message_lines(['answer', 'From: foo@bar.com', '', '> question'])
    'tsem'
    """
-    markers = ['e' for _ in lines]
+    markers = bytearray(len(lines))
    i = 0
    while i < len(lines):
        if not lines[i].strip():
@@ -242,11 +181,10 @@ def mark_message_lines(lines):
        else:
            # in case splitter is spread across several lines
            splitter = is_splitter('\n'.join(lines[i:i + SPLITTER_MAX_LINES]))
-
            if splitter:
                # append as many splitter markers as lines in splitter
                splitter_lines = splitter.group().splitlines()
-                for j in range(len(splitter_lines)):
+                for j in xrange(len(splitter_lines)):
                    markers[i + j] = 's'

                # skip splitter lines
@@ -256,7 +194,7 @@ def mark_message_lines(lines):
                markers[i] = 't'
        i += 1

-    return ''.join(markers)
+    return markers


 def process_marked_lines(lines, markers, return_flags=[False, -1, -1]):
@@ -270,7 +208,6 @@ def process_marked_lines(lines, markers, return_flags=[False, -1, -1]):
    return_flags = [were_lines_deleted, first_deleted_line,
                    last_deleted_line]
    """
-    markers = ''.join(markers)
    # if there are no splitter there should be no markers
    if 's' not in markers and not re.search('(me*){3}', markers):
        markers = markers.replace('m', 't')
@@ -316,27 +253,10 @@ def preprocess(msg_body, delimiter, content_type='text/plain'):
    Replaces link brackets so that they couldn't be taken for quotation marker.
    Splits line in two if splitter pattern preceded by some text on the same
    line (done only for 'On <date> <person> wrote:' pattern).
-
-    Converts msg_body into a unicode.
    """
-    msg_body = _replace_link_brackets(msg_body)
-
-    msg_body = _wrap_splitter_with_newline(msg_body, delimiter, content_type)
-
-    return msg_body
-
-
-def _replace_link_brackets(msg_body):
-    """
-    Normalize links i.e. replace '<', '>' wrapping the link with some symbols
-    so that '>' closing the link couldn't be mistakenly taken for quotation
-    marker.
-
-    Converts msg_body into a unicode
-    """
-    if isinstance(msg_body, bytes):
-        msg_body = msg_body.decode('utf8')
-
+    # normalize links i.e. replace '<', '>' wrapping the link with some symbols
+    # so that '>' closing the link couldn't be mistakenly taken for quotation
+    # marker.
    def link_wrapper(link):
        newline_index = msg_body[:link.start()].rfind("\n")
        if msg_body[newline_index + 1] == ">":
@@ -345,14 +265,7 @@ def _replace_link_brackets(msg_body):
            return "@@%s@@" % link.group(1)

    msg_body = re.sub(RE_LINK, link_wrapper, msg_body)
-    return msg_body

-
-def _wrap_splitter_with_newline(msg_body, delimiter, content_type='text/plain'):
-    """
-    Splits line in two if splitter pattern preceded by some text on the same
-    line (done only for 'On <date> <person> wrote:' pattern.
-    """
    def splitter_wrapper(splitter):
        """Wraps splitter with new line"""
        if splitter.start() and msg_body[splitter.start() - 1] != '\n':
@@ -380,8 +293,12 @@ def extract_from_plain(msg_body):

    delimiter = get_delimiter(msg_body)
    msg_body = preprocess(msg_body, delimiter)
+    lines = msg_body.splitlines()
+
    # don't process too long messages
-    lines = msg_body.splitlines()[:MAX_LINES_COUNT]
+    if len(lines) > MAX_LINES_COUNT:
+        return stripped_text
+
    markers = mark_message_lines(lines)
    lines = process_marked_lines(lines, markers)

@@ -406,59 +323,44 @@ def extract_from_html(msg_body):
    then extracting quotations from text,
    then checking deleted checkpoints,
    then deleting necessary tags.
-
-    Returns a unicode string.
    """
-    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')

-    result = _extract_from_html(msg_body)
-    if isinstance(result, bytes):
-        result = result.decode('utf8')
-
-    return result
-
-
-def _extract_from_html(msg_body):
-    """
-    Extract not quoted message from provided html message body
-    using tags and plain text algorithm.
-
-    Cut out the 'blockquote', 'gmail_quote' tags.
-    Cut Microsoft quotations.
-
-    Then use plain text algorithm to cut out splitter or
-    leftover quotation.
-    This works by adding checkpoint text to all html tags,
-    then converting html to text,
-    then extracting quotations from text,
-    then checking deleted checkpoints,
-    then deleting necessary tags.
-    """
-    if msg_body.strip() == b'':
+    if msg_body.strip() == '':
        return msg_body

-    msg_body = msg_body.replace(b'\r\n', b'\n')
-    html_tree = html_document_fromstring(msg_body)
-
-    if html_tree is None:
-        return msg_body
+    html_tree = html.document_fromstring(
+        msg_body,
+        parser=html.HTMLParser(encoding="utf-8")
+    )

    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
                      html_quotations.cut_microsoft_quote(html_tree) or
                      html_quotations.cut_by_id(html_tree) or
                      html_quotations.cut_from_block(html_tree)
                      )
+
    html_tree_copy = deepcopy(html_tree)

    number_of_checkpoints = html_quotations.add_checkpoint(html_tree, 0)
    quotation_checkpoints = [False] * number_of_checkpoints
-    plain_text = html_tree_to_text(html_tree)
-    plain_text = preprocess(plain_text, '\n', content_type='text/html')
+    msg_with_checkpoints = html.tostring(html_tree)
+
+    h = html2text.HTML2Text()
+    h.body_width = 0  # generate plain text without wrap
+
+    # html2text adds unnecessary star symbols. Remove them.
+    # Mask star symbols
+    msg_with_checkpoints = msg_with_checkpoints.replace('*', '3423oorkg432')
+    plain_text = h.handle(msg_with_checkpoints)
+    # Remove created star symbols
+    plain_text = plain_text.replace('*', '')
+    # Unmask saved star symbols
+    plain_text = plain_text.replace('3423oorkg432', '*')
+
+    delimiter = get_delimiter(plain_text)
+
+    plain_text = preprocess(plain_text, delimiter, content_type='text/html')
    lines = plain_text.splitlines()

    # Don't process too long messages
@@ -481,106 +383,25 @@ def _extract_from_html(msg_body):
    process_marked_lines(lines, markers, return_flags)
    lines_were_deleted, first_deleted, last_deleted = return_flags

-    if not lines_were_deleted and not cut_quotations:
-        return msg_body
-
    if lines_were_deleted:
        #collect checkpoints from deleted lines
-        for i in range(first_deleted, last_deleted):
+        for i in xrange(first_deleted, last_deleted):
            for checkpoint in line_checkpoints[i]:
                quotation_checkpoints[checkpoint] = True
+    else:
+        if cut_quotations:
+            return html.tostring(html_tree_copy)
+        else:
+            return msg_body

-        # Remove tags with quotation checkpoints
-        html_quotations.delete_quotation_tags(
-            html_tree_copy, 0, quotation_checkpoints
-        )
-
-    if _readable_text_empty(html_tree_copy):
-        return msg_body
+    # Remove tags with quotation checkpoints
+    html_quotations.delete_quotation_tags(
+        html_tree_copy, 0, quotation_checkpoints
+    )

    return html.tostring(html_tree_copy)


-def split_emails(msg):
-    """
-    Given a message (which may consist of an email conversation thread with
-    multiple emails), mark the lines to identify split lines, content lines and
-    empty lines.
-
-    Correct the split line markers inside header blocks. Header blocks are
-    identified by the regular expression RE_HEADER.
-
-    Return the corrected markers
-    """
-    msg_body = _replace_link_brackets(msg)
-
-    # don't process too long messages
-    lines = msg_body.splitlines()[:MAX_LINES_COUNT]
-    markers = remove_initial_spaces_and_mark_message_lines(lines)
-
-    markers = _mark_quoted_email_splitlines(markers, lines)
-
-    # we don't want splitlines in header blocks
-    markers = _correct_splitlines_in_headers(markers, lines)
-
-    return markers
-
-
-def _mark_quoted_email_splitlines(markers, lines):
-    """
-    When there are headers indented with '>' characters, this method will
-    attempt to identify if the header is a splitline header. If it is, then we
-    mark it with 's' instead of leaving it as 'm' and return the new markers.
-    """
-    # Create a list of markers to easily alter specific characters
-    markerlist = list(markers)
-    for i, line in enumerate(lines):
-        if markerlist[i] != 'm':
-            continue
-        for pattern in SPLITTER_PATTERNS:
-            matcher = re.search(pattern, line)
-            if matcher:
-                markerlist[i] = 's'
-                break
-
-    return "".join(markerlist)
-
-
-def _correct_splitlines_in_headers(markers, lines):
-    """
-    Corrects markers by removing splitlines deemed to be inside header blocks.
-    """
-    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':
-            if not in_header_block:
-                if bool(re.search(RE_HEADER, lines[i])):
-                    in_header_block = True
-            else:
-                if QUOT_PATTERN.match(lines[i]):
-                    m = 'm'
-                else:
-                    m = 't'
-
-        # If the line is not a header line, set in_header_block false.
-        if not bool(re.search(RE_HEADER, lines[i])):
-            in_header_block = False
-
-        # Add the marker to the new updated markers string.
-        updated_markers += m
-        i += 1
-
-    return updated_markers
-
-
-def _readable_text_empty(html_tree):
-    return not bool(html_tree_to_text(html_tree).strip())
-
-
 def is_splitter(line):
    '''
    Returns Matcher object if provided string is a splitter and
@@ -594,7 +415,7 @@ def is_splitter(line):

 def text_content(context):
    '''XPath Extension function to return a node text content.'''
-    return context.context_node.xpath("string()").strip()
+    return context.context_node.text_content().strip()


 def tail(context):
--- a/talon/signature/init.py
+++ b/talon/signature/init.py
@@ -20,7 +20,6 @@ trained against, don't forget to regenerate:
 * signature/data/classifier
 """

-from __future__ import absolute_import
 import os

 from . import extraction
--- a/talon/signature/bruteforce.py
+++ b/talon/signature/bruteforce.py
@@ -1,4 +1,3 @@
-from __future__ import absolute_import
 import logging

 import regex as re
@@ -112,7 +111,7 @@ def extract_signature(msg_body):

            return (stripped_body.strip(),
                    signature.strip())
-    except Exception as e:
+    except Exception, e:
        log.exception('ERROR extracting signature')
        return (msg_body, None)

--- a/talon/signature/data/classifier
+++ b/talon/signature/data/classifier
--- a/talon/signature/data/classifier_02.npy
+++ b/talon/signature/data/classifier_02.npy
--- a/talon/signature/data/classifier_03.npy
+++ b/talon/signature/data/classifier_03.npy
--- a/talon/signature/data/train.data
+++ b/talon/signature/data/train.data
--- a/talon/signature/extraction.py
+++ b/talon/signature/extraction.py
@@ -1,6 +1,5 @@
 # -*- coding: utf-8 -*-

-from __future__ import absolute_import
 import logging

 import regex as re
--- a/talon/signature/learning/classifier.py
+++ b/talon/signature/learning/classifier.py
@@ -5,7 +5,6 @@ The classifier could be used to detect if a certain line of the message
 body belongs to the signature.
 """

-from __future__ import absolute_import
 from numpy import genfromtxt
 from sklearn.svm import LinearSVC
 from sklearn.externals import joblib
--- a/talon/signature/learning/dataset.py
+++ b/talon/signature/learning/dataset.py
@@ -16,13 +16,11 @@ suffix and the corresponding sender file has the same name except for the
 suffix which should be `_sender`.
 """

-from __future__ import absolute_import
 import os
 import regex as re

 from talon.signature.constants import SIGNATURE_MAX_LINES
 from talon.signature.learning.featurespace import build_pattern, features
-from six.moves import range


 SENDER_SUFFIX = '_sender'
@@ -146,7 +144,7 @@ def build_extraction_dataset(folder, dataset_filename,
            if not sender or not msg:
                continue
            lines = msg.splitlines()
-            for i in range(1, min(SIGNATURE_MAX_LINES,
+            for i in xrange(1, min(SIGNATURE_MAX_LINES,
                                   len(lines)) + 1):
                line = lines[-i]
                label = -1
--- a/talon/signature/learning/featurespace.py
+++ b/talon/signature/learning/featurespace.py
@@ -7,12 +7,9 @@ The body and the message sender string are converted into unicode before
 applying features to them.
 """

-from __future__ import absolute_import
 from talon.signature.constants import (SIGNATURE_MAX_LINES,
                                       TOO_LONG_SIGNATURE_LINE)
 from talon.signature.learning.helpers import *
-from six.moves import zip
-from functools import reduce


 def features(sender=''):
--- a/talon/signature/learning/helpers.py
+++ b/talon/signature/learning/helpers.py
@@ -6,7 +6,6 @@

 """

-from __future__ import absolute_import
 import unicodedata
 import regex as re

@@ -17,7 +16,7 @@ from talon.signature.constants import SIGNATURE_MAX_LINES

 rc = re.compile

-RE_EMAIL = rc('\S@\S')
+RE_EMAIL = rc('@')
 RE_RELAX_PHONE = rc('(\(? ?[\d]{2,3} ?\)?.{,3}?){2,}')
 RE_URL = rc(r'''https?://|www\.[\S]+\.[\S]''')

@@ -121,7 +120,7 @@ def contains_sender_names(sender):
    names = names or sender
    if names != '':
        return binary_regex_search(re.compile(names))
-    return lambda s: 0
+    return lambda s: False


 def extract_names(sender):
@@ -135,7 +134,7 @@ def extract_names(sender):
    >>> extract_names('')
    []
    """
-    sender = to_unicode(sender, precise=True)
+    sender = to_unicode(sender)
    # Remove non-alphabetical characters
    sender = "".join([char if char.isalpha() else ' ' for char in sender])
    # Remove too short words and words from "black" list i.e.
@@ -162,7 +161,7 @@ def categories_percent(s, categories):
    50.0
    '''
    count = 0
-    s = to_unicode(s, precise=True)
+    s = to_unicode(s)
    for c in s:
        if unicodedata.category(c) in categories:
            count += 1
@@ -182,16 +181,15 @@ def punctuation_percent(s):

 def capitalized_words_percent(s):
    '''Returns capitalized words percent.'''
-    s = to_unicode(s, precise=True)
+    s = to_unicode(s)
    words = re.split('\s', s)
    words = [w for w in words if w.strip()]
-    words = [w for w in words if len(w) > 2]    
    capitalized_words_counter = 0
    valid_words_counter = 0
    for word in words:
        if not INVALID_WORD_START.match(word):
            valid_words_counter += 1
-            if word[0].isupper() and not word[1].isupper():
+            if word[0].isupper():
                capitalized_words_counter += 1
    if valid_words_counter > 0 and len(words) > 1:
        return 100 * float(capitalized_words_counter) / valid_words_counter
--- a/talon/utils.py
+++ b/talon/utils.py
@@ -1,19 +1,12 @@
 # coding:utf-8

-from __future__ import absolute_import
 import logging
 from random import shuffle
-import chardet
-import cchardet
-import regex as re
-
-from lxml.html import html5parser
-from lxml.cssselect import CSSSelector
-
-import html5lib

 from talon.constants import RE_DELIMITER
-import six
+
+
+log = logging.getLogger(__name__)


 def safe_format(format_string, *args, **kwargs):
@@ -32,7 +25,7 @@ def safe_format(format_string, *args, **kwargs):
    except (UnicodeEncodeError, UnicodeDecodeError):
        format_string = to_utf8(format_string)
        args = [to_utf8(p) for p in args]
-        kwargs = {k: to_utf8(v) for k, v in six.iteritems(kwargs)}
+        kwargs = {k: to_utf8(v) for k, v in kwargs.iteritems()}
        return format_string.format(*args, **kwargs)

    # ignore other errors
@@ -49,51 +42,19 @@ def to_unicode(str_or_unicode, precise=False):
        u'привет'
    If `precise` flag is True, tries to guess the correct encoding first.
    """
-    if not isinstance(str_or_unicode, six.text_type):
-        encoding = quick_detect_encoding(str_or_unicode) if precise else 'utf-8'
-        return six.text_type(str_or_unicode, encoding, 'replace')
+    encoding = detect_encoding(str_or_unicode) if precise else 'utf-8'
+    if isinstance(str_or_unicode, str):
+        return unicode(str_or_unicode, encoding, 'replace')
    return str_or_unicode


-def detect_encoding(string):
-    """
-    Tries to detect the encoding of the passed string.
-
-    Defaults to UTF-8.
-    """
-    assert isinstance(string, bytes)
-    try:
-        detected = chardet.detect(string)
-        if detected:
-            return detected.get('encoding') or 'utf-8'
-    except Exception as e:
-        pass
-    return 'utf-8'
-
-
-def quick_detect_encoding(string):
-    """
-    Tries to detect the encoding of the passed string.
-
-    Uses cchardet. Fallbacks to detect_encoding.
-    """
-    assert isinstance(string, bytes)
-    try:
-        detected = cchardet.detect(string)
-        if detected:
-            return detected.get('encoding') or detect_encoding(string)
-    except Exception as e:
-        pass
-    return detect_encoding(string)
-
-
 def to_utf8(str_or_unicode):
    """
    Safely returns a UTF-8 version of a given string
    >>> utils.to_utf8(u'hi')
        'hi'
    """
-    if not isinstance(str_or_unicode, six.text_type):
+    if isinstance(str_or_unicode, unicode):
        return str_or_unicode.encode("utf-8", "ignore")
    return str(str_or_unicode)

@@ -113,147 +74,3 @@ def get_delimiter(msg_body):
        delimiter = '\n'

    return delimiter
-
-
-def html_tree_to_text(tree):
-    for style in CSSSelector('style')(tree):
-        style.getparent().remove(style)
-
-    for c in tree.xpath('//comment()'):
-        parent = c.getparent()
-
-        # comment with no parent does not impact produced text
-        if parent is None:
-            continue
-
-        parent.remove(c)
-
-    text   = ""
-    for el in tree.iter():
-        el_text = (el.text or '') + (el.tail or '')
-        if len(el_text) > 1:
-            if el.tag in _BLOCKTAGS:
-                text += "\n"
-            if el.tag == 'li':
-                text += "  * "
-            text += el_text.strip() + " "
-
-            # add href to the output
-            href = el.attrib.get('href')
-            if href:
-                text += "(%s) " % href
-
-        if el.tag in _HARDBREAKS and text and not text.endswith("\n"):
-            text += "\n"
-
-    retval = _rm_excessive_newlines(text)
-    return _encode_utf8(retval)
-
-
-def html_to_text(string):
-    """
-    Dead-simple HTML-to-text converter:
-        >>> html_to_text("one<br>two<br>three")
-        >>> "one\ntwo\nthree"
-
-    NOTES:
-        1. the string is expected to contain UTF-8 encoded HTML!
-        2. returns utf-8 encoded str (not unicode)
-        3. if html can't be parsed returns None
-    """
-    if isinstance(string, six.text_type):
-        string = string.encode('utf8')
-
-    s = _prepend_utf8_declaration(string)
-    s = s.replace(b"\n", b"")
-    tree = html_fromstring(s)
-
-    if tree is None:
-        return None
-
-    return html_tree_to_text(tree)
-
-
-def html_fromstring(s):
-    """Parse html tree from string. Return None if the string can't be parsed.
-    """
-    try:
-        if html_too_big(s):
-            return None
-
-        return html5parser.fromstring(s, parser=_html5lib_parser())
-    except Exception:
-        pass
-
-
-def html_document_fromstring(s):
-    """Parse html tree from string. Return None if the string can't be parsed.
-    """
-    try:
-        if html_too_big(s):
-            return None
-
-        return html5parser.document_fromstring(s, parser=_html5lib_parser())
-    except Exception:
-        pass
-
-
-def cssselect(expr, tree):
-    return CSSSelector(expr)(tree)
-
-
-def html_too_big(s):
-    return s.count('<') > _MAX_TAGS_COUNT
-
-
-def _contains_charset_spec(s):
-    """Return True if the first 4KB contain charset spec
-    """
-    return s.lower().find(b'html; charset=', 0, 4096) != -1
-
-
-def _prepend_utf8_declaration(s):
-    """Prepend 'utf-8' encoding declaration if the first 4KB don't have any
-    """
-    return s if _contains_charset_spec(s) else _UTF8_DECLARATION + s
-
-
-def _rm_excessive_newlines(s):
-    """Remove excessive newlines that often happen due to tons of divs
-    """
-    return _RE_EXCESSIVE_NEWLINES.sub("\n\n", s).strip()
-
-
-def _encode_utf8(s):
-    """Encode in 'utf-8' if unicode
-    """
-    return s.encode('utf-8') if isinstance(s, six.text_type) else s
-
-
-def _html5lib_parser():
-    """
-    html5lib is a pure-python library that conforms to the WHATWG HTML spec
-    and is not vulnarable to certain attacks common for XML libraries
-    """
-    return html5lib.HTMLParser(
-        # build lxml tree
-        html5lib.treebuilders.getTreeBuilder("lxml"),
-        # remove namespace value from inside lxml.html.html5paser element tag
-        # otherwise it yields something like "{http://www.w3.org/1999/xhtml}div"
-        # instead of "div", throwing the algo off
-        namespaceHTMLElements=False
-    )
-
-
-_UTF8_DECLARATION = (b'<meta http-equiv="Content-Type" content="text/html;'
-                     b'charset=utf-8">')
-
-
-_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
--- a/tests/init.py
+++ b/tests/init.py
@@ -1,4 +1,3 @@
-from __future__ import absolute_import
 from nose.tools import *
 from mock import *

--- a/tests/fixtures/html_replies/hotmail.html
+++ b/tests/fixtures/html_replies/hotmail.html
@@ -1,4 +1,3 @@
-<?xml version="1.0" encoding="UTF-8"?>
 <html>
 <head>
 <style><!--
--- a/tests/fixtures/html_replies/ms_outlook_2010.html
+++ b/tests/fixtures/html_replies/ms_outlook_2010.html
@@ -1,87 +0,0 @@
-<html>
-<head>
-<meta http-equiv="Content-Type" content="text/html; charset=iso-2022-jp">
-<meta name="Generator" content="Microsoft Word 14 (filtered medium)">
-<style><!--
-/* Font Definitions */
-@font-face
-	{font-family:Calibri;
-	panose-1:2 15 5 2 2 2 4 3 2 4;}
-@font-face
-	{font-family:Tahoma;
-	panose-1:2 11 6 4 3 5 4 4 2 4;}
-/* Style Definitions */
-p.MsoNormal, li.MsoNormal, div.MsoNormal
-	{margin:0in;
-	margin-bottom:.0001pt;
-	font-size:12.0pt;
-	font-family:"Times New Roman","serif";}
-h3
-	{mso-style-priority:9;
-	mso-style-link:"Heading 3 Char";
-	mso-margin-top-alt:auto;
-	margin-right:0in;
-	mso-margin-bottom-alt:auto;
-	margin-left:0in;
-	font-size:13.5pt;
-	font-family:"Times New Roman","serif";
-	font-weight:bold;}
-a:link, span.MsoHyperlink
-	{mso-style-priority:99;
-	color:blue;
-	text-decoration:underline;}
-a:visited, span.MsoHyperlinkFollowed
-	{mso-style-priority:99;
-	color:purple;
-	text-decoration:underline;}
-p
-	{mso-style-priority:99;
-	mso-margin-top-alt:auto;
-	margin-right:0in;
-	mso-margin-bottom-alt:auto;
-	margin-left:0in;
-	font-size:12.0pt;
-	font-family:"Times New Roman","serif";}
-span.Heading3Char
-	{mso-style-name:"Heading 3 Char";
-	mso-style-priority:9;
-	mso-style-link:"Heading 3";
-	font-family:"Cambria","serif";
-	color:#4F81BD;
-	font-weight:bold;}
-span.EmailStyle19
-	{mso-style-type:personal-reply;
-	font-family:"Calibri","sans-serif";
-	color:#1F497D;}
-.MsoChpDefault
-	{mso-style-type:export-only;
-	font-family:"Calibri","sans-serif";}
-@page WordSection1
-	{size:8.5in 11.0in;
-	margin:1.0in 1.0in 1.0in 1.0in;}
-div.WordSection1
-	{page:WordSection1;}
--></style><!--[if gte mso 9]><xml>
-<o:shapedefaults v:ext="edit" spidmax="1026" />
-</xml><![endif]--><!--[if gte mso 9]><xml>
-<o:shapelayout v:ext="edit">
-<o:idmap v:ext="edit" data="1" />
-</o:shapelayout></xml><![endif]-->
-</head>
-<body lang="EN-US" link="blue" vlink="purple">
-<div class="WordSection1">
-<p class="MsoNormal"><span style="font-size:11.0pt;font-family:&quot;Calibri&quot;,&quot;sans-serif&quot;;color:#1F497D">Hi. I am fine.<o:p></o:p></span></p>
-<p class="MsoNormal"><span style="font-size:11.0pt;font-family:&quot;Calibri&quot;,&quot;sans-serif&quot;;color:#1F497D">Thanks,<o:p></o:p></span></p>
-<p class="MsoNormal"><span style="font-size:11.0pt;font-family:&quot;Calibri&quot;,&quot;sans-serif&quot;;color:#1F497D">Alex<o:p></o:p></span></p>
-<p class="MsoNormal"><b><span style="font-size:10.0pt;font-family:&quot;Tahoma&quot;,&quot;sans-serif&quot;">From:</span></b><span style="font-size:10.0pt;font-family:&quot;Tahoma&quot;,&quot;sans-serif&quot;"> Foo [mailto:foo@bar.com]
-<b>On Behalf Of </b>baz@bar.com<br>
-<b>Sent:</b> Monday, January 01, 2000 12:00 AM<br>
-<b>To:</b> john@bar.com<br>
-<b>Cc:</b> jane@bar.io<br>
-<b>Subject:</b> Conversation<o:p></o:p></span></p>
-<p class="MsoNormal"><o:p>&nbsp;</o:p></p>
-<p>Hello! How are you?<o:p></o:p></p>
-<p class="MsoNormal"><o:p>&nbsp;</o:p></p>
-</div>
-</body>
-</html>
--- a/tests/fixtures/standard_replies/apple_mail_2.eml
+++ b/tests/fixtures/standard_replies/apple_mail_2.eml
@@ -1,19 +0,0 @@
-Content-Type: text/plain;
-	charset=us-ascii
-Mime-Version: 1.0 (Mac OS X Mail 8.2 \(2104\))
-Subject: Re: Hello there
-X-Universally-Unique-Identifier: 85B1075D-5841-46A9-8565-FCB287A93AC4
-From: Adam Renberg <adam@tictail.com>
-In-Reply-To: <CABzQGhkMXDxUt_tSVQcg=43aniUhtsVfCZVzu-PG0kwS_uzqMw@mail.gmail.com>
-Date: Sat, 22 Aug 2015 19:22:20 +0200
-Content-Transfer-Encoding: 7bit
-X-Smtp-Server: smtp.gmail.com:adam@tictail.com
-Message-Id: <68001B29-8EA4-444C-A894-0537D2CA5208@tictail.com>
-References: <CABzQGhkMXDxUt_tSVQcg=43aniUhtsVfCZVzu-PG0kwS_uzqMw@mail.gmail.com>
-To: Adam Renberg <tgwizard@gmail.com>
-
-Hello
-> On 22 Aug 2015, at 19:21, Adam Renberg <tgwizard@gmail.com> wrote:
->
-> Hi there!
-
--- a/tests/html_quotations_test.py
+++ b/tests/html_quotations_test.py
@@ -1,12 +1,13 @@
 # -*- coding: utf-8 -*-

-from __future__ import absolute_import
 from . import *
 from . fixtures import *

 import regex as re

-from talon import quotations, utils as u
+from talon import quotations
+
+import html2text


 RE_WHITESPACE = re.compile("\s")
@@ -27,7 +28,7 @@ def test_quotation_splitter_inside_blockquote():

 </blockquote>"""

-    eq_("<html><head></head><body>Reply</body></html>",
+    eq_("<html><body><p>Reply</p></body></html>",
        RE_WHITESPACE.sub('', quotations.extract_from_html(msg_body)))


@@ -44,25 +45,7 @@ def test_quotation_splitter_outside_blockquote():
  </div>
 </blockquote>
 """
-    eq_("<html><head></head><body>Reply</body></html>",
-        RE_WHITESPACE.sub('', quotations.extract_from_html(msg_body)))
-
-
-def test_regular_blockquote():
-    msg_body = """Reply
-<blockquote>Regular</blockquote>
-
-<div>
-  On 11-Apr-2011, at 6:54 PM, Bob &lt;bob@example.com&gt; wrote:
-</div>
-
-<blockquote>
-  <div>
-    <blockquote>Nested</blockquote>
-  </div>
-</blockquote>
-"""
-    eq_("<html><head></head><body>Reply<blockquote>Regular</blockquote></body></html>",
+    eq_("<html><body><p>Reply</p><div></div></body></html>",
        RE_WHITESPACE.sub('', quotations.extract_from_html(msg_body)))


@@ -85,7 +68,6 @@ Reply

    reply = """
 <html>
-<head></head>
 <body>
 Reply

@@ -129,30 +111,7 @@ def test_gmail_quote():
    </div>
  </div>
 </div>"""
-    eq_("<html><head></head><body>Reply</body></html>",
-        RE_WHITESPACE.sub('', quotations.extract_from_html(msg_body)))
-
-
-def test_gmail_quote_compact():
-    msg_body = 'Reply' \
-               '<div class="gmail_quote">' \
-               '<div class="gmail_quote">On 11-Apr-2011, at 6:54 PM, Bob &lt;bob@example.com&gt; wrote:' \
-               '<div>Test</div>' \
-               '</div>' \
-               '</div>'
-    eq_("<html><head></head><body>Reply</body></html>",
-        RE_WHITESPACE.sub('', quotations.extract_from_html(msg_body)))
-
-
-def test_gmail_quote_blockquote():
-    msg_body = """Message
-<blockquote class="gmail_quote">
-  <div class="gmail_default">
-    My name is William Shakespeare.
-    <br/>
-  </div>
-</blockquote>"""
-    eq_(RE_WHITESPACE.sub('', msg_body),
+    eq_("<html><body><p>Reply</p></body></html>",
        RE_WHITESPACE.sub('', quotations.extract_from_html(msg_body)))


@@ -163,11 +122,11 @@ def test_unicode_in_reply():
  <br>
 </div>

-<blockquote>
+<blockquote class="gmail_quote">
  Quote
 </blockquote>""".encode("utf-8")

-    eq_("<html><head></head><body>Reply&#160;&#160;Text<br><div><br></div>"
+    eq_("<html><body><p>Reply&#160;&#160;Text<br></p><div><br></div>"
        "</body></html>",
        RE_WHITESPACE.sub('', quotations.extract_from_html(msg_body)))

@@ -193,7 +152,6 @@ def test_blockquote_disclaimer():

    stripped_html = """
 <html>
-  <head></head>
  <body>
  <div>
    <div>
@@ -225,7 +183,7 @@ def test_date_block():
  </div>
 </div>
 """
-    eq_('<html><head></head><body><div>message<br></div></body></html>',
+    eq_('<html><body><div>message<br></div></body></html>',
        RE_WHITESPACE.sub('', quotations.extract_from_html(msg_body)))


@@ -242,7 +200,7 @@ Subject: You Have New Mail From Mary!<br><br>
 text
 </div></div>
 """
-    eq_('<html><head></head><body><div>message<br></div></body></html>',
+    eq_('<html><body><div>message<br></div></body></html>',
        RE_WHITESPACE.sub('', quotations.extract_from_html(msg_body)))


@@ -260,7 +218,7 @@ def test_reply_shares_div_with_from_block():

  </div>
 </body>'''
-    eq_('<html><head></head><body><div>Blah<br><br></div></body></html>',
+    eq_('<html><body><div>Blah<br><br></div></body></html>',
        RE_WHITESPACE.sub('', quotations.extract_from_html(msg_body)))


@@ -271,47 +229,37 @@ def test_reply_quotations_share_block():


 def test_OLK_SRC_BODY_SECTION_stripped():
-    eq_('<html><head></head><body><div>Reply</div></body></html>',
+    eq_('<html><body><div>Reply</div></body></html>',
        RE_WHITESPACE.sub(
            '', quotations.extract_from_html(OLK_SRC_BODY_SECTION)))


 def test_reply_separated_by_hr():
-    eq_('<html><head></head><body><div>Hi<div>there</div></div></body></html>',
+    eq_('<html><body><div>Hi<div>there</div></div></body></html>',
        RE_WHITESPACE.sub(
            '', quotations.extract_from_html(REPLY_SEPARATED_BY_HR)))


-def test_from_block_and_quotations_in_separate_divs():
-    msg_body = '''
-Reply
-<div>
-  <hr/>
-  <div>
-    <font>
-      <b>From: bob@example.com</b>
-      <b>Date: Thu, 24 Mar 2016 08:07:12 -0700</b>
-    </font>
-  </div>
-  <div>
-    Quoted message
-  </div>
-</div>
-'''
-    eq_('<html><head></head><body>Reply<div><hr></div></body></html>',
-        RE_WHITESPACE.sub('', quotations.extract_from_html(msg_body)))
+RE_REPLY = re.compile(r"^Hi\. I am fine\.\s*\n\s*Thanks,\s*\n\s*Alex\s*$")


 def extract_reply_and_check(filename):
    f = open(filename)

-    msg_body = f.read()
+    msg_body = f.read().decode("utf-8")
    reply = quotations.extract_from_html(msg_body)
-    plain_reply = u.html_to_text(reply)
-    plain_reply = plain_reply.decode('utf8')

-    eq_(RE_WHITESPACE.sub('', "Hi. I am fine.\n\nThanks,\nAlex"),
-        RE_WHITESPACE.sub('', plain_reply))
+    h = html2text.HTML2Text()
+    h.body_width = 0
+    plain_reply = h.handle(reply)
+
+    #remove &nbsp; spaces
+    plain_reply = plain_reply.replace(u'\xa0', u' ')
+
+    if RE_REPLY.match(plain_reply):
+        eq_(1, 1)
+    else:
+        eq_("Hi. I am fine.\n\nThanks,\nAlex", plain_reply)


 def test_gmail_reply():
@@ -334,10 +282,6 @@ def test_ms_outlook_2007_reply():
    extract_reply_and_check("tests/fixtures/html_replies/ms_outlook_2007.html")


-def test_ms_outlook_2010_reply():
-    extract_reply_and_check("tests/fixtures/html_replies/ms_outlook_2010.html")
-
-
 def test_thunderbird_reply():
    extract_reply_and_check("tests/fixtures/html_replies/thunderbird.html")

@@ -348,74 +292,3 @@ def test_windows_mail_reply():

 def test_yandex_ru_reply():
    extract_reply_and_check("tests/fixtures/html_replies/yandex_ru.html")
-
-
-def test_CRLF():
-    """CR is not converted to '&#13;'
-    """
-    symbol = '&#13;'
-    extracted = quotations.extract_from_html('<html>\r\n</html>')
-    assert_false(symbol in extracted)
-    eq_('<html></html>', RE_WHITESPACE.sub('', extracted))
-
-    msg_body = """My
-reply
-<blockquote>
-
-  <div>
-    On 11-Apr-2011, at 6:54 PM, Bob &lt;bob@example.com&gt; wrote:
-  </div>
-
-  <div>
-    Test
-  </div>
-
-</blockquote>"""
-    msg_body = msg_body.replace('\n', '\r\n')
-    extracted = quotations.extract_from_html(msg_body)
-    assert_false(symbol in extracted)    
-    # Keep new lines otherwise "My reply" becomes one word - "Myreply" 
-    eq_("<html><head></head><body>My\nreply\n</body></html>", extracted)
-
-
-def test_gmail_forwarded_msg():
-    msg_body = """<div dir="ltr"><br><div class="gmail_quote">---------- Forwarded message ----------<br>From: <b class="gmail_sendername">Bob</b> <span dir="ltr">&lt;<a href="mailto:bob@example.com">bob@example.com</a>&gt;</span><br>Date: Fri, Feb 11, 2010 at 5:59 PM<br>Subject: Bob WFH today<br>To: Mary &lt;<a href="mailto:mary@example.com">mary@example.com</a>&gt;<br><br><br><div dir="ltr">eom</div>
-</div><br></div>"""
-    extracted = quotations.extract_from_html(msg_body)
-    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 &lt;bob@example.com&gt; 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>
-  Reply
-  <div>
-    On 11-Apr-2011, at 6:54 PM, Bob &lt;bob@example.com&gt; wrote:
-  </div>
-
-  <div>
-    Test
-  </div>
-
-</blockquote>"""
-
-    eq_(RE_WHITESPACE.sub('', msg_body),
-        RE_WHITESPACE.sub('', quotations.extract_from_html(msg_body)))
-
-
-@patch.object(quotations, 'html_document_fromstring', Mock(return_value=None))
-def test_bad_html():
-    bad_html = "<html></html>"
-    eq_(bad_html, quotations.extract_from_html(bad_html))
--- a/tests/quotations_test.py
+++ b/tests/quotations_test.py
@@ -1,6 +1,5 @@
 # -*- coding: utf-8 -*-

-from __future__ import absolute_import
 from . import *
 from . fixtures import *

--- a/tests/signature/bruteforce_test.py
+++ b/tests/signature/bruteforce_test.py
@@ -1,6 +1,5 @@
 # -*- coding: utf-8 -*-

-from __future__ import absolute_import
 from .. import *

 from talon.signature import bruteforce
--- a/tests/signature/extraction_test.py
+++ b/tests/signature/extraction_test.py
@@ -1,6 +1,5 @@
 # -*- coding: utf-8 -*-

-from __future__ import absolute_import
 from .. import *

 import os
@@ -9,7 +8,6 @@ from talon.signature.learning import dataset
 from talon import signature
 from talon.signature import extraction as e
 from talon.signature import bruteforce
-from six.moves import range


 def test_message_shorter_SIGNATURE_MAX_LINES():
@@ -77,31 +75,6 @@ def test_basic():
        signature.extract(msg_body, 'Sergey'))


-def test_capitalized():
-    msg_body = """Hi Mary,
-
-Do you still need a DJ for your wedding? I've included a video demo of one of our DJs available for your wedding date.
-
-DJ Doe 
-http://example.com
-Password: SUPERPASSWORD
-
-Would you like to check out more?
-
-
-At your service,
-
-John Smith
-Doe Inc
-555-531-7967"""
-
-    sig = """John Smith
-Doe Inc
-555-531-7967"""
-
-    eq_(sig, signature.extract(msg_body, 'Doe')[1])
-
-
 def test_over_2_text_lines_after_signature():
    body = """Blah

@@ -154,20 +127,20 @@ def test_mark_lines():

 def test_process_marked_lines():
    # no signature found
-    eq_((list(range(5)), None), e._process_marked_lines(list(range(5)), 'telt'))
+    eq_((range(5), None), e._process_marked_lines(range(5), 'telt'))

    # signature in the middle of the text
-    eq_((list(range(9)), None), e._process_marked_lines(list(range(9)), 'tesestelt'))
+    eq_((range(9), None), e._process_marked_lines(range(9), 'tesestelt'))

    # long line splits signature
-    eq_((list(range(7)), [7, 8]),
-        e._process_marked_lines(list(range(9)), 'tsslsless'))
+    eq_((range(7), [7, 8]),
+        e._process_marked_lines(range(9), 'tsslsless'))

-    eq_((list(range(20)), [20]),
-        e._process_marked_lines(list(range(21)), 'ttttttstttesllelelets'))
+    eq_((range(20), [20]),
+        e._process_marked_lines(range(21), 'ttttttstttesllelelets'))

    # some signature lines could be identified as text
-    eq_(([0], list(range(1, 9))), e._process_marked_lines(list(range(9)), 'tsetetest'))
+    eq_(([0], range(1, 9)), e._process_marked_lines(range(9), 'tsetetest'))

-    eq_(([], list(range(5))),
-        e._process_marked_lines(list(range(5)), "ststt"))
+    eq_(([], range(5)),
+        e._process_marked_lines(range(5), "ststt"))
--- a/tests/signature/learning/dataset_test.py
+++ b/tests/signature/learning/dataset_test.py
@@ -1,6 +1,5 @@
 # -*- coding: utf-8 -*-

-from __future__ import absolute_import
 from ... import *
 import os

--- a/tests/signature/learning/featurespace_test.py
+++ b/tests/signature/learning/featurespace_test.py
@@ -1,15 +1,12 @@
 # -*- coding: utf-8 -*-

-from __future__ import absolute_import
 from ... import *

 from talon.signature.learning import featurespace as fs


 def test_apply_features():
-    s = '''This is John Doe
-
-Tuesday @3pm suits. I'll chat to you then.
+    s = '''John Doe

 VP Research and Development, Xxxx Xxxx Xxxxx

@@ -22,12 +19,11 @@ john@example.com'''
    # note that we don't consider the first line because signatures don't
    # usually take all the text, empty lines are not considered
    eq_(result, [[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
-                 [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                 [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                 [0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0],
                 [0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0]])

-    with patch.object(fs, 'SIGNATURE_MAX_LINES', 5):
+    with patch.object(fs, 'SIGNATURE_MAX_LINES', 4):
        features = fs.features(sender)
        new_result = fs.apply_features(s, features)
        # result remains the same because we don't consider empty lines
--- a/tests/signature/learning/helpers_test.py
+++ b/tests/signature/learning/helpers_test.py
@@ -1,13 +1,11 @@
 # -*- coding: utf-8 -*-

-from __future__ import absolute_import
 from ... import *

 import regex as re

 from talon.signature.learning import helpers as h
 from talon.signature.learning.helpers import *
-from six.moves import range

 # First testing regex constants.
 VALID = '''
@@ -156,7 +154,7 @@ def test_extract_names():
        # check that extracted names could be compiled
        try:
            re.compile("|".join(extracted_names))
-        except Exception as e:
+        except Exception, e:
            ok_(False, ("Failed to compile extracted names {}"
                        "\n\nReason: {}").format(extracted_names, e))
        if expected_names:
@@ -192,11 +190,10 @@ def test_punctuation_percent(categories_percent):
 def test_capitalized_words_percent():
    eq_(0.0, h.capitalized_words_percent(''))
    eq_(100.0, h.capitalized_words_percent('Example Corp'))
-    eq_(50.0, h.capitalized_words_percent('Qqq qqq Aqs 123 sss'))
+    eq_(50.0, h.capitalized_words_percent('Qqq qqq QQQ 123 sss'))
    eq_(100.0, h.capitalized_words_percent('Cell 713-444-7368'))
    eq_(100.0, h.capitalized_words_percent('8th Floor'))
    eq_(0.0, h.capitalized_words_percent('(212) 230-9276'))
-    eq_(50.0, h.capitalized_words_percent('Password: REMARKABLE'))


 def test_has_signature():
@@ -207,7 +204,7 @@ def test_has_signature():
                        'sender@example.com'))
    assert_false(h.has_signature('http://www.example.com/555-555-5555',
                                 'sender@example.com'))
-    long_line = ''.join(['q' for e in range(28)])
+    long_line = ''.join(['q' for e in xrange(28)])
    assert_false(h.has_signature(long_line + ' sender', 'sender@example.com'))
    # wont crash on an empty string
    assert_false(h.has_signature('', ''))
--- a/tests/text_quotations_test.py
+++ b/tests/text_quotations_test.py
@@ -1,6 +1,5 @@
 # -*- coding: utf-8 -*-

-from __future__ import absolute_import
 from . import *
 from . fixtures import *

@@ -8,19 +7,16 @@ import os

 import email.iterators
 from talon import quotations
-import six
-from six.moves import range
-from six import StringIO


@patch.object(quotations, 'MAX_LINES_COUNT', 1)
 def test_too_many_lines():
    msg_body = """Test reply
-Hi
+
 -----Original Message-----

 Test"""
-    eq_("Test reply", quotations.extract_from_plain(msg_body))
+    eq_(msg_body, quotations.extract_from_plain(msg_body))


 def test_pattern_on_date_somebody_wrote():
@@ -28,32 +24,6 @@ def test_pattern_on_date_somebody_wrote():

 On 11-Apr-2011, at 6:54 PM, Roman Tkachenko <romant@example.com> wrote:

->
-> Test
->
-> Roman"""
-
-    eq_("Test reply", quotations.extract_from_plain(msg_body))
-
-def test_pattern_on_date_polymail():
-    msg_body = """Test reply
-
-On Tue, Apr 11, 2017 at 10:07 PM John Smith
-
-<
-mailto:John Smith <johnsmith@gmail.com>
-> wrote:
-Test quoted data
-"""
-
-    eq_("Test reply", quotations.extract_from_plain(msg_body))
-
-
-def test_pattern_sent_from_samsung_smb_wrote():
-    msg_body = """Test reply
-
-Sent from Samsung MobileName <address@example.com> wrote:
-
 >
 > Test
 >
@@ -84,18 +54,6 @@ On 04/19/2011 07:10 AM, Roman Tkachenko wrote:
    eq_("Test reply", quotations.extract_from_plain(msg_body))


-def test_date_time_email_splitter():
-    msg_body = """Test reply
-
-2014-10-17 11:28 GMT+03:00 Postmaster <
-postmaster@sandboxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx.mailgun.org>:
-
-> First from site
->
-    """
-    eq_("Test reply", quotations.extract_from_plain(msg_body))
-
-
 def test_pattern_on_date_somebody_wrote_allows_space_in_front():
    msg_body = """Thanks Thanmai
 On Mar 8, 2012 9:59 AM, "Example.com" <
@@ -155,8 +113,7 @@ def _check_pattern_original_message(original_message_indicator):
 -----{}-----

 Test"""
-    eq_('Test reply', quotations.extract_from_plain(
-        msg_body.format(six.text_type(original_message_indicator))))
+    eq_('Test reply', quotations.extract_from_plain(msg_body.format(unicode(original_message_indicator))))

 def test_english_original_message():
    _check_pattern_original_message('Original Message')
@@ -179,17 +136,6 @@ Test reply"""
    eq_("Test reply", quotations.extract_from_plain(msg_body))


-def test_android_wrote():
-    msg_body = """Test reply
-
---- John Smith wrote ----
-
-> quoted
-> text
-"""
-    eq_("Test reply", quotations.extract_from_plain(msg_body))
-
-
 def test_reply_wraps_quotations():
    msg_body = """Test reply

@@ -365,33 +311,6 @@ Emne: The manager has commented on your Loop
 Blah-blah-blah
 """))

-def test_swedish_from_block():
-    eq_('Allo! Follow up MIME!', quotations.extract_from_plain(
-    u"""Allo! Follow up MIME!
-Från: Anno Sportel [mailto:anno.spoel@hsbcssad.com]
-Skickat: den 26 augusti 2015 14:45
-Till: Isacson Leiff
-Ämne: RE: Week 36
-
-Blah-blah-blah
-"""))
-
-def test_swedish_from_line():
-    eq_('Lorem', quotations.extract_from_plain(
-    """Lorem
-Den 14 september, 2015 02:23:18, Valentino Rudy (valentino@rudy.be) skrev:
-
-Veniam laborum mlkshk kale chips authentic. Normcore mumblecore laboris, fanny pack readymade eu blog chia pop-up freegan enim master cleanse.
-"""))
-
-def test_norwegian_from_line():
-    eq_('Lorem', quotations.extract_from_plain(
-    u"""Lorem
-På 14 september 2015 på 02:23:18, Valentino Rudy (valentino@rudy.be) skrev:
-
-Veniam laborum mlkshk kale chips authentic. Normcore mumblecore laboris, fanny pack readymade eu blog chia pop-up freegan enim master cleanse.
-"""))
-
 def test_dutch_from_block():
    eq_('Gluten-free culpa lo-fi et nesciunt nostrud.', quotations.extract_from_plain(
    """Gluten-free culpa lo-fi et nesciunt nostrud. 
@@ -691,15 +610,6 @@ def test_preprocess_postprocess_2_links():
    eq_(msg_body, quotations.extract_from_plain(msg_body))


-def body_iterator(msg, decode=False):
-    for subpart in msg.walk():
-        payload = subpart.get_payload(decode=decode)
-        if isinstance(payload, six.text_type):
-            yield payload
-        else:
-            yield payload.decode('utf8')
-
-
 def test_standard_replies():
    for filename in os.listdir(STANDARD_REPLIES):
        filename = os.path.join(STANDARD_REPLIES, filename)
@@ -707,8 +617,8 @@ def test_standard_replies():
            continue
        with open(filename) as f:
            message = email.message_from_file(f)
-            body = next(email.iterators.typed_subpart_iterator(message, subtype='plain'))
-            text = ''.join(body_iterator(body, True))
+            body = email.iterators.typed_subpart_iterator(message, subtype='plain').next()
+            text = ''.join(email.iterators.body_line_iterator(body, True))

            stripped_text = quotations.extract_from_plain(text)
            reply_text_fn = filename[:-4] + '_reply_text'
@@ -721,52 +631,3 @@ def test_standard_replies():
                "'%(reply)s' != %(stripped)s for %(fn)s" % \
                {'reply': reply_text, 'stripped': stripped_text,
                 'fn': filename}
-
-
-def test_split_email():
-    msg = """From: Mr. X
-    Date: 24 February 2016
-    To: Mr. Y
-    Subject: Hi
-    Attachments: none
-    Goodbye.
-    From: Mr. Y
-    To: Mr. X
-    Date: 24 February 2016
-    Subject: Hi
-    Attachments: none
-
-    Hello.
-
-        On 24th February 2016 at 09.32am, Conal wrote:
-
-        Hey!
-
-        On Mon, 2016-10-03 at 09:45 -0600, Stangel, Dan wrote:
-        > Mohan,
-        >
-        > We have not yet migrated the systems.
-        >
-        > Dan
-        >
-        > > -----Original Message-----
-        > > Date: Mon, 2 Apr 2012 17:44:22 +0400
-        > > Subject: Test
-        > > From: bob@xxx.mailgun.org
-        > > To: xxx@gmail.com; xxx@hotmail.com; xxx@yahoo.com; xxx@aol.com; xxx@comcast.net; xxx@nyc.rr.com
-        > >
-        > > Hi
-        > >
-        > > > From: bob@xxx.mailgun.org
-        > > > To: xxx@gmail.com; xxx@hotmail.com; xxx@yahoo.com; xxx@aol.com; xxx@comcast.net; xxx@nyc.rr.com
-        > > > Date: Mon, 2 Apr 2012 17:44:22 +0400
-        > > > Subject: Test
-        > > > Hi
-        > > >
-        > >
-        >
-        >
-"""
-    expected_markers = "stttttsttttetesetesmmmmmmssmmmmmmsmmmmmmmm"
-    markers = quotations.split_emails(msg)
-    eq_(markers, expected_markers)
--- a/tests/utils_test.py
+++ b/tests/utils_test.py
@@ -1,162 +1,9 @@
-# coding:utf-8
-
-from __future__ import absolute_import
 from . import *

-from talon import utils as u
-import cchardet
-import six
-from lxml import html
+from talon import utils


 def test_get_delimiter():
-    eq_('\r\n', u.get_delimiter('abc\r\n123'))
-    eq_('\n', u.get_delimiter('abc\n123'))
-    eq_('\n', u.get_delimiter('abc'))
-
-
-def test_unicode():
-    eq_ (u'hi', u.to_unicode('hi'))
-    eq_ (type(u.to_unicode('hi')), six.text_type )
-    eq_ (type(u.to_unicode(u'hi')), six.text_type )
-    eq_ (type(u.to_unicode('привет')), six.text_type )
-    eq_ (type(u.to_unicode(u'привет')), six.text_type )
-    eq_ (u"привет", u.to_unicode('привет'))
-    eq_ (u"привет", u.to_unicode(u'привет'))
-    # some latin1 stuff
-    eq_ (u"Versión", u.to_unicode(u'Versi\xf3n'.encode('iso-8859-2'), precise=True))
-
-
-def test_detect_encoding():
-    eq_ ('ascii', u.detect_encoding(b'qwe').lower())
-    ok_ (u.detect_encoding(
-        u'Versi\xf3n'.encode('iso-8859-2')).lower() in [
-            'iso-8859-1', 'iso-8859-2'])
-    eq_ ('utf-8', u.detect_encoding(u'привет'.encode('utf8')).lower())
-    # fallback to utf-8
-    with patch.object(u.chardet, 'detect') as detect:
-        detect.side_effect = Exception
-        eq_ ('utf-8', u.detect_encoding('qwe'.encode('utf8')).lower())
-
-
-def test_quick_detect_encoding():
-    eq_ ('ascii', u.quick_detect_encoding(b'qwe').lower())
-    ok_ (u.quick_detect_encoding(
-        u'Versi\xf3n'.encode('windows-1252')).lower() in [
-            'windows-1252', 'windows-1250'])
-    eq_ ('utf-8', u.quick_detect_encoding(u'привет'.encode('utf8')).lower())
-
-
-@patch.object(cchardet, 'detect')
-@patch.object(u, 'detect_encoding')
-def test_quick_detect_encoding_edge_cases(detect_encoding, cchardet_detect):
-    cchardet_detect.return_value = {'encoding': 'ascii'}
-    eq_('ascii', u.quick_detect_encoding(b"qwe"))
-    cchardet_detect.assert_called_once_with(b"qwe")
-
-    # fallback to detect_encoding
-    cchardet_detect.return_value = {}
-    detect_encoding.return_value = 'utf-8'
-    eq_('utf-8', u.quick_detect_encoding(b"qwe"))
-
-    # exception
-    detect_encoding.reset_mock()
-    cchardet_detect.side_effect = Exception()
-    detect_encoding.return_value = 'utf-8'
-    eq_('utf-8', u.quick_detect_encoding(b"qwe"))
-    ok_(detect_encoding.called)
-
-
-def test_html_to_text():
-    html = """<body>
-<p>Hello world!</p>
-<br>
-<ul>
-<li>One!</li>
-<li>Two</li>
-</ul>
-<p>
-Haha
-</p>
-</body>"""
-    text = u.html_to_text(html)
-    eq_(b"Hello world! \n\n  * One! \n  * Two \nHaha", text)
-    eq_(u"привет!", u.html_to_text("<b>привет!</b>").decode('utf8'))
-
-    html = '<body><br/><br/>Hi</body>'
-    eq_ (b'Hi', u.html_to_text(html))
-
-    html = """Hi
-<style type="text/css">
-
-div, p, li {
-
-font: 13px 'Lucida Grande', Arial, sans-serif;
-
-}
-</style>
-
-<style type="text/css">
-
-h1 {
-
-font: 13px 'Lucida Grande', Arial, sans-serif;
-
-}
-</style>"""
-    eq_ (b'Hi', u.html_to_text(html))
-
-    html = """<div>
-<!-- COMMENT 1 -->
-<span>TEXT 1</span>
-<p>TEXT 2 <!-- COMMENT 2 --></p>
-</div>"""
-    eq_(b'TEXT 1 \nTEXT 2', u.html_to_text(html))
-
-
-def test_comment_no_parent():
-    s = "<!-- COMMENT 1 --> no comment"
-    d = u.html_document_fromstring(s)
-    eq_("no comment", u.html_tree_to_text(d))
-
-
-@patch.object(u.html5parser, 'fromstring', Mock(side_effect=Exception()))
-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_("Hello", u.html_to_text("<div>Hello</div>"))
-    eq_(None, u.html_to_text("<div><span>Hi</span></div>"))
+    eq_('\r\n', utils.get_delimiter('abc\r\n123'))
+    eq_('\n', utils.get_delimiter('abc\n123'))
+    eq_('\n', utils.get_delimiter('abc'))
--- a/train.py
+++ b/train.py
@@ -1,4 +1,3 @@
-from __future__ import absolute_import
 from talon.signature import EXTRACTOR_FILENAME, EXTRACTOR_DATA
 from talon.signature.learning.classifier import train, init