diff options
Diffstat (limited to 'backend/injector.py')
| -rw-r--r-- | backend/injector.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/backend/injector.py b/backend/injector.py index 16ced852..9b4fe353 100644 --- a/backend/injector.py +++ b/backend/injector.py @@ -48,6 +48,10 @@ class Tab: await self.client.close() return res + async def get_steam_resource(self, url): + res = await self.evaluate_js(f'(async function test() {{ return await (await fetch("{url}")).text() }})()', True) + return res["result"]["result"]["value"] + def __repr__(self): return self.title @@ -93,3 +97,15 @@ async def tab_has_global_var(tab_name, var_name): return False return res["result"]["result"]["value"] + +async def tab_has_element(tab_name, element_name): + try: + tab = await get_tab(tab_name) + except ValueError: + return False + res = await tab.evaluate_js(f"document.getElementById('{element_name}') != null", False) + + if not "result" in res or not "result" in res["result"] or not "value" in res["result"]["result"]: + return False + + return res["result"]["result"]["value"]
\ No newline at end of file |