Merge pull request #110 from Salehbigdeli/patch-decorator-with-optional-argument

Patch decorator with optional argument

Author: jph00

fastcore/foundation.py

@@ -39,10 +39,11 @@ def _inner(f):
     return _inner
 
 # Cell
-def patch(f):
+def patch(f=None, *, as_prop=False, cls_method=False):
     "Decorator: add `f` to the first parameter's class (based on f's type annotations)"
+    if f is None: return partial(patch, as_prop=as_prop, cls_method=cls_method)
     cls = next(iter(f.__annotations__.values()))
-    return patch_to(cls)(f)
+    return patch_to(cls, as_prop=as_prop, cls_method=cls_method)(f)
 
 # Cell
 def patch_property(f):

nbs/01_foundation.ipynb

@@ -203,8 +203,7 @@
     "@patch_to(_T5, cls_method=True)\n",
     "def func(cls, x): return cls.attr + x # you can access class attributes in the normal way\n",
     "\n",
-    "t = _T5()\n",
-    "test_eq(t.func(4), 7)"
+    "test_eq(_T5.func(4), 7)"
    ]
   },
   {
@@ -259,10 +258,11 @@
    "outputs": [],
    "source": [
     "#export\n",
-    "def patch(f):\n",
+    "def patch(f=None, *, as_prop=False, cls_method=False):\n",
     "    \"Decorator: add `f` to the first parameter's class (based on f's type annotations)\"\n",
+    "    if f is None: return partial(patch, as_prop=as_prop, cls_method=cls_method)\n",
     "    cls = next(iter(f.__annotations__.values()))\n",
-    "    return patch_to(cls)(f)"
+    "    return patch_to(cls, as_prop=as_prop, cls_method=cls_method)(f)"
    ]
   },
   {
@@ -315,6 +315,40 @@
     "test_eq(t.func2.__qualname__, '_T9.func2')"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Just like `patch_to` decorator you can use `as_propas_prop` and `cls_method` parameters with `patch` decorator:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "@patch(as_prop=True)\n",
+    "def add_ten(self:_T5): return self + 10\n",
+    "\n",
+    "t = _T5(4)\n",
+    "test_eq(t.add_ten, 14)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "class _T5(int): attr = 3 # attr is a class attribute we will access in a later method\n",
+    "    \n",
+    "@patch(cls_method=True)\n",
+    "def func(cls:_T5, x): return cls.attr + x # you can access class attributes in the normal way\n",
+    "\n",
+    "test_eq(_T5.func(4), 7)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,