6
6
7
7
8
8
class Chart (Renderer ):
9
- def render_sheet (self , sheet , chart_type = 'bar' ,
10
- embed = False , ** keywords ):
9
+ def render_sheet (self , sheet , chart_type = 'bar' , embed = False , ** keywords ):
11
10
charter = MANAGER .get_a_plugin (chart_type )
12
11
chart_instance = charter .render_sheet (
13
12
sheet , ** keywords )
14
13
15
- self ._write_content (chart_instance )
14
+ self ._write_content (chart_instance , embed )
16
15
17
16
with tempfile .NamedTemporaryFile (suffix = ".html" ) as fout :
18
17
chart_instance .render (path = fout .name )
@@ -23,13 +22,16 @@ def render_book(self, book, chart_type='bar', embed=False, **keywords):
23
22
charter = MANAGER .get_a_plugin (chart_type )
24
23
chart_instance = charter .render_book (book ,
25
24
** keywords )
26
- self ._write_content (chart_instance )
25
+ self ._write_content (chart_instance , embed )
27
26
28
- def _write_content (self , instance ):
29
- with tempfile .NamedTemporaryFile (suffix = ".html" ) as fout :
30
- instance .render (path = fout .name )
31
- fout .seek (0 )
32
- if PY2 :
33
- self ._stream .write (fout .read ())
34
- else :
35
- self ._stream .write (fout .read ().decode ('utf-8' ))
27
+ def _write_content (self , instance , embed ):
28
+ if embed :
29
+ content = instance .render_embed ()
30
+ else :
31
+ with tempfile .NamedTemporaryFile (suffix = ".html" ) as fout :
32
+ instance .render (path = fout .name )
33
+ fout .seek (0 )
34
+ content = fout .read ()
35
+ if not PY2 :
36
+ content = content .decode ('utf-8' )
37
+ self ._stream .write (content )
0 commit comments