CIRCL
/
lookyloo
mirror of https://github.com/CIRCL/lookyloo
2
0
Fork 0
Code Issues Releases Wiki Activity

Make it possible to render tree containing ImgFace 

See https://github.com/etetoolkit/ete/issues/302
pull/1/head
Raphaël Vinot 2017-09-03 01:14:21 +02:00
parent 7cfd51bbbe
commit b059c7b1f3
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lookyloo/ete3_webserver/tree_handler.py
View File

@ -4,6 +4,8 @@ import random
# import logging as log # import logging as log
from ete3 import Tree # , TreeStyle from ete3 import Tree # , TreeStyle
from ete3.parser.newick import NewickError from ete3.parser.newick import NewickError
import tempfile
import base64
def timeit(f): def timeit(f):
@ -38,10 +40,13 @@ class WebTreeHandler(object):
@timeit @timeit
def redraw(self): def redraw(self):
base64_img, img_map = self.tree.render("%%return.PNG", tree_style=self.tree.tree_style) with tempfile.NamedTemporaryFile(suffix='.PNG') as temp:
img_map = self.tree.render(temp.name, tree_style=self.tree.tree_style)
temp.seek(0)
base64_img = base64.b64encode(temp.read())
nodes, faces = self.get_html_map(img_map) nodes, faces = self.get_html_map(img_map)
base64 = base64_img.data().decode() base64_img = base64_img.decode()
return nodes, faces, base64 return nodes, faces, base64_img
def get_html_map(self, img_map): def get_html_map(self, img_map):
nodes = [] nodes = []