Improvements with Steam.

This commit is contained in:
Storm Dragon
2026-04-07 19:03:46 -04:00
parent 633ff214a4
commit 2d9790de88
2 changed files with 149 additions and 0 deletions
@@ -44,6 +44,8 @@ class Utilities(ChromiumUtilities):
def clearSteamVirtualizedListCaches(self) -> None:
self.clearContentCache()
self._steamInferredButtonLabels = {}
self._isUselessImage = {}
self._shouldFilter = {}
def isSteamVirtualizedList(self, obj) -> bool:
if not (obj and self.inDocumentContent(obj)):
@@ -76,6 +78,21 @@ class Utilities(ChromiumUtilities):
cache[obj] = inferredLabel
return inferredLabel
def isUselessImage(self, obj) -> bool:
if not (obj and self.inDocumentContent(obj)):
return super().isUselessImage(obj)
cached = self._isUselessImage.get(hash(obj))
if cached is not None:
return cached
rv = super().isUselessImage(obj)
if not rv:
rv = self._isRedundantSteamImage(obj)
self._isUselessImage[hash(obj)] = rv
return rv
def _shouldInferSteamButtonLabel(self, obj) -> bool:
if not (obj and self.inDocumentContent(obj)):
return False
@@ -113,6 +130,34 @@ class Utilities(ChromiumUtilities):
return "Add Friend"
return ""
def _isRedundantSteamImage(self, obj) -> bool:
if not AXUtilities.is_image_or_canvas(obj):
return False
if AXObject.get_name(obj) or AXObject.get_description(obj):
return False
if AXObject.get_child_count(obj):
return False
if AXUtilities.is_focusable(obj):
return False
if not AXObject.has_action(obj, "click-ancestor"):
return False
roleDescription = self._normalizeSteamLabelText(AXObject.get_role_description(obj) or "")
if roleDescription and roleDescription.casefold() not in ["unlabeled image", "image"]:
return False
nearbyLabel = self._getSteamNearbyImageLabel(obj)
if not self._isUsefulSteamLabel(nearbyLabel):
return False
tokens = ["STEAM: Treating redundant image as useless:", obj, "(label:", nearbyLabel, ")"]
debug.printTokens(debug.LEVEL_INFO, tokens, True)
return True
def _getSteamNearbyButtonLabel(self, obj) -> str:
parent = AXObject.get_parent(obj)
if parent is None:
@@ -132,6 +177,25 @@ class Utilities(ChromiumUtilities):
return self._getSteamLabelFromChildren(grandParent, ignore=parent)
def _getSteamNearbyImageLabel(self, obj) -> str:
parent = AXObject.get_parent(obj)
if parent is None:
return ""
siblingLabel = self._getSteamLabelFromChildren(parent, ignore=obj)
if siblingLabel:
return siblingLabel
parentLabel = self._getSteamReadableText(parent)
if self._isUsefulSteamLabel(parentLabel):
return parentLabel
grandParent = AXObject.get_parent(parent)
if grandParent is None:
return ""
return self._getSteamLabelFromChildren(grandParent, ignore=parent)
def _getSteamLabelFromChildren(self, obj, ignore=None) -> str:
for child in AXObject.iter_children(obj):
if child == ignore:
+85
View File
@@ -267,5 +267,90 @@ class SteamLabelRecoveryTests(unittest.TestCase):
self.assertEqual(utilities.displayedLabel(button), "Add Friend")
class SteamRedundantImageTests(unittest.TestCase):
def test_is_useless_image_treats_unlabeled_click_ancestor_image_with_nearby_text_as_useless(self):
testScript = mock.Mock(generatorCache={})
utilities = steam_script_utilities.Utilities(testScript)
image = object()
parent = object()
entry = object()
utilities.inDocumentContent = mock.Mock(return_value=True)
def get_parent(obj):
if obj in (image, entry):
return parent
return None
def get_name(obj):
if obj is entry:
return "Search for games or profiles..."
return ""
def get_child_count(obj):
if obj is parent:
return 2
return 0
def has_action(obj, actionName):
return obj is image and actionName == "click-ancestor"
def iter_children(obj, pred=None):
children = [image, entry] if obj is parent else []
if pred is not None:
children = [child for child in children if pred(child)]
return iter(children)
with (
mock.patch.object(steam_script_utilities.ChromiumUtilities, "isUselessImage", return_value=False),
mock.patch.object(steam_script_utilities.AXObject, "get_parent", side_effect=get_parent),
mock.patch.object(steam_script_utilities.AXObject, "get_name", side_effect=get_name),
mock.patch.object(steam_script_utilities.AXObject, "get_description", return_value=""),
mock.patch.object(steam_script_utilities.AXObject, "get_child_count", side_effect=get_child_count),
mock.patch.object(steam_script_utilities.AXObject, "has_action", side_effect=has_action),
mock.patch.object(steam_script_utilities.AXObject, "iter_children", side_effect=iter_children),
mock.patch.object(steam_script_utilities.AXObject, "supports_text", return_value=False),
mock.patch.object(
steam_script_utilities.AXObject,
"get_role_description",
side_effect=lambda obj: "Unlabeled image" if obj is image else "",
),
mock.patch.object(
steam_script_utilities.AXUtilities,
"is_image_or_canvas",
side_effect=lambda obj: obj is image,
),
mock.patch.object(steam_script_utilities.AXUtilities, "is_focusable", return_value=False),
mock.patch.object(steam_script_utilities.AXUtilities, "is_button", return_value=False),
mock.patch.object(steam_script_utilities.AXUtilities, "is_push_button", return_value=False),
):
self.assertTrue(utilities.isUselessImage(image))
def test_is_useless_image_defers_to_generic_logic_without_nearby_text(self):
testScript = mock.Mock(generatorCache={})
utilities = steam_script_utilities.Utilities(testScript)
image = object()
utilities.inDocumentContent = mock.Mock(return_value=True)
with (
mock.patch.object(steam_script_utilities.ChromiumUtilities, "isUselessImage", return_value=False),
mock.patch.object(steam_script_utilities.AXObject, "get_parent", return_value=None),
mock.patch.object(steam_script_utilities.AXObject, "get_name", return_value=""),
mock.patch.object(steam_script_utilities.AXObject, "get_description", return_value=""),
mock.patch.object(steam_script_utilities.AXObject, "get_child_count", return_value=0),
mock.patch.object(steam_script_utilities.AXObject, "has_action", return_value=True),
mock.patch.object(steam_script_utilities.AXObject, "supports_text", return_value=False),
mock.patch.object(steam_script_utilities.AXObject, "get_role_description", return_value=""),
mock.patch.object(
steam_script_utilities.AXUtilities,
"is_image_or_canvas",
side_effect=lambda obj: obj is image,
),
mock.patch.object(steam_script_utilities.AXUtilities, "is_focusable", return_value=False),
):
self.assertFalse(utilities.isUselessImage(image))
if __name__ == "__main__":
unittest.main()