33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
|
from selenium import webdriver
|
||
|
|
from selenium.webdriver.common.by import By
|
||
|
|
import os
|
||
|
|
import json
|
||
|
|
|
||
|
|
|
||
|
|
def is_english_content(
|
||
|
|
driver: webdriver.firefox.webdriver.WebDriver, content_id: int
|
||
|
|
) -> bool:
|
||
|
|
string_json_path: str = os.path.join("content", "is_english_content.json")
|
||
|
|
|
||
|
|
with open(string_json_path, "r") as file:
|
||
|
|
string_dict = json.load(file)
|
||
|
|
|
||
|
|
found_element_list = driver.find_elements(By.TAG_NAME, "div")
|
||
|
|
|
||
|
|
for element in found_element_list:
|
||
|
|
dtab = str(element.get_dom_attribute("data-table"))
|
||
|
|
duid = str(element.get_dom_attribute("data-uid"))
|
||
|
|
dtyp = str(element.get_dom_attribute("data-ctype"))
|
||
|
|
if (
|
||
|
|
(dtab == "tt_content")
|
||
|
|
and (duid == str(content_id))
|
||
|
|
and (dtyp == "textmedia")
|
||
|
|
):
|
||
|
|
sub_element_list = element.find_elements(By.TAG_NAME, "img")
|
||
|
|
for sub_element in sub_element_list:
|
||
|
|
img_name = str(sub_element.get_dom_attribute("src"))
|
||
|
|
if img_name.endswith(string_dict["search_for_string"]):
|
||
|
|
return True
|
||
|
|
|
||
|
|
return False
|