duckdb imprevements

Author: kindly

README.md

@@ -296,10 +296,10 @@ session2 = noteql.Session('postgresql://user:password@localhost/dbname')
 %nql SELECT 1;
 ```
 
-To swap between sessions you can `use` on the session you want:
+To swap between sessions you can `set` on the session you want:
 
 ```python
-session.use()
+session.set()
 ```
 or use the `SESSION` command:
 

noteql/__init__.py

@@ -34,7 +34,7 @@
 def get_engine(dburi, connect_args=None):
     if not dburi:
         dburi = os.environ.get("NOTEQL_DBURI")
-    return sqlalchemy.create_engine(dburi, connect_args=connect_args or {})
+    return sqlalchemy.create_engine(str(dburi), connect_args=connect_args or {})
 
 
 def generate_rows(result, limit):
@@ -203,7 +203,7 @@ def set(self):
             connection = self.datasette_url
         else:
             connection = self.dburi
-            if self.engine.url.password:
+            if self.engine.url.password and connection:
                 connection = connection.replace(self.engine.url.password, '***')
 
         print(
@@ -244,21 +244,39 @@ def create_view(self, view, sql, params=None):
         return self.create_relation(view, sql, "VIEW", params)
 
     def create_relation(self, relation, sql, type, params=None):
-        self.run_sql(
-            """
-            DROP {type} IF EXISTS {relation};
-        """.format(
-                relation=relation, type=type
+        if not (self.database_type == 'duckdb' and relation.startswith("'")):
+            self.run_sql(
+                """
+                DROP {type} IF EXISTS {relation};
+            """.format(
+                    relation=relation, type=type
+                )
             )
-        )
 
-        sql = """
-            CREATE {type} {relation}
-            AS
-            {sql}
-        """.format(
-            relation=relation, sql=sql, type=type
-        )
+            sql = """
+                CREATE {type} {relation}
+                AS
+                {sql}
+            """.format(
+                relation=relation, sql=sql, type=type
+            )
+        else:
+            if relation.endswith(".csv'"):
+                sql = """
+                    COPY (
+                      {sql}
+                    ) TO {relation} with (HEADER 1)
+                """.format(
+                    relation=relation, sql=sql
+                )
+            else:
+                sql = """
+                    COPY (
+                      {sql}
+                    ) TO {relation}
+                """.format(
+                    relation=relation, sql=sql
+                )
         return self.run_sql(sql, params=params)
 
     def get_dataframe(
@@ -583,7 +601,9 @@ def get_parsers(self, noteql_session):
         create = pp.Keyword("create", caseless=True).suppress() + (
             pp.Word(pp.alphanums + "_")
             | pp.QuotedString('"', escQuote='"', unquoteResults=False)
+            | pp.QuotedString("'", escQuote="'", unquoteResults=False)
         )("create")
+
         view = pp.Keyword("view", caseless=True).suppress() + (
             pp.Word(pp.alphanums + "_")
             | pp.QuotedString('"', escQuote='"', unquoteResults=False)
@@ -670,6 +690,8 @@ def execute_part(self, parsed_line, sql):
         ns = self.shell.user_ns
 
         session = self.find_session()
+        if not session:
+            return
 
         actions = {}
         arg_params = {}
@@ -857,14 +879,20 @@ def execute_part(self, parsed_line, sql):
         view_name = actions.get("view")
 
         if view_name:
-            session.create_view(view_name, sql, params)
+            if create_name:
+                session.create_view(view_name, f"select * from {create_name}")
+            else:
+                session.create_view(view_name, sql, params)
 
         return df
 
     @line_cell_magic
     def nql(self, line, cell=None):
 
         session = self.find_session()
+        if not session:
+            return
+
         if cell:
             dfs = []
             magic_line_parser, cell_parser = self.get_parsers(session)