jupyter-duckdb 1.1.3.1__py3-none-any.whl → 1.1.3.4__py3-none-any.whl

duckdb_kernel/kernel.py

@@ -1,3 +1,4 @@
+import csv
 import json
 import math
 import os
@@ -43,6 +44,7 @@ class DuckDBKernel(Kernel):
             MagicCommand('max_rows').arg('count').on(self._max_rows_magic),
             MagicCommand('query_max_rows').arg('count').on(self._query_max_rows_magic),
             MagicCommand('schema').flag('td').opt('only').on(self._schema_magic),
+            MagicCommand('store').arg('file').flag('noheader').result(True).on(self._store_magic),
             MagicCommand('ra').flag('analyze').code(True).on(self._ra_magic),
             MagicCommand('dc').code(True).on(self._dc_magic)
         )
@@ -434,6 +436,26 @@ class DuckDBKernel(Kernel):
 
         self.print_data(svg)
 
+    def _store_magic(self, silent: bool, result_columns: List[str], result: List[List], file: str, noheader: bool):
+        _, ext = file.rsplit('.', 1)
+
+        # csv
+        if ext == 'csv':
+            with open(file, 'w', encoding='utf-8') as f:
+                writer = csv.writer(f)
+
+                if not noheader:
+                    writer.writerow(result_columns)
+
+                for row in result:
+                    writer.writerow(row)
+
+            self.print(f'result stored to {file}', name='stderr')
+
+        # unsupported
+        else:
+            raise ValueError(f'extension {ext} not supported')
+
     def _ra_magic(self, silent: bool, code: str, analyze: bool):
         if self._db is None:
             raise AssertionError('load a database first')

duckdb_kernel/magics/MagicCommandHandler.py

@@ -20,7 +20,7 @@ class MagicCommandHandler:
 
         while True:
             # ensure code starts with '%' or '%%' but not with '%%%'
-            match = re.match(r'^%{1,2}([^% ]+?)($| .+?$)', code, re.MULTILINE | re.IGNORECASE)
+            match = re.match(r'^%{1,2}([^% ]+?)([ \t]*$| .+?$)', code, re.MULTILINE | re.IGNORECASE)
 
             if match is None:
                 break

