Adapted for pytask 0.4.1

Author: mj023

README.md

@@ -5,14 +5,11 @@ This is a plugin for Pytask. It is needed to use the Pytask VS Code Extension.
 
 ## Installation
 
-pytask-vscode is available on [PyPI](https://pypi.org/project/pytask-vscode) and [Anaconda.org](https://anaconda.org/conda-forge/pytask-vscode). Install it with
+pytask-vscode is available on [PyPI](https://test.pypi.org/project/pytask-vscode/). Install it with
 
 ```console
-$ pip install pytask-vscode
+$ pip install -i https://test.pypi.org/simple/ pytask-vscode
 
-# or
-
-$ conda install -c conda-forge pytask-vscode
 ```
 
 ## Changes

environment.yml

@@ -11,7 +11,7 @@ dependencies:
   - toml
 
   # Package dependencies
-  - pytask >=0.2
+  - pytask >=0.4.1
   - pytask-parallel >=0.2
   - requests
 

setup.cfg

@@ -29,7 +29,7 @@ project_urls =
 packages = find:
 install_requires =
     click
-    pytask>=0.2
+    pytask>=0.4.1
     requests
 python_requires = >=3.7
 include_package_data = True

src/pytask_vscode/execution.py

@@ -1,12 +1,14 @@
 import pytask
 import json
 import requests
+from contextlib import redirect_stdout
+import io
 
 @pytask.hookimpl(tryfirst=True)
-def pytask_collect_log(session: pytask.Session, reports: list[pytask.CollectionReport], tasks: list[pytask.Task]) -> None:
+def pytask_collect_log(session: pytask.Session, reports: list[pytask.CollectionReport], tasks: list[pytask.PTask]) -> None:
         try:    
             if session.config['command'] == 'collect':
-                result = [{'name' : task.short_name, 'path' : str(task.path)} for task in tasks]
+                result = [{'name' : task.name.split('/')[-1], 'path' : str(task.path)} if isinstance(task,pytask.PTaskWithPath) else {'name' : task.name, 'path' : ''} for task in tasks]
                 res = requests.post('http://localhost:6000/pytask', json={"exitcode" : session.exit_code, "tasks": result}, timeout=0.0001)
         except requests.exceptions.ReadTimeout: 
             pass
@@ -17,8 +19,14 @@ def pytask_collect_log(session: pytask.Session, reports: list[pytask.CollectionR
 @pytask.hookimpl(tryfirst=True)
 def pytask_execute_task_log_end(session: pytask.Session, report: pytask.ExecutionReport) -> None:
 
-    try:    
-        result = {'type': 'task', 'name' : report.task.short_name, 'outcome' : str(report.outcome)}
+    try:
+        if report.outcome == pytask.TaskOutcome.FAIL:
+            with pytask.console.capture() as capture:
+                pytask.console.print(pytask.render_exc_info(report.exc_info[0], report.exc_info[1], report.exc_info[2]))
+            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)}
         res = requests.post('http://localhost:6000/pytask', json=result, timeout=0.00001)
     except requests.exceptions.ReadTimeout: 
         pass