VSCode Logging

.pre-commit-config.yaml

@@ -64,6 +64,7 @@ repos:
             click,
             optree,
             pluggy,
+            types-requests,
             rich,
             sqlalchemy,
             types-setuptools,

environment.yml

@@ -17,6 +17,7 @@ dependencies:
   - networkx >=2.4
   - pluggy >=1.0.0
   - optree >=0.9
+  - requests
   - rich
   - sqlalchemy >=1.4.36
   - tomli >=1.0.0

pyproject.toml

@@ -40,6 +40,7 @@ dependencies = [
   "packaging",
   "pluggy>=1",
   "rich",
+  "requests",
   "sqlalchemy>=1.4.36",
   'tomli>=1; python_version < "3.11"',
   'typing-extensions; python_version < "3.9"',

src/_pytask/collect.py

@@ -1,17 +1,20 @@
 """Implement functionality to collect tasks."""
 from __future__ import annotations
 
+import contextlib
 import inspect
 import itertools
 import os
 import sys
 import time
 from pathlib import Path
+from threading import Thread
 from typing import Any
 from typing import Generator
 from typing import Iterable
 from typing import TYPE_CHECKING
 
+import requests
 from _pytask.collect_utils import create_name_of_python_node
 from _pytask.collect_utils import parse_dependencies_from_task_function
 from _pytask.collect_utils import parse_products_from_task_function
@@ -29,6 +32,7 @@
 from _pytask.node_protocols import PPathNode
 from _pytask.node_protocols import PTask
 from _pytask.nodes import PathNode
+from _pytask.nodes import PTaskWithPath
 from _pytask.nodes import PythonNode
 from _pytask.nodes import Task
 from _pytask.nodes import TaskWithoutPath
@@ -469,13 +473,40 @@
     return id_to_short_id
 
 
+def send_logging_vscode(url: str, json: dict[str, Any], timeout: float) -> None:
+    """Send logging information to VSCode."""
+    with contextlib.suppress(requests.exceptions.RequestException):
+        requests.post(url=url, json=json, timeout=timeout)
+
+
 @hookimpl
 def pytask_collect_log(
     session: Session, reports: list[CollectionReport], tasks: list[PTask]
 ) -> None:
     """Log collection."""
     session.collection_end = time.time()
 
+    if session.config["command"] == "collect":
+        exitcode = 0
+        for report in reports:
+            if report.outcome == CollectionOutcome.FAIL:
+                exitcode = 3
+        result = [
+            {"name": task.name.split("/")[-1], "path": str(task.path)}
+            if isinstance(task, PTaskWithPath)
+            else {"name": task.name, "path": ""}
+            for task in tasks
+        ]
+        thread = Thread(
+            target=send_logging_vscode,
+            args=(
+                "http://localhost:6000/pytask",
+                {"exitcode": exitcode, "tasks": result},
+                0.00001,
+            ),
+        )
+        thread.start()
+
     console.print(f"Collected {len(tasks)} task{'' if len(tasks) == 1 else 's'}.")
 
     failed_reports = [r for r in reports if r.outcome == CollectionOutcome.FAIL]

src/_pytask/live.py

@@ -1,17 +1,20 @@
 """Contains code related to live objects."""
 from __future__ import annotations
 
+from threading import Thread
 from typing import Any
 from typing import Generator
 from typing import NamedTuple
 from typing import TYPE_CHECKING
 
 import click
+from _pytask.collect import send_logging_vscode
 from _pytask.config import hookimpl
 from _pytask.console import console
 from _pytask.console import format_task_name
 from _pytask.outcomes import CollectionOutcome
 from _pytask.outcomes import TaskOutcome
+from _pytask.traceback import Traceback
 from attrs import define
 from attrs import field
 from rich.box import ROUNDED
@@ -168,6 +171,27 @@ def pytask_execute_task_log_start(self, task: PTask) -> bool:
     @hookimpl
     def pytask_execute_task_log_end(self, report: ExecutionReport) -> bool:
         """Mark a task as being finished and update outcome."""
+        if report.outcome == TaskOutcome.FAIL and report.exc_info is not None:
+            with console.capture() as capture:
+                console.print(Traceback(report.exc_info))
+            s = capture.get()
+            result = {
+                "type": "task",
+                "name": report.task.name.split("/")[-1],
+                "outcome": str(report.outcome),
+                "exc_info": s,
+            }
+        else:
+            result = {
+                "type": "task",
+                "name": report.task.name.split("/")[-1],
+                "outcome": str(report.outcome),
+            }
+        thread = Thread(
+            target=send_logging_vscode,
+            args=("http://localhost:6000/pytask", result, 0.00001),
+        )
+        thread.start()
         self.update_reports(report)
         return True