{jupyter_duckdb-1.1.3.1.dist-info → jupyter_duckdb-1.1.3.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: jupyter-duckdb
-Version: 1.1.3.1
+Version: 1.1.3.4
 Summary: a basic wrapper kernel for DuckDB
 Home-page: https://github.com/erictroebs/jupyter-duckdb
 Author: Eric Tröbs
@@ -12,9 +12,19 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 Requires-Dist: jupyter
-Requires-Dist: graphviz ==0.20.3
+Requires-Dist: graphviz==0.20.3
 Requires-Dist: checkmarkandcross
-Requires-Dist: duckdb ==1.1.3
+Requires-Dist: duckdb==1.1.3
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: project-url
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 # DuckDB Kernel for Jupyter
 

{jupyter_duckdb-1.1.3.1.dist-info → jupyter_duckdb-1.1.3.4.dist-info}/RECORD

@@ -1,7 +1,7 @@
 duckdb_kernel/__init__.py,sha256=6auU6zeJrsA4fxPSr2PYamS8fG-SMXTn5YQFXF2cseo,33
 duckdb_kernel/__main__.py,sha256=Z3GwHEBWoQjNm2Y84ijnbA0Lk66L7nsFREuqhZ_ptk0,165
 duckdb_kernel/kernel.json,sha256=_7E8Ci2FSdCvnzCjsOaue8QE8AvpS5JLQuxORO5IGtA,127
-duckdb_kernel/kernel.py,sha256=wGHvQRhCu6aHvIRPQYJwb9PfRTYju4kYGx5yIeBsPvw,18666
+duckdb_kernel/kernel.py,sha256=wEEH7ZH7lbOvCa8NUe_5yuGT0P45Qg37NPzGT1-BMLc,19396
 duckdb_kernel/db/Column.py,sha256=GM5P6sFdlYK92hiKln5-6038gIDOTxh1AYbR4kiga_w,559
 duckdb_kernel/db/Connection.py,sha256=5pH-CwGh-r9Q2QwJKGSxvoINBU-sqmvZyG4Q1digfeE,599
 duckdb_kernel/db/Constraint.py,sha256=1YgUHk7s8mHCVedbcuJKyXDykj7_ybbwT3Dk9p2VMis,287
@@ -21,7 +21,7 @@ duckdb_kernel/db/implementation/sqlite/__init__.py,sha256=HKogB1es4wOiQUoh7_eT32
 duckdb_kernel/magics/MagicCommand.py,sha256=OoQ6j4cNtIYjaK4MPVzJyv1eYTNu4_a7qoRx-5G3Hg0,2346
 duckdb_kernel/magics/MagicCommandCallback.py,sha256=r1kkJyRR7sZnrnlMH3w4bGqDAJL-BVTIB4-Kn66ynlM,764
 duckdb_kernel/magics/MagicCommandException.py,sha256=MwuWkpA6NoCqz437urdI0RVXhbSbVdziuRoi7slYFPc,49
-duckdb_kernel/magics/MagicCommandHandler.py,sha256=leyzavFL_sY97Ylv1fs9OxTRVfRAGYcwrVeZ5uwNcBQ,2743
+duckdb_kernel/magics/MagicCommandHandler.py,sha256=4njm49cNnLBH9j8GazrmA-wF8XkXfSioJDuCoDyumqc,2749
 duckdb_kernel/magics/__init__.py,sha256=DA8gnQeRCUt1Scy3_NQ9w5CPmMEY9i8YwB-g392pN1U,204
 duckdb_kernel/parser/DCParser.py,sha256=ciNdM_ZFfa10bBlAc_bB2tKmPOxnXBcAGpYROjbAkVY,2120
 duckdb_kernel/parser/LogicParser.py,sha256=PI4NTe4UZIsnEvoAe_LgpEtmGraTTmYOsUk5_Qr3QOk,1137
@@ -76,7 +76,7 @@ duckdb_kernel/visualization/Drawer.py,sha256=D0LkiGMvuJ2v6cQSg_axLTGaM4VXAJEQJAy
 duckdb_kernel/visualization/RATreeDrawer.py,sha256=j-Vy1zpYMzwZ3CsphyfPW-J7ou9a9tM6aXXgAlQTgDI,2128
 duckdb_kernel/visualization/SchemaDrawer.py,sha256=9K-TUUmyeGdMYMTFQJ7evIU3p8p2KyMKeizUc7-y8co,3015
 duckdb_kernel/visualization/__init__.py,sha256=5eMJmxJ01XAXcgWDn3t70eSZF2PGaXdNo6GK-x-0H3s,78
-jupyter_duckdb-1.1.3.1.dist-info/METADATA,sha256=wX-USnHaOptE1CkvS0Sx643zmq7NBzvp1FA69Y9CNVI,7764
-jupyter_duckdb-1.1.3.1.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-jupyter_duckdb-1.1.3.1.dist-info/top_level.txt,sha256=KvRRPMnmkQNuhyBsXoPmwyt26LRDp0O-0HN6u0Dm5jA,14
-jupyter_duckdb-1.1.3.1.dist-info/RECORD,,
+jupyter_duckdb-1.1.3.4.dist-info/METADATA,sha256=A9678InK3tin-SiZ5i9k0LFHtzybk9L-DIg-WUQw01E,7980
+jupyter_duckdb-1.1.3.4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+jupyter_duckdb-1.1.3.4.dist-info/top_level.txt,sha256=KvRRPMnmkQNuhyBsXoPmwyt26LRDp0O-0HN6u0Dm5jA,14
+jupyter_duckdb-1.1.3.4.dist-info/RECORD,,

{jupyter_duckdb-1.1.3.1.dist-info → jupyter_duckdb-1.1.3.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.3.0)
+Generator: setuptools (75.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any