33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from ..client import Client
|
|
from ..context import context
|
|
|
|
|
|
def add_head_html(code: str, *, shared: bool = False) -> None:
|
|
"""Add HTML code to the head of the page.
|
|
|
|
:param code: HTML code to add
|
|
:param shared: if True, the code is added to all pages
|
|
"""
|
|
if shared:
|
|
Client.shared_head_html += code + '\n'
|
|
else:
|
|
client = context.client
|
|
if client.has_socket_connection:
|
|
client.run_javascript(f'document.head.insertAdjacentHTML("beforeend", {code!r});')
|
|
client._head_html += code + '\n' # pylint: disable=protected-access
|
|
|
|
|
|
def add_body_html(code: str, *, shared: bool = False) -> None:
|
|
"""Add HTML code to the body of the page.
|
|
|
|
:param code: HTML code to add
|
|
:param shared: if True, the code is added to all pages
|
|
"""
|
|
if shared:
|
|
Client.shared_body_html += code + '\n'
|
|
else:
|
|
client = context.client
|
|
if client.has_socket_connection:
|
|
client.run_javascript(f'document.querySelector("#app").insertAdjacentHTML("beforebegin", {code!r});')
|
|
client._body_html += code + '\n' # pylint: disable=protected-access
|