5 # Languages mapping as used by markdown/pandoc
6 shortname2language
= dict(
20 def read(text
, argv
=sys
.argv
[3:]):
21 lines
= text
.splitlines()
22 # First read all include statements
23 for i
in range(len(lines
)):
24 if lines
[i
].startswith('#include "'):
25 filename
= lines
[i
].split('"')[1]
26 with open(filename
, 'r') as f
:
27 include_text
= f
.read()
28 lines
[i
] = include_text
29 text
= '\n'.join(lines
)
34 key
, value
= arg
.split('=')
35 mako_kwargs
[key
] = value
41 print('Cannot import mako - mako is not run')
45 from mako
.template
import Template
46 from mako
.lookup
import TemplateLookup
47 lookup
= TemplateLookup(directories
=[os
.curdir
])
48 # text = text.encode('utf-8')
49 temp
= Template(text
=text
, lookup
=lookup
, strict_undefined
=True)
50 text
= temp
.render(**mako_kwargs
)
53 lines
= text
.splitlines()
55 inside
= None # indicates which type of cell we are inside
56 fullname
= None # full language name in code cells
58 if line
.startswith('-----'):
59 # New cell, what type?
60 m
= re
.search(r
'-----([a-z0-9-]+)?', line
)
62 shortname
= m
.group(1)
64 # Check if code is to be typeset as static
65 # Markdown code (e.g., shortname=py-t)
66 astext
= shortname
[-2:] == '-t'
69 shortname
= shortname
[:-2]
71 cells
.append(['markdown', 'code', ['\n']])
72 cells
[-1][2].append('```%s\n' % fullname
)
75 if shortname
in shortname2language
:
76 fullname
= shortname2language
[shortname
]
78 cells
.append(['codecell', fullname
, []])
82 cells
.append(['markdown', 'text', ['\n']])
84 raise SyntaxError('Wrong syntax of cell delimiter:\n{}'
87 # Ordinary line in a cell
88 if inside
in ('markdown', 'codecell'):
89 cells
[-1][2].append(line
)
91 raise SyntaxError('line\n {}\nhas no beginning cell delimiter'
93 # Merge the lines in each cell to a string
94 for i
in range(len(cells
)):
95 if cells
[i
][0] == 'markdown' and cells
[i
][1] == 'code':
96 # Add an ending ``` of code
97 cells
[i
][2].append('```\n')
98 cells
[i
][2] = '\n'.join(cells
[i
][2])
102 """Turn cells list into valid IPython notebook code."""
103 # Use Jupyter nbformat functionality for writing the notebook
105 from nbformat
.v4
import (
106 new_code_cell
, new_markdown_cell
, new_notebook
, writes
)
109 for cell_tp
, language
, block
in cells
:
110 if cell_tp
== 'markdown':
112 new_markdown_cell(source
=block
))
113 elif cell_tp
== 'codecell':
114 nb_cells
.append(new_code_cell(source
=block
))
116 nb
= new_notebook(cells
=nb_cells
)
121 """Compile a document and its variables."""
123 inputfile
= sys
.argv
[1]
124 with open(inputfile
, 'r') as f
:
126 # Assuming file extension .gj (generate Jupyter); TODO: less strict
127 outputfile
= inputfile
[:-3]+'.ipynb' if (len(sys
.argv
)<=2 or sys
.argv
[2]=='-') \
129 except (IndexError, IOError) as e
:
130 print('Usage: %s inputfile [outputfile|- [Mako args]]' % (sys
.argv
[0]))
133 cells
= read(text
, argv
=sys
.argv
[3:])
134 filestr
= write(cells
)
135 with open(outputfile
, 'w') as f
:
138 if __name__
== '__main__':