aboutsummaryrefslogtreecommitdiff
path: root/mammoth/writers/markdown.py
blob: 76e4f90335914124311b7a98fa2a69d9b7b8f228 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
from __future__ import unicode_literals

import re


class MarkdownWriter(object):
    def __init__(self):
        self._fragments = []
    
    def text(self, text):
        self._fragments.append(_escape_markdown(text))
    
    def start(self, name, attributes=None):
        pass

    def end(self, name):
        if name == "p":
            self._fragments.append("\n\n")
    
    def self_closing(self, name, attributes=None):
        pass
    
    def append(self, other):
        self._fragments.append(other)
    
    def as_string(self):
        return "".join(self._fragments)


def _escape_markdown(value):
    return re.sub(r"([\`\*_\{\}\[\]\(\)\#\+\-\.\!])", r"\\\1", re.sub("\\\\", "\\\\\\\\", value))