jupyter-duckdb 1.2.101__py3-none-any.whl → 1.2.103__py3-none-any.whl

duckdb_kernel/db/Connection.py

@@ -10,6 +10,9 @@ class Connection:
     def close(self):
         pass
 
+    def copy(self) -> 'Connection':
+        raise NotImplementedError
+
     @staticmethod
     def plain_explain() -> bool:
         return False

duckdb_kernel/db/implementation/duckdb/Connection.py

@@ -15,6 +15,9 @@ class Connection(Base):
     def close(self):
         self.con.close()
 
+    def copy(self) -> 'Connection':
+        return Connection(self.path)
+
     @staticmethod
     def plain_explain() -> bool:
         return True

duckdb_kernel/db/implementation/postgres/Connection.py

@@ -19,6 +19,7 @@ class Connection(Base):
         self.host: str = host
         self.port: int = port
         self.username: Optional[str] = username
+        self.password: Optional[str] = password
 
         options = {
             'host': host,
@@ -43,6 +44,9 @@ class Connection(Base):
     def close(self):
         self.con.close()
 
+    def copy(self) -> 'Connection':
+        return Connection(self.host, self.port, self.username, self.password, self.database_name)
+
     def __str__(self) -> str:
         user = f'{self.username}@' if self.username is not None else ''
         return f'PostgreSQL: {user}{self.host}:{self.port}/{self.database_name}'

duckdb_kernel/db/implementation/sqlite/Connection.py

@@ -16,6 +16,9 @@ class Connection(Base):
     def close(self):
         self.con.close()
 
+    def copy(self) -> 'Connection':
+        return Connection(self.path)
+
     @staticmethod
     def multiple_statements_per_query() -> bool:
         return False