Add support for sqlite.

Author: Diadochokinetic

pandas/io/sql.py

@@ -30,7 +30,6 @@
 import warnings
 
 import numpy as np
-from sqlalchemy.exc import ProgrammingError
 
 from pandas._config import using_string_dtype
 
@@ -975,6 +974,11 @@ def _drop_temporary_table(self):
         self.pd_sql.execute(query)
 
     def _exists_temporary(self):
+        from sqlalchemy.exc import (
+            OperationalError,
+            ProgrammingError,
+        )
+
         if self.schema is None:
             query = f"SELECT * FROM {self.name} LIMIT 1"
         else:
@@ -986,6 +990,8 @@ def _exists_temporary(self):
             # Some DBMS (e.g. postgres) require a rollback after a caught exception
             self.pd_sql.execute("rollback")
             return False
+        except OperationalError:
+            return False
 
     def exists(self):
         if self.is_temporary:
@@ -2816,6 +2822,7 @@ def to_sql(
         chunksize: int | None = None,
         dtype: DtypeArg | None = None,
         method: Literal["multi"] | Callable | None = None,
+        prefixes: Sequence[str] | None = None,
         engine: str = "auto",
         **engine_kwargs,
     ) -> int | None:
@@ -2856,6 +2863,9 @@ def to_sql(
 
             Details and a sample callable implementation can be found in the
             section :ref:`insert method <io.sql.method>`.
+        prefixs : sequence, optional
+            A list of strings to insert after CREATE in the CREATE TABLE statement.
+            They will be separated by spaces.
         """
         if dtype:
             if not is_dict_like(dtype):
@@ -2881,6 +2891,7 @@ def to_sql(
             if_exists=if_exists,
             index_label=index_label,
             dtype=dtype,
+            prefixes=prefixes,
         )
         table.create()
         return table.insert(chunksize, method)

pandas/tests/io/test_sql.py

@@ -17,8 +17,11 @@
 
 import numpy as np
 import pytest
-from sqlalchemy import text
-from sqlalchemy.engine import Connection
+from sqlalchemy import (
+    create_engine,
+    text,
+)
+from sqlalchemy.engine import Engine
 
 from pandas._config import using_string_dtype
 
@@ -4353,7 +4356,7 @@ def test_xsqlite_if_exists(sqlite_buildin):
     drop_table(table_name, sqlite_buildin)
 
 
-@pytest.mark.parametrize("conn", mysql_connectable + postgresql_connectable)
+@pytest.mark.parametrize("conn", sqlalchemy_connectable)
 def test_exists_temporary_table(conn, test_frame1, request):
     conn = request.getfixturevalue(conn)
 
@@ -4372,14 +4375,18 @@ def test_exists_temporary_table(conn, test_frame1, request):
     assert True if table.exists() else False
 
 
-@pytest.mark.parametrize("conn", mysql_connectable + postgresql_connectable)
+@pytest.mark.parametrize("conn", sqlalchemy_connectable)
 def test_to_sql_temporary_table_replace(conn, test_frame1, request):
     conn = request.getfixturevalue(conn)
 
-    if isinstance(conn, Connection):
-        con = conn
-    else:
+    # some DBMS only allow temporary tables to exist within a connection, therefore
+    # we can only test for a connection and not all types of connectables.
+    if isinstance(conn, Engine):
         con = conn.connect()
+    elif isinstance(conn, str):
+        con = create_engine(conn).connect()
+    else:
+        con = conn
 
     test_frame1.to_sql(
         name="test_frame1",
@@ -4402,14 +4409,18 @@ def test_to_sql_temporary_table_replace(conn, test_frame1, request):
     assert_frame_equal(test_frame1, df_test)
 
 
-@pytest.mark.parametrize("conn", mysql_connectable + postgresql_connectable)
+@pytest.mark.parametrize("conn", sqlalchemy_connectable)
 def test_to_sql_temporary_table_fail(conn, test_frame1, request):
     conn = request.getfixturevalue(conn)
 
-    if isinstance(conn, Connection):
-        con = conn
-    else:
+    # some DBMS only allow temporary tables to exist within a connection, therefore
+    # we can only test for a connection and not all types of connectables.
+    if isinstance(conn, Engine):
         con = conn.connect()
+    elif isinstance(conn, str):
+        con = create_engine(conn).connect()
+    else:
+        con = conn
 
     test_frame1.to_sql(
         name="test_frame1",