From 29722b72667be38ae94d7b9ff99c8623cfa7b2be Mon Sep 17 00:00:00 2001 From: qvalentin Date: Sun, 29 Aug 2021 18:28:34 +0200 Subject: [PATCH] improve javaScript --- server.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/server.py b/server.py index f4127a6..41116c9 100644 --- a/server.py +++ b/server.py @@ -17,10 +17,11 @@ from pygments.formatters.html import HtmlFormatter from pygments.lexers import guess_lexer_for_filename -def render_gist_js(code): - template = Template("document.write( '{{ code|tojson }}' );") +def render_gist_js(code,path): + template = Template("""script = document.querySelector('script[src$="{{ path }}"]') + script.insertAdjacentHTML( 'afterend','{{ code|tojson }}' );""") - return template.render(code=code) + return template.render(code=code,path=path) class HTTPRequestHandler(BaseHTTPRequestHandler): @@ -60,15 +61,13 @@ class HTTPRequestHandler(BaseHTTPRequestHandler): result = highlighted_code + f"" - js_rendered = render_gist_js(result) + js_rendered = render_gist_js(result,self.path) self.send_response(200) self.send_header('Content-Type', 'text/html') self.end_headers() self.wfile.write(js_rendered.encode('utf8')) - - else: self.send_response(404)