process.py

import bleach
import re

def clean_input(string):
    """
    Sanatises invalid tags/attributes & strip white space from strings
    """
    allowed_tags = ['\r', '\n',
                    'a', 'abbr', 'acronym', 'address', 'b', 'br', 'div', 'dl', 'dt',
                    'em', 'h1', 'h2', 'h3', 'h4', 'h5', 'h6', 'hr', 'i', 'img',
                    'li', 'ol', 'p', 'pre', 'q', 's', 'small', 'strike', 'strong',
                    'span', 'sub', 'sup', 'table', 'tbody', 'td', 'tfoot', 'th',
                    'thead', 'tr', 'tt', 'u', 'ul']

    allowed_attrs = {
        'a': ['href', 'target', 'title'],
        'img': ['src', 'alt', 'width', 'height'],
    }   

    string = bleach.clean(string, tags=allowed_tags, attributes=allowed_attrs)

    string = string.rstrip().lstrip()
    return string


def title_process(title):
    """
    Process title field.
    """
    title = clean_input(title)
    if len(title) == 0:
        title = "your wall of text"
    output_title = ""
    list_of_words = title.split()
    for elem in list_of_words:
        if len(output_title) > 0:
            output_title = output_title + " " + elem.strip().capitalize()
        else:
            output_title = elem.capitalize()
    if not output_title:
        return title
    else:
        return output_title


def text_process(text):
    """ 
    Process user input field.

    text, takes user sumitted plain text and formats it as per the WoT
    specification.
    """
    if (text[-1] != "."):
        text = text + "."
    text = re.sub("\r\n\r\n", "<hr>", text)
    text = clean_input(text) 
    text_list = re.findall('.*?[.!\?]+', text)
    # TODO: also need to check for speach in text ""
    new_list = []

    for item in text_list:
        new_string = item.lstrip()
        new_string = new_string.capitalize()
        new_string = "<li>" + new_string + "</li>"
        new_string = re.sub('<li><hr>', '<hr><br><li>', new_string)
        # FIXME: edge case, this does not capatalize first letter on new line
        # due to additional <hr> tags at start of list item.:w
        new_list.append(new_string)
    
    str1 = ""  
    
    for ele in new_list:  
        str1 += ele   
   
    return str1