39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
#!/usr/bin/env python3
|
|
import json
|
|
import sys
|
|
import gi
|
|
gi.require_version('Atspi', '2.0')
|
|
from gi.repository import Atspi
|
|
|
|
mode, limit = sys.argv[1], int(sys.argv[2])
|
|
Atspi.init()
|
|
desktop = Atspi.get_desktop(0)
|
|
nodes = []
|
|
def walk(node, depth=0):
|
|
if len(nodes) >= limit or node is None or depth > 30:
|
|
return
|
|
try:
|
|
role = node.get_role_name() or ''
|
|
name = node.get_name() or ''
|
|
component = node.get_component()
|
|
rect = component.get_extents(Atspi.CoordType.SCREEN) if component else None
|
|
if role and (name or rect):
|
|
nodes.append({'role': role, 'name': name, 'rect': [rect.x, rect.y, rect.width, rect.height] if rect else None, 'state': [str(s) for s in node.get_state_set().get_states()]})
|
|
for i in range(node.get_child_count()):
|
|
walk(node.get_child_at_index(i), depth + 1)
|
|
except Exception:
|
|
return
|
|
if mode == 'focused':
|
|
for i in range(desktop.get_child_count()):
|
|
app = desktop.get_child_at_index(i)
|
|
try:
|
|
for j in range(app.get_child_count()):
|
|
window = app.get_child_at_index(j)
|
|
if window.get_state_set().contains(Atspi.StateType.ACTIVE):
|
|
walk(window)
|
|
except Exception:
|
|
continue
|
|
else:
|
|
walk(desktop)
|
|
print(json.dumps(nodes, ensure_ascii=False))
|