update afterware arg parsing

Author: comhar

fasthtml/core.py

@@ -619,8 +619,8 @@ async def _f(req):
         req.body_wrap = body_wrap
         if not resp: resp = await _wrap_call(f, req, sig.parameters)
         for a in self.after:
-            _,*wreq = await _wrap_req(req, _params(a))
-            nr = a(resp, *wreq)
+            wreq = await _wrap_req(req, _params(a))
+            nr = a(*wreq.values())
             if nr: resp = nr
         return _resp(req, resp, sig.return_annotation)
     return _f

nbs/api/00_core.ipynb

@@ -131,7 +131,7 @@
     {
      "data": {
       "text/plain": [
-       "datetime.datetime(2025, 8, 22, 14, 0)"
+       "datetime.datetime(2025, 9, 11, 14, 0)"
       ]
      },
      "execution_count": null,
@@ -1724,8 +1724,8 @@
     "        req.body_wrap = body_wrap\n",
     "        if not resp: resp = await _wrap_call(f, req, sig.parameters)\n",
     "        for a in self.after:\n",
-    "            _,*wreq = await _wrap_req(req, _params(a))\n",
-    "            nr = a(resp, *wreq)\n",
+    "            wreq = await _wrap_req(req, _params(a))\n",
+    "            nr = a(*wreq.values())\n",
     "            if nr: resp = nr\n",
     "        return _resp(req, resp, sig.return_annotation)\n",
     "    return _f"
@@ -2878,13 +2878,13 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Set to 2025-08-22 01:21:53.772537\n"
+      "Set to 2025-09-11 09:53:37.344743\n"
      ]
     },
     {
      "data": {
       "text/plain": [
-       "'Session time: 2025-08-22 01:21:53.772537'"
+       "'Session time: 2025-09-11 09:53:37.344743'"
       ]
      },
      "execution_count": null,
@@ -3055,6 +3055,60 @@
     "test_eq(cli.get('/locked', auth=(\"testuser\",\"spycraft\")).text, 'Hello, testuser')"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "967001f3",
+   "metadata": {},
+   "source": [
+    "Check that afterware like toasts are called correctly."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "09a5fb96",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from fasthtml.toaster import *"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "c7500555",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "app,cli,rt = get_cli(FastHTML())\n",
+    "setup_toasts(app)\n",
+    "\n",
+    "@rt(\"/\")\n",
+    "def index(sess): \n",
+    "    add_toast(sess, \"Successful Page Load!\", \"success\")\n",
+    "    return P(\"I love toast!\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "0ddf5964",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "r = cli.get('/')"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "86386f6d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "assert 'Successful Page Load!' in r.text"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "6a014add",
@@ -3423,7 +3477,7 @@
     {
      "data": {
       "text/plain": [
-       "'Cookie was set at time 01:21:59.994611'"
+       "'Cookie was set at time 09:53:37.455388'"
       ]
      },
      "execution_count": null,