Add more resources.

Author: tobiasraabe

collec.ipynb

@@ -0,0 +1,104 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import inspect"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def fibonacci(n):\n",
+    "    \"\"\"Return the nth fibonacci number.\"\"\"\n",
+    "    if n < 2:\n",
+    "        return n\n",
+    "    return fibonacci(n-1) + fibonacci(n-2)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 12,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "source = inspect.getsource(fibonacci)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 13,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "'Return the nth fibonacci number.'"
+      ]
+     },
+     "execution_count": 13,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "inspect.getdoc(fibonacci)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "def fibonacci(n):\n",
+      "    if n < 2:\n",
+      "        return n\n",
+      "    return fibonacci(n-1) + fibonacci(n-2)\n",
+      "\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(source)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "pytask",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.11.7"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

collect_app.py

@@ -18,12 +18,20 @@
 from textual.widgets import Header
 from textual.widgets import Static
 from textual.widgets import Tree
+from rich._inspect import Inspect
 
 if TYPE_CHECKING:
     from textual.widgets.tree import TreeNode
     from _pytask.node_protocols import PTask
 
 
+class Task(NamedTuple):
+    """A task."""
+
+    task: PTask
+    reasons_rerun: list[str]
+
+
 class Module(NamedTuple):
     """A module with tasks."""
 
@@ -39,7 +47,8 @@ def show_node(self, node: TreeNode) -> None:
             return
         if isinstance(pytask_node, RenderableType):
             self.update(pytask_node)
-        # self.update(Inspect(pytask_node, all=False))
+            return
+        self.update(Inspect(pytask_node, all=False))
 
 
 class CollectionApp(App):

src/_pytask/collect_command.py

@@ -44,6 +44,14 @@
 if TYPE_CHECKING:
     from textual.widgets.tree import TreeNode
     from typing import NoReturn
+    import networkx as nx
+
+
+class Task(NamedTuple):
+    """A task."""
+
+    task: PTask
+    reasons_rerun: list[str]
 
 
 class Module(NamedTuple):
@@ -185,7 +193,7 @@ def _select_tasks_by_expressions_and_marker(session: Session) -> list[PTask]:
     return [task for task in session.tasks if task.signature in remaining]
 
 
-def _organize_tasks(tasks: list[PTaskWithPath]) -> list[Module]:
+def _organize_tasks(tasks: list[PTaskWithPath], dag: nx.DiGraph) -> list[Module]:
     """Organize tasks in a dictionary.
 
     The dictionary has file names as keys and then a dictionary with task names and
@@ -197,11 +205,22 @@ def _organize_tasks(tasks: list[PTaskWithPath]) -> list[Module]:
         module = name_to_module.get(
             task.path.as_posix(), Module(task.path.as_posix(), [])
         )
+        reasons = _find_reasons_to_rerun(task, dag)
+        task_wrap = Task(task=task, reasons_rerun=[])
         module.tasks.append(task)
         name_to_module[module.name] = module
     return [name_to_module[name] for name in sorted(name_to_module)]
 
 
+def _find_reasons_to_rerun(task: PTask, dag: nx.DiGraph) -> list[str]:
+    """Find the reasons to rerun a task."""
+    reasons = []
+    for task in dag.nodes:
+        if node.task == task:
+            reasons.append(node.name)
+    return reasons
+
+
 def _print_collected_tasks(
     dictionary: dict[Path, list[PTaskWithPath]],
     show_nodes: bool,

src/_pytask/nodes.py

@@ -24,6 +24,10 @@
 from rich._inspect import Inspect
 from rich.console import Console, ConsoleOptions, RenderResult, group
 from rich.panel import Panel
+from rich.syntax import Syntax
+from rich.table import Table
+from _pytask.tree_util import tree_leaves
+from rich import box
 
 
 if TYPE_CHECKING:
@@ -149,12 +153,15 @@ def execute(self, **kwargs: Any) -> Any:
         return self.function(**kwargs)
 
     def __rich_console__(self, console: Console, options: ConsoleOptions) -> RenderResult:
-        @group()
-        def _():
-            yield self.name
-            yield self.function.__doc__ or ""
-
-        yield Panel(_())
+        table = Table("", "", title="Information", show_header=False, expand=True, box=box.ROUNDED)
+        table.add_row("Name", self.name)
+        table.add_row("Path", self.path.as_posix())
+        table.add_row("# Dependecies", str(len(tree_leaves(self.depends_on))))
+        table.add_row("# Products", str(len(tree_leaves(self.produces))))
+
+        yield table
+        yield Panel(inspect.getdoc(self.function), title="Docstring")
+        yield Panel(Syntax(inspect.getsource(self.function), lexer="python"), title="Source Code")
 
 
 @define(kw_only=True)

tasks/out.txt

@@ -0,0 +1 @@
+Hello World!

tasks/out_2.txt

@@ -0,0 +1 @@
+Hello World!

tasks/task_example.py

@@ -0,0 +1,16 @@
+from pathlib import Path
+from typing import Annotated
+from pytask import Product
+
+
+
+def task_example(path: Path = Path("out.txt")) -> Annotated[Path, Path("out_2.txt")]:
+    """Compute the square of 2.
+
+    Example
+    -------
+    >>> task_example(2)
+    4
+
+    """
+    return path.read_text()

tasks/task_module.py

@@ -0,0 +1,7 @@
+from pathlib import Path
+from typing import Annotated
+from pytask import Product
+
+
+def task_example(path: Annotated[Path, Product] = Path("out.txt")) -> None:
+    path.write_text("Hello World!")