Function titlecase

Lib/pagebot/contributions/filibuster/titlecase.py:23–58

Titlecases input text. This filter changes all words to Title Caps, and attempts to be clever about *uncapitalizing* SMALL words like a/an/the in the input. The list of "SMALL words" which are not capped comes from the New York Times Manual of Style, plus 'vs' and 'v'.

(text)

SUBPHRASE = re.compile('([:.;?!][ ])(%s)' % SMALL)

def titlecase(text):
    """Titlecases input text.

    This filter changes all words to Title Caps, and attempts to be clever
    about *uncapitalizing* SMALL words like a/an/the in the input.

    The list of "SMALL words" which are not capped comes from the New York
    Times Manual of Style, plus 'vs' and 'v'.
    """
    words = re.split(r'\s', text)
    line = []

    for word in words:
        if INLINE_PERIOD.search(word) or UC_ELSEWHERE.match(word):
            line.append(word)
            continue
        if SMALL_WORDS.match(word):
            line.append(word.lower())
            continue
        line.append(CAPFIRST.sub(lambda m: m.group(0).upper(), word))

    line = " ".join(line)

    line = SMALL_FIRST.sub(lambda m: '%s%s' % (
        m.group(1),
        m.group(2).capitalize()
    ), line)

    line = SMALL_LAST.sub(lambda m: m.group(0).capitalize(), line)

    line = SUBPHRASE.sub(lambda m: '%s%s' % (
        m.group(1),
        m.group(2).capitalize()
    ), line)

    return line

class TitlecaseTests(unittest.TestCase):
    """Tests to ensure titlecase follows all of the rules"""