flyte 0.0.1b3__py3-none-any.whl → 0.2.0a0__py3-none-any.whl

union/remote/_logs.py

@@ -1,74 +0,0 @@
-import asyncio
-from dataclasses import dataclass
-from typing import AsyncGenerator, AsyncIterator
-
-from rich.console import Console
-from rich.live import Live
-from rich.panel import Panel
-from rich.text import Text
-
-from union._api_commons import syncer
-from union._initialize import get_client, requires_client
-from union._protos.logs.dataplane import payload_pb2
-from union._protos.workflow import run_definition_pb2, run_logs_service_pb2
-
-
-class AsyncLogViewer:
-    def __init__(self, log_source: AsyncIterator, max_lines: int = 30):
-        self.console = Console()
-        self.log_source = log_source
-        self.max_lines = max_lines
-        self.lines = []
-
-    def _format_line(self, logline: payload_pb2.LogLine) -> Text:
-        style_map = {
-            payload_pb2.LogLineOriginator.SYSTEM: "dim",
-            payload_pb2.LogLineOriginator.USER: "cyan",
-            payload_pb2.LogLineOriginator.UNKNOWN: "light red",
-        }
-        style = style_map.get(logline.originator, "")
-        return Text(
-            f"[{logline.originator}] [{logline.timestamp.ToDatetime().isoformat()}]{logline.message}", style=style
-        )
-
-    def _render(self):
-        log_text = Text()
-        for line in self.lines[-self.max_lines :]:
-            log_text.append(line + "\n")
-        return Panel(log_text, title="Async Log Viewer", border_style="cyan")
-
-    async def run(self):
-        with Live(self._render(), refresh_per_second=10, console=self.console) as live:
-            try:
-                async for logline in self.log_source:
-                    formatted = self._format_line(logline)
-                    self.lines.append(formatted)
-                    live.update(self._render())
-            except asyncio.CancelledError:
-                pass
-
-
-@dataclass
-class Logs:
-    @classmethod
-    @requires_client
-    @syncer.wrap
-    async def tail(cls, action_id: run_definition_pb2.ActionIdentifier, attempt: int = 1) -> AsyncGenerator[str, None]:
-        """
-        Tail the logs for a given action ID and attempt.
-        :param action_id: The action ID to tail logs for.
-        :param attempt: The attempt number (default is 0).
-        """
-        resp = get_client().logs_service.TailLogs(
-            run_logs_service_pb2.TailLogsRequest(action_id=action_id, attempt=attempt)
-        )
-        async for log_set in resp:
-            if log_set.logs:
-                for log in log_set.logs:
-                    for line in log.lines:
-                        yield line
-
-    @classmethod
-    async def create_viewer(cls, action_id: run_definition_pb2.ActionIdentifier, attempt: int = 1, max_lines: int = 30):
-        viewer = AsyncLogViewer(log_source=cls.tail.aio(cls, action_id=action_id, attempt=attempt), max_lines=max_lines)
-        await viewer.run()

union/remote/_project.py

@@ -1,86 +0,0 @@
-from __future__ import annotations
-
-import typing
-from dataclasses import dataclass
-from typing import AsyncGenerator, Literal, Tuple
-
-import rich.repr
-from flyteidl.admin import common_pb2, project_pb2
-
-from union._api_commons import syncer
-from union._initialize import get_client, get_common_config, requires_client
-
-
-@dataclass
-class Project:
-    """
-    A class representing a project in the Union API.
-    """
-
-    _pb2: project_pb2.Project
-
-    @classmethod
-    @requires_client
-    @syncer.wrap
-    async def get(cls, name: str, org: str | None = None) -> Project:
-        """
-        Get a run by its ID or name. If both are provided, the ID will take precedence.
-
-        :param name: The name of the project.
-        :param org: The organization of the project (if applicable).
-        """
-        service = get_client().project_domain_service
-        resp = await service.GetProject(
-            project_pb2.ProjectGetRequest(
-                id=name,
-                org=org,
-            )
-        )
-        return cls(resp)
-
-    @classmethod
-    @requires_client
-    @syncer.wrap
-    async def listall(
-        cls,
-        filters: str | None = None,
-        sort_by: Tuple[str, Literal["asc", "desc"]] | None = None,
-    ) -> typing.Union[typing.Iterator[Project], AsyncGenerator[Project, None]]:
-        """
-        Get a run by its ID or name. If both are provided, the ID will take precedence.
-
-        :param filters: The filters to apply to the project list.
-        :param sort_by: The sorting criteria for the project list, in the format (field, order).
-        :return: An iterator of projects.
-        """
-        token = None
-        sort_by = sort_by or ("created_at", "asc")
-        sort_pb2 = common_pb2.Sort(
-            key=sort_by[0], direction=common_pb2.Sort.ASCENDING if sort_by[1] == "asc" else common_pb2.Sort.DESCENDING
-        )
-        org = get_common_config().org
-        while True:
-            resp = await get_client().project_domain_service.ListProjects(
-                project_pb2.ProjectListRequest(
-                    limit=100,
-                    token=token,
-                    filters=filters,
-                    sort_by=sort_pb2,
-                    org=org,
-                )
-            )
-            token = resp.token
-            for p in resp.projects:
-                yield cls(p)
-            if not token:
-                break
-
-    def __rich_repr__(self) -> rich.repr.Result:
-        yield "name", self._pb2.name
-        yield "id", self._pb2.id
-        yield "description", self._pb2.description
-        yield "state", project_pb2.Project.ProjectState.Name(self._pb2.state)
-        yield (
-            "labels",
-            ", ".join([f"{k}: {v}" for k, v in self._pb2.labels.values.items()]) if self._pb2.labels else None,
-        )