|
0
|
1 |
########## converters |
|
|
2 |
from django.core.cache import cache |
|
|
3 |
|
|
|
4 |
# adapted [to django] from http://code.activestate.com/recipes/325205/ |
|
|
5 |
def memoize(f): |
|
|
6 |
def g(*args, **kwargs): |
|
|
7 |
key = ( f.__name__, f, tuple(args), frozenset(kwargs.items()) ) |
|
|
8 |
val = cache.get(key) |
|
|
9 |
if not val: |
|
|
10 |
val = f(*args, **kwargs) |
|
|
11 |
cache.set(key,val) |
|
|
12 |
return val |
|
|
13 |
return g |
|
|
14 |
|
|
|
15 |
|
|
|
16 |
def to_unicode(string): |
|
|
17 |
if type(string) != 'unicode': |
|
|
18 |
return string.decode('utf8') |
|
|
19 |
else: |
|
|
20 |
return string |
|
|
21 |
|
|
|
22 |
def to_utf8(string): |
|
|
23 |
if type(string) != 'str': |
|
|
24 |
return string.encode('utf8') |
|
|
25 |
else: |
|
|
26 |
return string |
|
|
27 |
|
|
|
28 |
#@memoize |
|
|
29 |
def rst_to_html(rst): |
|
|
30 |
from docutils import core, io |
|
|
31 |
html, pub = _get_html_and_pub(rst) |
|
|
32 |
parts = pub.writer.parts |
|
|
33 |
return parts['stylesheet']+parts['body'] |
|
|
34 |
|
|
|
35 |
#@memoize |
|
|
36 |
def rst_to_fullhtml(rst): |
|
|
37 |
html, pub = _get_html_and_pub(rst) |
|
|
38 |
parts = pub.writer.parts |
|
|
39 |
return html |
|
|
40 |
#return '<html><head>' + parts['stylesheet'] + '</head><body>' + parts['body'] + '</body></html>' |
|
|
41 |
|
|
|
42 |
def markdown_to_html(markdown): |
|
|
43 |
return _markdown_to_html(markdown) |
|
|
44 |
|
|
|
45 |
def markdown_to_fullhtml(markdown): |
|
|
46 |
return '<html><body>'+_markdown_to_html(markdown) + '</body></html>' |
|
|
47 |
|
|
|
48 |
def _markdown_to_html(markdown): |
|
|
49 |
from markdown import Markdown |
|
|
50 |
md = Markdown() |
|
|
51 |
html = md.convert(markdown) |
|
|
52 |
return html |
|
|
53 |
|
|
|
54 |
def _get_html_and_pub(rst): |
|
|
55 |
from docutils import core, io |
|
|
56 |
html, pub = core.publish_programmatically( |
|
|
57 |
source_class=io.StringInput, source=rst, |
|
|
58 |
source_path=None, |
|
|
59 |
destination_class=io.StringOutput, destination=None, |
|
|
60 |
destination_path=None, |
|
|
61 |
reader=None, reader_name='standalone', |
|
|
62 |
parser=None, parser_name='restructuredtext', |
|
|
63 |
writer=None, writer_name='HTML', |
|
|
64 |
settings=None, settings_spec=None, settings_overrides=None, |
|
|
65 |
config_section=None, enable_exit_status=None) |
|
|
66 |
return html, pub |
|
|
67 |
|
|
|
68 |
#@memoize |
|
|
69 |
def html_to_pdf(html): |
|
|
70 |
html = to_utf8(html) |
|
|
71 |
|
|
|
72 |
import sx.pisa3 as pisa |
|
|
73 |
import StringIO |
|
|
74 |
dst = StringIO.StringIO() |
|
|
75 |
result = pisa.CreatePDF(html, dst) |
|
|
76 |
if not result.err: |
|
|
77 |
pdf = dst.getvalue() |
|
|
78 |
dst.close() |
|
|
79 |
return pdf |
|
|
80 |
else: |
|
|
81 |
return None |
|
|
82 |
|
|
|
83 |
# http://www.aaronsw.com/2002/html2text/ |
|
|
84 |
#@memoize |
|
|
85 |
def html_to_markdown(html): |
|
|
86 |
from com.ext.html2text import html2text |
|
|
87 |
return html2text(html) |
|
|
88 |
|
|
|
89 |
########## formats |
|
|
90 |
|
|
|
91 |
FORMATS = { |
|
|
92 |
'HTML' : {'name': 'HTML', |
|
|
93 |
'to_format' : {'Markdown' : html_to_markdown, } |
|
|
94 |
}, |
|
|
95 |
'FULLHTML' : {'name': 'FULLHTML', |
|
|
96 |
'to_format' : {'PDF' : html_to_pdf, } |
|
|
97 |
}, |
|
|
98 |
'RST' : {'name': 'reStructuredText', |
|
|
99 |
'to_format' : {'HTML' : rst_to_html, |
|
|
100 |
'FULLHTML' : rst_to_fullhtml, |
|
|
101 |
} |
|
|
102 |
}, |
|
|
103 |
|
|
|
104 |
'Markdown' : {'name': 'Markdown', |
|
|
105 |
'to_format' : {'HTML' : markdown_to_html, |
|
|
106 |
'FULLHTML' : markdown_to_fullhtml, |
|
|
107 |
} |
|
|
108 |
}, |
|
|
109 |
'Textile' : {'name': 'Textile', |
|
|
110 |
}, |
|
|
111 |
'PDF' : {'name': 'PDF', |
|
|
112 |
}, |
|
|
113 |
} |
|
|
114 |
|
|
|
115 |
CHOICES_FORMATS = [ (k,v.get('name')) for k,v in FORMATS.items()] |
|
|
116 |
|
|
|
117 |
INPUT_FORMATS = ['RST','Markdown'] |
|
|
118 |
|
|
|
119 |
DEFAULT_INPUT_FORMAT = 'Markdown' |
|
|
120 |
|
|
|
121 |
CHOICES_INPUT_FORMATS = [ (k,v.get('name')) for k,v in FORMATS.items() if k in INPUT_FORMATS] |
|
|
122 |
|
|
|
123 |
def get_supported_conversions(from_format): |
|
|
124 |
return FORMATS[from_format]['to_format'].keys() |
|
|
125 |
|
|
|
126 |
def is_supported_conversion(from_format, to_format): |
|
|
127 |
infos = FORMATS.get(from_format) |
|
|
128 |
return infos.get('to_format') and infos.get('to_format').get(to_format) |
|
|
129 |
|
|
|
130 |
def convert(content, from_format, to_format): |
|
|
131 |
if is_supported_conversion(from_format, to_format): |
|
|
132 |
infos = FORMATS.get(from_format) |
|
|
133 |
conv_fun = infos.get('to_format').get(to_format) |
|
|
134 |
return conv_fun(content) |
|
|
135 |
else: |
|
|
136 |
pass |