Add allure support to ydb github export

d6cba27d09fb5e50a99c36070a6a3545c8393ea1

1 files changed, 74 insertions, 0 deletions

diff --git a/contrib/python/allure-python-commons/allure_commons/logger.py b/contrib/python/allure-python-commons/allure_commons/logger.py
new file mode 100644
index 0000000000..d0ac1e2491
--- /dev/null
+++ b/contrib/python/allure-python-commons/allure_commons/logger.py
@@ -0,0 +1,74 @@
+import io
+import os
+from pathlib import Path
+import json
+import uuid
+import shutil
+from attr import asdict
+from allure_commons import hookimpl
+
+INDENT = 4
+
+
+class AllureFileLogger:
+
+    def __init__(self, report_dir, clean=False):
+        self._report_dir = Path(report_dir).absolute()
+        if self._report_dir.is_dir() and clean:
+            shutil.rmtree(self._report_dir)
+        self._report_dir.mkdir(parents=True, exist_ok=True)
+
+    def _report_item(self, item):
+        indent = INDENT if os.environ.get("ALLURE_INDENT_OUTPUT") else None
+        filename = item.file_pattern.format(prefix=uuid.uuid4())
+        data = asdict(item, filter=lambda _, v: v or v is False)
+        with io.open(self._report_dir / filename, 'w', encoding='utf8') as json_file:
+            json.dump(data, json_file, indent=indent, ensure_ascii=False)
+
+    @hookimpl
+    def report_result(self, result):
+        self._report_item(result)
+
+    @hookimpl
+    def report_container(self, container):
+        self._report_item(container)
+
+    @hookimpl
+    def report_attached_file(self, source, file_name):
+        destination = self._report_dir / file_name
+        shutil.copy2(source, destination)
+
+    @hookimpl
+    def report_attached_data(self, body, file_name):
+        destination = self._report_dir / file_name
+        with open(destination, 'wb') as attached_file:
+            if isinstance(body, str):
+                attached_file.write(body.encode('utf-8'))
+            else:
+                attached_file.write(body)
+
+
+class AllureMemoryLogger:
+
+    def __init__(self):
+        self.test_cases = []
+        self.test_containers = []
+        self.attachments = {}
+
+    @hookimpl
+    def report_result(self, result):
+        data = asdict(result, filter=lambda _, v: v or v is False)
+        self.test_cases.append(data)
+
+    @hookimpl
+    def report_container(self, container):
+        data = asdict(container, filter=lambda _, v: v or v is False)
+        self.test_containers.append(data)
+
+    @hookimpl
+    def report_attached_file(self, source, file_name):
+        self.attachments[file_name] = source
+
+    @hookimpl
+    def report_attached_data(self, body, file_name):
+        self.attachments[file_name] = body