108 lines
3.9 KiB
Python
108 lines
3.9 KiB
Python
"""Модуль для работы с JSON-контейнерами на веб-страницах.
|
||
|
||
Содержит компонент для чтения и проверки JSON-данных в контейнерах.
|
||
Использует playwright для взаимодействия с элементами страницы.
|
||
"""
|
||
|
||
import json
|
||
import jsondiff
|
||
from typing import Any, Dict
|
||
from playwright.sync_api import Page
|
||
from tools.logger import get_logger
|
||
from components.base_component import BaseComponent
|
||
|
||
logger = get_logger("JSON_CONTAINER")
|
||
|
||
|
||
class JsonContainerComponent(BaseComponent):
|
||
"""Компонент для работы с JSON-данными на странице.
|
||
|
||
Предоставляет методы чтения и проверки JSON-данных в контейнерах.
|
||
|
||
"""
|
||
|
||
def __init__(self, page: Page):
|
||
"""Инициализирует JSON-контейнер.
|
||
|
||
Args:
|
||
page: Экземпляр страницы Playwright.
|
||
"""
|
||
|
||
super().__init__(page)
|
||
|
||
# Действия:
|
||
def read_data(self, locator: Any) -> Dict:
|
||
"""Читает и форматирует JSON-данные из указанного локатора.
|
||
|
||
Args:
|
||
locator: Локатор элемента с JSON-данными.
|
||
|
||
Returns:
|
||
dict: Распарсенный JSON-объект.
|
||
|
||
Raises:
|
||
json.JSONDecodeError: Если данные не могут быть преобразованы в JSON.
|
||
"""
|
||
|
||
def format_json_string(json_string):
|
||
"""Форматирует строку JSON для корректного парсинга.
|
||
|
||
Args:
|
||
json_string: Сырая строка с JSON-данными.
|
||
|
||
Returns:
|
||
str: Отформатированная строка JSON.
|
||
"""
|
||
|
||
substrings = json_string.splitlines()
|
||
formatted_string_list = []
|
||
last_substring = substrings.pop()
|
||
|
||
for substring in substrings:
|
||
if substring.find(':') == -1:
|
||
if substring == '{':
|
||
formatted_string_list.append(substring)
|
||
elif substring == '}':
|
||
s1 = formatted_string_list.pop()
|
||
formatted_string_list.append(s1.rstrip(','))
|
||
formatted_string_list.append(substring + ',')
|
||
else:
|
||
formatted_string_list.append(substring + ',')
|
||
continue
|
||
|
||
key, value = substring.split(':')
|
||
s = ':'.join(['"' + key + '" ', " " + value])
|
||
|
||
if value == '{':
|
||
formatted_string_list.append(s)
|
||
else:
|
||
formatted_string_list.append(s + ',')
|
||
|
||
s2 = formatted_string_list.pop()
|
||
formatted_string_list.append(s2.rstrip(','))
|
||
formatted_string_list.append(last_substring)
|
||
|
||
return " ".join(formatted_string_list)
|
||
|
||
# Чтение JSON-содержимого из рабочей области
|
||
loc = self.get_locator(locator)
|
||
json_string = loc.inner_text()
|
||
formatted_json_string = format_json_string(json_string)
|
||
return json.loads(formatted_json_string)
|
||
|
||
# Проверки:
|
||
def check_json_equals(self, actual: Any, expected: Any, msg: str) -> None:
|
||
"""Сравнивает JSON-объекты на идентичность.
|
||
|
||
Args:
|
||
actual: Фактический JSON-объект.
|
||
expected: Ожидаемый JSON-объект.
|
||
msg: Сообщение об ошибке.
|
||
|
||
Raises:
|
||
AssertionError: Если объекты не идентичны.
|
||
"""
|
||
|
||
diff = jsondiff.diff(expected, actual, syntax='symmetric')
|
||
assert len(diff) == 0, f"{msg}. DIFF is {diff}"
|