@@ -203,7 +203,6 @@ def create_and_load_iris_postgresql(conn, iris_file: Path):
203
203
def create_and_load_iris (conn , iris_file : Path , dialect : str ):
204
204
from sqlalchemy import insert
205
205
206
- import_optional_dependency ("adbc_driver_manager.dbapi" , errors = "ignore" )
207
206
iris = iris_table_metadata (dialect )
208
207
209
208
with iris_file .open (newline = None , encoding = "utf-8" ) as csvfile :
@@ -223,7 +222,12 @@ def create_and_load_iris_view(conn):
223
222
cur = conn .cursor ()
224
223
cur .execute (stmt )
225
224
else :
226
- adbc = import_optional_dependency ("adbc_driver_manager.dbapi" , errors = "ignore" )
225
+ if pa_version_under8p0 :
226
+ adbc = None
227
+ else :
228
+ adbc = import_optional_dependency (
229
+ "adbc_driver_manager.dbapi" , errors = "ignore"
230
+ )
227
231
if adbc and isinstance (conn , adbc .Connection ):
228
232
with conn .cursor () as cur :
229
233
cur .execute (stmt )
@@ -301,8 +305,12 @@ def create_and_load_types_sqlite3(conn, types_data: list[dict]):
301
305
302
306
def create_and_load_types_postgresql (conn , types_data : list [dict ]):
303
307
# Boolean support not added until 0.8.0
304
- adbc = import_optional_dependency ("adbc_driver_manager" )
305
- if Version (adbc .__version__ ) < Version ("0.8.0" ):
308
+ if pa_version_under8p0 :
309
+ adbc = None
310
+ else :
311
+ adbc = import_optional_dependency ("adbc_driver_manager" , errors = "ignore" )
312
+
313
+ if adbc and Version (adbc .__version__ ) < Version ("0.8.0" ):
306
314
bool_type = "INTEGER"
307
315
else :
308
316
bool_type = "BOOLEAN"
@@ -363,7 +371,10 @@ def check_iris_frame(frame: DataFrame):
363
371
364
372
def count_rows (conn , table_name : str ):
365
373
stmt = f"SELECT count(*) AS count_1 FROM { table_name } "
366
- adbc = import_optional_dependency ("adbc_driver_manager.dbapi" , errors = "ignore" )
374
+ if pa_version_under8p0 :
375
+ adbc = None
376
+ else :
377
+ adbc = import_optional_dependency ("adbc_driver_manager.dbapi" , errors = "ignore" )
367
378
if isinstance (conn , sqlite3 .Connection ):
368
379
cur = conn .cursor ()
369
380
return cur .execute (stmt ).fetchone ()[0 ]
@@ -495,7 +506,12 @@ def get_all_views(conn):
495
506
c = conn .execute ("SELECT name FROM sqlite_master WHERE type='view'" )
496
507
return [view [0 ] for view in c .fetchall ()]
497
508
else :
498
- adbc = import_optional_dependency ("adbc_driver_manager.dbapi" , errors = "ignore" )
509
+ if pa_version_under8p0 :
510
+ adbc = None
511
+ else :
512
+ adbc = import_optional_dependency (
513
+ "adbc_driver_manager.dbapi" , errors = "ignore"
514
+ )
499
515
if adbc and isinstance (conn , adbc .Connection ):
500
516
results = []
501
517
info = conn .adbc_get_objects ().read_all ().to_pylist ()
@@ -520,7 +536,13 @@ def get_all_tables(conn):
520
536
c = conn .execute ("SELECT name FROM sqlite_master WHERE type='table'" )
521
537
return [table [0 ] for table in c .fetchall ()]
522
538
else :
523
- adbc = import_optional_dependency ("adbc_driver_manager.dbapi" , errors = "ignore" )
539
+ if pa_version_under8p0 :
540
+ adbc = None
541
+ else :
542
+ adbc = import_optional_dependency (
543
+ "adbc_driver_manager.dbapi" , errors = "ignore"
544
+ )
545
+
524
546
if adbc and isinstance (conn , adbc .Connection ):
525
547
results = []
526
548
info = conn .adbc_get_objects ().read_all ().to_pylist ()
@@ -547,7 +569,12 @@ def drop_table(
547
569
conn .commit ()
548
570
549
571
else :
550
- adbc = import_optional_dependency ("adbc_driver_manager.dbapi" , errors = "ignore" )
572
+ if pa_version_under8p0 :
573
+ adbc = None
574
+ else :
575
+ adbc = import_optional_dependency (
576
+ "adbc_driver_manager.dbapi" , errors = "ignore"
577
+ )
551
578
if adbc and isinstance (conn , adbc .Connection ):
552
579
with conn .cursor () as cur :
553
580
cur .execute (f'DROP TABLE IF EXISTS "{ table_name } "' )
@@ -564,7 +591,12 @@ def drop_view(
564
591
conn .execute (f"DROP VIEW IF EXISTS { sql ._get_valid_sqlite_name (view_name )} " )
565
592
conn .commit ()
566
593
else :
567
- adbc = import_optional_dependency ("adbc_driver_manager.dbapi" , errors = "ignore" )
594
+ if pa_version_under8p0 :
595
+ adbc = None
596
+ else :
597
+ adbc = import_optional_dependency (
598
+ "adbc_driver_manager.dbapi" , errors = "ignore"
599
+ )
568
600
if adbc and isinstance (conn , adbc .Connection ):
569
601
with conn .cursor () as cur :
570
602
cur .execute (f'DROP VIEW IF EXISTS "{ view_name } "' )
@@ -1796,7 +1828,7 @@ def test_api_timedelta(conn, request):
1796
1828
)
1797
1829
1798
1830
if "adbc" in conn_name :
1799
- exp_warning = None
1831
+ exp_warning = FutureWarning # pyarrow warns is_sparse is deprecated
1800
1832
else :
1801
1833
exp_warning = UserWarning
1802
1834
0 commit comments