Merge pull request #243 from fastai/obj2dict

add obj2dict

Author: Hamel Husain

fastcore/_nbdev.py

@@ -134,6 +134,7 @@
          "read_config_file": "02_foundation.ipynb",
          "Config": "02_foundation.ipynb",
          "dict2obj": "03_xtras.ipynb",
+         "obj2dict": "03_xtras.ipynb",
          "repr_dict": "03_xtras.ipynb",
          "AttrDict.__repr__": "03_xtras.ipynb",
          "is_listy": "03_xtras.ipynb",

fastcore/xtras.py

@@ -1,9 +1,10 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: nbs/03_xtras.ipynb (unless otherwise specified).
 
-__all__ = ['dict2obj', 'repr_dict', 'is_listy', 'shufflish', 'mapped', 'IterLen', 'ReindexCollection', 'maybe_open',
-           'image_size', 'bunzip', 'join_path_file', 'loads', 'untar_dir', 'repo_details', 'run', 'open_file',
-           'save_pickle', 'load_pickle', 'stringfmt_names', 'PartialFormatter', 'partial_format', 'utc2local',
-           'local2utc', 'trace', 'round_multiple', 'modified_env', 'ContextManagers', 'str2bool', 'sort_by_run']
+__all__ = ['dict2obj', 'obj2dict', 'repr_dict', 'is_listy', 'shufflish', 'mapped', 'IterLen', 'ReindexCollection',
+           'maybe_open', 'image_size', 'bunzip', 'join_path_file', 'loads', 'untar_dir', 'repo_details', 'run',
+           'open_file', 'save_pickle', 'load_pickle', 'stringfmt_names', 'PartialFormatter', 'partial_format',
+           'utc2local', 'local2utc', 'trace', 'round_multiple', 'modified_env', 'ContextManagers', 'str2bool',
+           'sort_by_run']
 
 # Cell
 from .imports import *
@@ -24,6 +25,13 @@ def dict2obj(d):
     if not isinstance(d, dict): return d
     return AttrDict(**{k:dict2obj(v) for k,v in d.items()})
 
+# Cell
+def obj2dict(d):
+    "Convert (possibly nested) AttrDicts (or lists of AttrDicts) to `dict`"
+    if isinstance(d, (L,list)): return list(L(d).map(obj2dict))
+    if not isinstance(d, dict): return d
+    return dict(**{k:obj2dict(v) for k,v in d.items()})
+
 # Cell
 def _repr_dict(d, lvl):
     if isinstance(d,dict):

nbs/02_foundation.ipynb

@@ -685,7 +685,7 @@
     {
      "data": {
       "text/plain": [
-       "b'\\x80\\x03cinspect\\nSignature\\nq\\x00(cinspect\\nParameter\\nq\\x01X\\x05\\x00\\x00\\x00itemsq\\x02cinspect\\n_ParameterKind\\nq\\x03K\\x01\\x85q\\x04Rq\\x05\\x86q\\x06Rq\\x07}q\\x08(X\\x08\\x00\\x00\\x00_defaultq\\tNX\\x0b\\x00\\x00\\x00_annotationq\\ncinspect\\n_empty\\nq\\x0bubh\\x01X\\x04\\x00\\x00\\x00restq\\x0ch\\x03K\\x02\\x85q\\rRq\\x0e\\x86q\\x0fRq\\x10}q\\x11(h\\th\\x0bh\\nh\\x0bubh\\x01X\\x08\\x00\\x00\\x00use_listq\\x12h\\x03K\\x03\\x85q\\x13Rq\\x14\\x86q\\x15Rq\\x16}q\\x17(h\\t\\x89h\\nh\\x0bubh\\x01X\\x05\\x00\\x00\\x00matchq\\x18h\\x14\\x86q\\x19Rq\\x1a}q\\x1b(h\\tNh\\nh\\x0bubtq\\x1c\\x85q\\x1dRq\\x1e}q\\x1fX\\x12\\x00\\x00\\x00_return_annotationq h\\x0bsb.'"
+       "b'\\x80\\x04\\x95\\x01\\x01\\x00\\x00\\x00\\x00\\x00\\x00\\x8c\\x07inspect\\x94\\x8c\\tSignature\\x94\\x93\\x94(h\\x00\\x8c\\tParameter\\x94\\x93\\x94\\x8c\\x05items\\x94h\\x00\\x8c\\x0e_ParameterKind\\x94\\x93\\x94K\\x01\\x85\\x94R\\x94\\x86\\x94R\\x94}\\x94(\\x8c\\x08_default\\x94N\\x8c\\x0b_annotation\\x94h\\x00\\x8c\\x06_empty\\x94\\x93\\x94ubh\\x04\\x8c\\x04rest\\x94h\\x07K\\x02\\x85\\x94R\\x94\\x86\\x94R\\x94}\\x94(h\\rh\\x10h\\x0eh\\x10ubh\\x04\\x8c\\x08use_list\\x94h\\x07K\\x03\\x85\\x94R\\x94\\x86\\x94R\\x94}\\x94(h\\r\\x89h\\x0eh\\x10ubh\\x04\\x8c\\x05match\\x94h\\x19\\x86\\x94R\\x94}\\x94(h\\rNh\\x0eh\\x10ubt\\x94\\x85\\x94R\\x94}\\x94\\x8c\\x12_return_annotation\\x94h\\x10sb.'"
       ]
      },
      "execution_count": null,
@@ -818,7 +818,7 @@
     {
      "data": {
       "text/plain": [
-       "['k', 1, 0]"
+       "[10, 3, 7]"
       ]
      },
      "execution_count": null,
@@ -860,6 +860,8 @@
     "t = L(arr, use_list=None)\n",
     "test_eq(t[1,2], arr[[1,2]])\n",
     "\n",
+    "\n",
+    "import pandas as pd\n",
     "df = pd.DataFrame({'a':[1,2,3]})\n",
     "t = L(df, use_list=None)\n",
     "test_eq(t[1,2], L(pd.DataFrame({'a':[2,3]}, index=[1,2]), use_list=None))"
@@ -1983,10 +1985,13 @@
       "Converted 01_basics.ipynb.\n",
       "Converted 02_foundation.ipynb.\n",
       "Converted 03_xtras.ipynb.\n",
+      "Converted 03a_parallel.ipynb.\n",
+      "Converted 03b_net.ipynb.\n",
       "Converted 04_dispatch.ipynb.\n",
       "Converted 05_transform.ipynb.\n",
       "Converted 07_meta.ipynb.\n",
       "Converted 08_script.ipynb.\n",
+      "Converted Untitled1.ipynb.\n",
       "Converted index.ipynb.\n"
      ]
     }

nbs/03_xtras.ipynb

@@ -101,10 +101,42 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "ds = dict2obj(L(d1, d1))\n",
+    "_list_of_dicts = [d1, d1]\n",
+    "ds = dict2obj(_list_of_dicts)\n",
     "test_eq(ds[0].b.c, 2)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#export\n",
+    "def obj2dict(d):\n",
+    "    \"Convert (possibly nested) AttrDicts (or lists of AttrDicts) to `dict`\"\n",
+    "    if isinstance(d, (L,list)): return list(L(d).map(obj2dict))\n",
+    "    if not isinstance(d, dict): return d\n",
+    "    return dict(**{k:obj2dict(v) for k,v in d.items()})"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "`obj2dict` can be used to reverse what is done by `dict2obj`:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "test_eq(obj2dict(d2), d1)\n",
+    "test_eq(obj2dict(ds), _list_of_dicts) "
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,