rebase

src/sql/magic_plot.py

@@ -76,13 +76,8 @@ def execute(self, line="", cell="", local_ns=None):
         if cmd.args.with_:
             util.show_deprecation_warning()
         if cmd.args.line[0] in {"box", "boxplot"}:
-            with_ = None
-            if util.is_saved_snippet(table):
-                with_ = [table]
-            else:
-                util.is_table_exists(table)
-            if with_:
-                print(f"Plotting using saved snippet : {with_}")
+            with_ = self._check_table_exists(table)
+
             return plot.boxplot(
                 table=table,
                 column=column,
@@ -91,13 +86,7 @@ def execute(self, line="", cell="", local_ns=None):
                 conn=None,
             )
         elif cmd.args.line[0] in {"hist", "histogram"}:
-            with_ = None
-            if util.is_saved_snippet(table):
-                with_ = [table]
-            else:
-                util.is_table_exists(table)
-            if with_:
-                print(f"Plotting using saved snippet : {with_}")
+            with_ = self._check_table_exists(table)
             return plot.histogram(
                 table=table,
                 column=column,
@@ -109,3 +98,12 @@ def execute(self, line="", cell="", local_ns=None):
             raise exceptions.UsageError(
                 f"Unknown plot {cmd.args.line[0]!r}. Must be: 'histogram' or 'boxplot'"
             )
+
+    @staticmethod
+    def _check_table_exists(table):
+        with_ = None
+        if util.is_saved_snippet(table):
+            with_ = [table]
+        else:
+            util.is_table_exists(table)
+        return with_

src/tests/test_command.py

@@ -104,9 +104,7 @@ def test_parsed_sql_when_using_with(ip, sql_magic):
         """,
     )
 
-    cmd = SQLCommand(
-        sql_magic, ip.user_ns, line="", cell="SELECT * FROM author_one"
-    )
+    cmd = SQLCommand(sql_magic, ip.user_ns, line="", cell="SELECT * FROM author_one")
 
     sql = "WITH `author_one` AS (\n\n        SELECT * FROM author LIMIT 1\n        )\n\
 SELECT * FROM author_one"