Handle extra error code pystata glitch

Author: hugetim

nbs/14_kernel.ipynb

@@ -174,6 +174,29 @@
     "    }"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#| hide\n",
+    "#| eval: False\n",
+    "from nbstata.config import launch_stata\n",
+    "from nbstata.stata import run_direct"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#| hide\n",
+    "#| eval: False\n",
+    "launch_stata(splash=False)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -183,6 +206,8 @@
     "#| export\n",
     "def print_stata_error(text):\n",
     "    lines = text.splitlines()\n",
+    "    if len(lines) >= 2 and lines[-2] == lines[-1]:\n",
+    "        lines.pop(-1) # remove duplicate error code glitch in pystata.stata.run multi-line (ex. below)\n",
     "    if len(lines) > 2:\n",
     "        print(\"\\n\".join(lines[:-2]))\n",
     "    print_red(\"\\n\".join(lines[-2:]))"
@@ -197,6 +222,68 @@
     "from textwrap import dedent"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      ". disp 1\n",
+      "1\n",
+      "\n",
+      ". disp error\n",
+      "error not found\n",
+      "r(111);\n",
+      "r(111);\n",
+      "\n"
+     ]
+    }
+   ],
+   "source": [
+    "#| hide\n",
+    "#| eval: False\n",
+    "try:\n",
+    "    run_direct(dedent(\"\"\"\\\n",
+    "        disp 1\n",
+    "        disp error\"\"\"), echo=False)\n",
+    "except SystemError as err:\n",
+    "    print(str(err))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      ". disp 1\n",
+      "1\n",
+      "\n",
+      ". disp error\n",
+      "\u001b[31merror not found\n",
+      "r(111);\u001b[0m\n"
+     ]
+    }
+   ],
+   "source": [
+    "#| hide\n",
+    "#| eval: False\n",
+    "try:\n",
+    "    run_direct(dedent(\"\"\"\\\n",
+    "        disp 1\n",
+    "        disp error\"\"\"), echo=False)\n",
+    "except SystemError as err:\n",
+    "    print_stata_error(str(err))"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -220,6 +307,24 @@
     "    \"\"\"))"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\u001b[31mone line\u001b[0m\n"
+     ]
+    }
+   ],
+   "source": [
+    "#| hide\n",
+    "print_stata_error(\"one line\")"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

nbstata/kernel.py

@@ -78,14 +78,16 @@ def _handle_stata_import_error(err, silent, execution_count):
         'execution_count': execution_count,
     }
 
-# %% ../nbs/14_kernel.ipynb 11
+# %% ../nbs/14_kernel.ipynb 13
 def print_stata_error(text):
     lines = text.splitlines()
+    if len(lines) >= 2 and lines[-2] == lines[-1]:
+        lines.pop(-1) # remove duplicate error code glitch in pystata.stata.run multi-line (ex. below)
     if len(lines) > 2:
         print("\n".join(lines[:-2]))
     print_red("\n".join(lines[-2:]))
 
-# %% ../nbs/14_kernel.ipynb 14
+# %% ../nbs/14_kernel.ipynb 19
 def _handle_stata_error(err, silent, execution_count):
     reply_content = {
         "traceback": [],
@@ -105,12 +107,12 @@ def _handle_stata_error(err, silent, execution_count):
     })
     return reply_content
 
-# %% ../nbs/14_kernel.ipynb 15
+# %% ../nbs/14_kernel.ipynb 20
 @patch_to(PyStataKernel)
 def post_do_hook(self):
     self.inspect_output = ""
 
-# %% ../nbs/14_kernel.ipynb 16
+# %% ../nbs/14_kernel.ipynb 21
 @patch_to(PyStataKernel)
 def do_execute(self, code, silent,
                store_history=True, user_expressions=None, allow_stdin=False):
@@ -134,7 +136,7 @@ def do_execute(self, code, silent,
         'user_expressions': {},
     }
 
-# %% ../nbs/14_kernel.ipynb 17
+# %% ../nbs/14_kernel.ipynb 22
 @patch_to(PyStataKernel)
 def do_complete(self, code, cursor_pos):
     """Provide context-aware suggestions"""
@@ -154,13 +156,13 @@ def do_complete(self, code, cursor_pos):
         'matches': matches,
     }
 
-# %% ../nbs/14_kernel.ipynb 18
+# %% ../nbs/14_kernel.ipynb 23
 @patch_to(PyStataKernel)
 def do_is_complete(self, code):
     """Overrides IPythonKernel with kernelbase default"""
     return {"status": "unknown"}
 
-# %% ../nbs/14_kernel.ipynb 19
+# %% ../nbs/14_kernel.ipynb 24
 @patch_to(PyStataKernel)
 def do_inspect(self, code, cursor_pos, detail_level=0, omit_sections=()):
     """Display Stata 'describe' output regardless of cursor position"""
@@ -169,7 +171,7 @@ def do_inspect(self, code, cursor_pos, detail_level=0, omit_sections=()):
     data = {'text/plain': self.inspect_output}
     return {"status": "ok", "data": data, "metadata": {}, "found": True}
 
-# %% ../nbs/14_kernel.ipynb 20
+# %% ../nbs/14_kernel.ipynb 25
 @patch_to(PyStataKernel)
 def do_history(
     self,