1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
from typing import List, Optional, Union
from webview.dom import ManipulationMode
from webview.dom.element import Element
from webview.util import escape_quotes
class DOM:
def __init__(self, window):
self.__window = window
window.events.loaded += self.__on_loaded
self._elements = {}
def __on_loaded(self):
self._elements = {}
@property
def body(self) -> Element:
self._elements.get('body', Element(self.__window, 'body'))
@property
def document(self) -> Element:
return self._elements.get('document', Element(self.__window, 'document'))
@property
def window(self) -> Element:
return self._elements.get('window', Element(self.__window, 'window'))
def create_element(self, html: str, parent: Union[Element, str]=None, mode=ManipulationMode.LastChild) -> Element:
self.__window.events.loaded.wait()
if isinstance(parent, Element):
parent_command = parent._query_command
elif isinstance(parent, str):
parent_command = f'var element = document.querySelector("{parent}");'
else:
parent_command = 'var element = document.body;'
node_id = self.__window.evaluate_js(f"""
{parent_command};
var template = document.createElement('template');
template.innerHTML = '{escape_quotes(html)}'.trim();
var newElement = template.content.firstChild;
pywebview._insertNode(newElement, element, '{mode.value}')
pywebview._getNodeId(newElement);
""")
return Element(self.__window, node_id)
def get_element(self, selector: str) -> Optional[Element]:
self.__window.events.loaded.wait()
node_id = self.__window.evaluate_js(f"""
var element = document.querySelector('{selector}');
pywebview._getNodeId(element);
""")
return Element(self.__window, node_id) if node_id else None
def get_elements(self, selector: str) -> List[Element]:
self.__window.events.loaded.wait()
code = f"""
var elements = document.querySelectorAll('{selector}');
nodeIds = [];
for (var i = 0; i < elements.length; i++) {{
var nodeId = pywebview._getNodeId(elements[i]);
nodeIds.push(nodeId);
}}
nodeIds
"""
node_ids = self.__window.evaluate_js(code)
return [Element(self.__window, node_id) for node_id in node_ids]
|
