exec_import

Author: jph00

fastcore/_nbdev.py

@@ -133,6 +133,7 @@
          "add_props": "01_basics.ipynb",
          "typed": "01_basics.ipynb",
          "exec_new": "01_basics.ipynb",
+         "exec_import": "01_basics.ipynb",
          "working_directory": "02_foundation.ipynb",
          "add_docs": "02_foundation.ipynb",
          "docs": "02_foundation.ipynb",

fastcore/basics.py

@@ -14,7 +14,7 @@
            'arg0', 'arg1', 'arg2', 'arg3', 'arg4', 'bind', 'mapt', 'map_ex', 'compose', 'maps', 'partialler',
            'instantiate', 'using_attr', 'Self', 'Self', 'copy_func', 'patch_to', 'patch', 'patch_property',
            'compile_re', 'ImportEnum', 'StrEnum', 'str_enum', 'Stateful', 'PrettyString', 'even_mults', 'num_cpus',
-           'add_props', 'typed', 'exec_new']
+           'add_props', 'typed', 'exec_new', 'exec_import']
 
 # Cell
 from .imports import *
@@ -998,6 +998,13 @@ def _f(*args,**kwargs):
 # Cell
 def exec_new(code):
     "Execute `code` in a new environment and return it"
-    g = {}
+    pkg = None if __name__=='__main__' else Path().cwd().name
+    g = {'__name__': __name__, '__package__': pkg}
     exec(code, g)
-    return g
+    return g
+
+# Cell
+def exec_import(mod, sym):
+    "Import `sym` from `mod` in a new environment"
+    pref = '' if __name__=='__main__' or mod[0]=='.' else '.'
+    return exec_new(f'from {pref}{mod} import {sym}')

nbs/01_basics.ipynb

@@ -5492,7 +5492,8 @@
     "#export\n",
     "def exec_new(code):\n",
     "    \"Execute `code` in a new environment and return it\"\n",
-    "    g = {}\n",
+    "    pkg = None if __name__=='__main__' else Path().cwd().name\n",
+    "    g = {'__name__': __name__, '__package__': pkg}\n",
     "    exec(code, g)\n",
     "    return g"
    ]
@@ -5501,21 +5502,23 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "1"
-      ]
-     },
-     "execution_count": null,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
+   "outputs": [],
    "source": [
     "g = exec_new('a=1')\n",
-    "g['a']"
+    "test_eq(g['a'], 1)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#export\n",
+    "def exec_import(mod, sym):\n",
+    "    \"Import `sym` from `mod` in a new environment\"\n",
+    "    pref = '' if __name__=='__main__' or mod[0]=='.' else '.'\n",
+    "    return exec_new(f'from {pref}{mod} import {sym}')"
    ]
   },
   {
@@ -5693,7 +5696,27 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Converted 00_test.ipynb.\n",
+      "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 06_docments.ipynb.\n",
+      "Converted 07_meta.ipynb.\n",
+      "Converted 08_script.ipynb.\n",
+      "Converted index.ipynb.\n",
+      "Converted parallel_win.ipynb.\n"
+     ]
+    }
+   ],
    "source": [
     "#hide\n",
     "from nbdev.export import notebook2script\n",