sqlspec 0.14.0__py3-none-any.whl → 0.15.0__py3-none-any.whl

{sqlspec-0.14.0.dist-info → sqlspec-0.15.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlspec
-Version: 0.14.0
+Version: 0.15.0
 Summary: SQL Experiments in Python
 Project-URL: Discord, https://discord.gg/litestar
 Project-URL: Issue, https://github.com/litestar-org/sqlspec/issues/
@@ -11,8 +11,9 @@ License-Expression: MIT
 License-File: LICENSE
 License-File: NOTICE
 Requires-Python: <4.0,>=3.9
-Requires-Dist: click
 Requires-Dist: eval-type-backport; python_version < '3.10'
+Requires-Dist: mypy-extensions
+Requires-Dist: rich-click
 Requires-Dist: sqlglot>=19.9.0
 Requires-Dist: typing-extensions
 Provides-Extra: adbc
@@ -28,6 +29,9 @@ Provides-Extra: asyncmy
 Requires-Dist: asyncmy; extra == 'asyncmy'
 Provides-Extra: asyncpg
 Requires-Dist: asyncpg; extra == 'asyncpg'
+Provides-Extra: attrs
+Requires-Dist: attrs; extra == 'attrs'
+Requires-Dist: cattrs; extra == 'attrs'
 Provides-Extra: bigquery
 Requires-Dist: google-cloud-bigquery; extra == 'bigquery'
 Provides-Extra: cli
@@ -44,6 +48,7 @@ Provides-Extra: litestar
 Requires-Dist: litestar; extra == 'litestar'
 Provides-Extra: msgspec
 Requires-Dist: msgspec; extra == 'msgspec'
+Provides-Extra: mypyc
 Provides-Extra: nanoid
 Requires-Dist: fastnanoid>=0.4.1; extra == 'nanoid'
 Provides-Extra: obstore

sqlspec-0.15.0.dist-info/RECORD

@@ -0,0 +1,134 @@
+sqlspec/__init__.py,sha256=8_TR9bXd7bkA4qBGCzTHNawH7KXaJ4YlnCJzJwvgud8,2066
+sqlspec/__main__.py,sha256=lXBKZMOXA1uY735Rnsb-GS7aXy0nt22tYmd2X9FcxrY,253
+sqlspec/__metadata__.py,sha256=IUw6MCTy1oeUJ1jAVYbuJLkOWbiAWorZ5W-E-SAD9N4,395
+sqlspec/_serialization.py,sha256=6U5-smk2h2yl0i6am2prtOLJTdu4NJQdcLlSfSUMaUQ,2590
+sqlspec/_sql.py,sha256=GggpMLOqVQxuirCcpSj-wBQX2N89jOqNibP7r88-dMY,39567
+sqlspec/_typing.py,sha256=jv-7QHGLrJLfnP86bR-Xcmj3PDoddNZEKDz_vYRBiAU,22684
+sqlspec/base.py,sha256=lVLzFD-nzEU6QnwnU0kRWh3XWjbvXWX0XnnUViYBoQk,21767
+sqlspec/cli.py,sha256=3ZxPwl4neNWyrAkM9J9ccC_gaFigDJbhuZfx15JVE7E,9903
+sqlspec/config.py,sha256=s7csxGK0SlTvB9jOvHlKKm4Y272RInQrUd6hGXwy31Q,14974
+sqlspec/exceptions.py,sha256=mCqNJ0JSPA-TUPpAfdctwwqJWbiNsWap5ATNNRdczwU,6159
+sqlspec/loader.py,sha256=sIK4I8L1Qe6hyoi6OHuaaAUUTdj4UrsRca16fuMs8HM,27259
+sqlspec/protocols.py,sha256=nWj5DwWGBkY3i5KV5MsXSSALBbbR8vV_i17YoWNpW4U,12632
+sqlspec/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sqlspec/typing.py,sha256=JEq2VsTaLdNZ1HAqj3n_HgQgdwlQQOCcZx3VBiPs0UA,7393
+sqlspec/adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sqlspec/adapters/adbc/__init__.py,sha256=CqBaEahvHbQ5NxmBJuuCtZ8v5wuaPgbqSSe9WUhglWY,336
+sqlspec/adapters/adbc/_types.py,sha256=htTZ20qVo6Iod_ooUTLGd0xAL8AUYA4qeOexTaLQG3Q,351
+sqlspec/adapters/adbc/config.py,sha256=UdS_4q5tsXKPEilD2dLtk9H1BI02CTSvhIWEDChrPEs,13871
+sqlspec/adapters/adbc/driver.py,sha256=nYS6e6DXxZ8lHzUFvXvC8qtcGSBTvK8IF_NlBbw56pg,20217
+sqlspec/adapters/aiosqlite/__init__.py,sha256=q2YXU9WCL2RSYFvd0DiXz5PeCaezCRmrqmQVd86V2so,577
+sqlspec/adapters/aiosqlite/_types.py,sha256=M8VqaW6iCigfExhdJM2yPAN09Ue2yBoBL59-QXGBObo,355
+sqlspec/adapters/aiosqlite/config.py,sha256=F_trn-H2KcB1sH1fXdGKYNLcDFwu28WszTBoESK8D0c,9036
+sqlspec/adapters/aiosqlite/driver.py,sha256=pU6uyvZSDrUsgkLkBW1guq6wno5FOyqt9Tg8dg6L7HM,9644
+sqlspec/adapters/asyncmy/__init__.py,sha256=wBgak4MA3ySaGQHUxrnv3HdSVQKKRWf1_DmENL0LkWs,531
+sqlspec/adapters/asyncmy/_types.py,sha256=WbwREzJkLYmqd4c7A7cu04WaD5g7-n35ZETHZvbP1Z8,250
+sqlspec/adapters/asyncmy/config.py,sha256=8fGiiFjIaT_y6OcVj_7dTK9sSuLqucMRsGcRPoYigy8,6557
+sqlspec/adapters/asyncmy/driver.py,sha256=W_RLuGrGlqLl5aZBfMu4ZxBcgKZDPOmoJuAD-wpZeKM,11215
+sqlspec/adapters/asyncpg/__init__.py,sha256=h3lg3pLBLrS9ABBt5yKFKhGL69RpOlASYSqVCuAuURY,567
+sqlspec/adapters/asyncpg/_types.py,sha256=p65WrqfUl2cx5Md759_aV8m_gHP-PyCzNk48bLhDQkc,425
+sqlspec/adapters/asyncpg/config.py,sha256=KdJ2BEtkIA3X3BY4C86zCOUH1-U3AT8ZPNYOpbN0e3A,8752
+sqlspec/adapters/asyncpg/driver.py,sha256=jKDnh5Cz-LTCQvSGaYPgDysl7aocVZ9Wfxy_ERz0RUw,14286
+sqlspec/adapters/bigquery/__init__.py,sha256=1nzkWubO4ZObuK27QoeGfS05ed5v02NU4GaXrvqjlw0,504
+sqlspec/adapters/bigquery/_types.py,sha256=gECmbiSmZmnR-xJpmznhnva4SbhhdcYQ09jBdqMAmOs,255
+sqlspec/adapters/bigquery/config.py,sha256=KYVvhcd4MyU2ak0t1mX3bd3EXlVMEciO7YPaDtZsPc8,11975
+sqlspec/adapters/bigquery/driver.py,sha256=3y8gwrYTt6ktPKE0ihGk5Dj7yvL52Gc8uT1b1ynaCuA,24016
+sqlspec/adapters/duckdb/__init__.py,sha256=iovQlYGfizYpIZpbc1YymK2xt6pZ4ApQj6I0Q0lHc3I,603
+sqlspec/adapters/duckdb/_types.py,sha256=4p5nuD1yNs0fmQ5sGxKFx2ru6jUCPrpsYgqfh-uZHM8,270
+sqlspec/adapters/duckdb/config.py,sha256=krVy5BsS7f2Uv4vmZrwpxlXzbe-MweS6JbHm809NJqE,18610
+sqlspec/adapters/duckdb/driver.py,sha256=HDRPYh1k0JnnnaPpyv8oONn7UAYDCQyTK4fsgFDv6uk,15003
+sqlspec/adapters/oracledb/__init__.py,sha256=AUsZ8m9tDgNwxv9m4LtkIs9Ib6hOtBrGNm-Ee4HWNq0,843
+sqlspec/adapters/oracledb/_types.py,sha256=yPJTQE37HYVgvxI3USonWvFJ2VcfaNnXqfgKNVphmcQ,400
+sqlspec/adapters/oracledb/config.py,sha256=yZITXSWuHKwrjVcuaTLyR9RtVb3WtzmWlBQTjvyUtPo,11383
+sqlspec/adapters/oracledb/driver.py,sha256=mckrpid-XI-3juWbTXfy0JuYuZl3NS-ZB9P2VcJMSTI,22534
+sqlspec/adapters/psqlpy/__init__.py,sha256=ABve2Oj-G-fxMO8WRJ0XzxEw2cs5H3INDnmUI99l8gc,526
+sqlspec/adapters/psqlpy/_types.py,sha256=tG4jwQtBB6mCX5KU5x3hAhxsQUQlQEzbCsYhU3IouVc,269
+sqlspec/adapters/psqlpy/config.py,sha256=h9IZr0vv6-rS7RLHSVsaIdCGgLsTdXQJSSgw4HI99d4,7866
+sqlspec/adapters/psqlpy/driver.py,sha256=e9gHrmjXj7QGE_DxCTA3DhjvckSuFGO7pcdm3kqNv1c,20346
+sqlspec/adapters/psycopg/__init__.py,sha256=swmTz8xlj6LvB-i78tBSE_T-sez2e6lFJW2cMzNJEVE,862
+sqlspec/adapters/psycopg/_types.py,sha256=UJBKDWgARoSR6UxSy8zcIP7HSHTpXU5moJTsjDj0M4M,563
+sqlspec/adapters/psycopg/config.py,sha256=tyxP1WbLQlwvdWdykG_iluIaQg0O6GLUwMLtfaRQ3sE,16268
+sqlspec/adapters/psycopg/driver.py,sha256=qMRaoUI1RF0JLGEEIWzZJTyPd5jEaRFq6ZY0dkukrCQ,33434
+sqlspec/adapters/sqlite/__init__.py,sha256=wDN0StuJxCy0VX1fVYDngTFUP5Cx8d0-foeJeJaEVOw,484
+sqlspec/adapters/sqlite/_types.py,sha256=4Nqolhk8s4mwLw13BwUjuyhAbM9BsKwJCvcmjMWkhaY,246
+sqlspec/adapters/sqlite/config.py,sha256=jTomeuHahwMN9huOnQclAuJyiFL1m8hoexQxWCn0Ckg,9166
+sqlspec/adapters/sqlite/driver.py,sha256=uAhasoCNOV30gTvl1EUpofRcm8YiEW5RnVy07XyypzI,12103
+sqlspec/builder/__init__.py,sha256=E3UGimdSOUK_ZGxFEOsCduwWWwTxcqoLe3Rvh4KbGNY,1429
+sqlspec/builder/_base.py,sha256=vApAFc9-Vw4DsiQP7iQhtfIEjGy10mvi-6DjJnulZiM,17234
+sqlspec/builder/_column.py,sha256=sGDoRhgeqNv69M2jIcNDWxTsMV3TKRNtktpDwl_ssSI,13224
+sqlspec/builder/_ddl.py,sha256=6TvCMiJAuPBtwXbAupGKnrJ9Wb9-VK5P6cfEFbE9CT8,50284
+sqlspec/builder/_ddl_utils.py,sha256=qlKT9TqlaymwZLiZjHXIVvFofiWekCLNu-AvPFzUiac,4115
+sqlspec/builder/_delete.py,sha256=YCAV7GUutY-j1Z6tQ9Q0CE4K4Z5LbZXldMOpS2uxAi4,2301
+sqlspec/builder/_insert.py,sha256=nsAaJpAWU0g8LnAJynScn5Iqiw8f-bxvKlT-g5yToxw,8489
+sqlspec/builder/_merge.py,sha256=pabh0G7DsMZoOG8cGep0qLLdgnyoxQFRUUglhE5yScE,1570
+sqlspec/builder/_parsing_utils.py,sha256=0Bu2G5VAWOym-tIpdFau3_sa71CMmM-BwZzgDoD89xE,5384
+sqlspec/builder/_select.py,sha256=oxi-5ezjsAt7n6OQm1FNCSO_wFo8djWBPL9f-kiUo9g,5992
+sqlspec/builder/_update.py,sha256=l9tHwC96VSmk8zGMhzB_S4YuD2_6q4cBXMboqoqvLJ0,5969
+sqlspec/builder/mixins/__init__.py,sha256=YXhAzKmQbQtne5j26SKWY8PUxwosl0RhlhLoahAdkj0,1885
+sqlspec/builder/mixins/_cte_and_set_ops.py,sha256=BPPkDTFo2Feja7Vers0BrC_JIYcpfOYSTZ9GIV_UOZ4,8317
+sqlspec/builder/mixins/_delete_operations.py,sha256=0f8ZZMFx5b53EUx1oF264owLb6aWP45naeB6g-BZQys,1068
+sqlspec/builder/mixins/_insert_operations.py,sha256=0Pz0QSa4bgGIiKT2Y8zPs6C2uxEQPyQqXdhKADttSJ0,5528
+sqlspec/builder/mixins/_join_operations.py,sha256=5qloj7JmRWBi7pFBpAY1acwg_O_1CpcHeYMbzkBKfBE,5507
+sqlspec/builder/mixins/_merge_operations.py,sha256=JpP33NevlhfCHnKMpLQOU8THN2ImFKMUwdWiI6UydPI,16745
+sqlspec/builder/mixins/_order_limit_operations.py,sha256=NuNUb_DJKTdnbjasjOh27zLtbr1JZ4ygS0NH76G4foI,4514
+sqlspec/builder/mixins/_pivot_operations.py,sha256=nkV9kG15rSoX0IWRAmaZjxyY8J7_5sScgxrRtYjddqE,5886
+sqlspec/builder/mixins/_select_operations.py,sha256=-zo75-lp9T3j7D7zRpsa1D3oCd0Jt2xZ_OslhZeumt0,24692
+sqlspec/builder/mixins/_update_operations.py,sha256=N6Lm-BrMhEPfFQz9V3-4ZgvSEsGfWtuQwlWbzdgLVWc,6904
+sqlspec/builder/mixins/_where_clause.py,sha256=lb6iGqbKPX7rInY1YwYpVMQxUlRftRZthZdwDbnYs20,25729
+sqlspec/core/__init__.py,sha256=rU_xGsXhqIOnBbyB2InhJknYePm5NQ2DSWdBigror4g,1775
+sqlspec/core/cache.py,sha256=CQ_xTBQVJaBxB5-ipiw4Oj6JObrcMFtXYRLMSvbXGnE,27130
+sqlspec/core/compiler.py,sha256=we_ifDr7JOAu3EycmWDxynOYq8ufjwcZl9MDXFTf5cQ,13959
+sqlspec/core/filters.py,sha256=J2tpDo8Ton98Gwc9Tx0Ki3X0JgO6xrnbE0_stfBeP3E,31126
+sqlspec/core/hashing.py,sha256=4KyAFWtFDMYreoBGGPQppEuMWO6_NrRYlw9Lm-qeJqo,10429
+sqlspec/core/parameters.py,sha256=X5e9SKGS1gkn7Sv6qAkzZtllIQDmhuUxAnLFlxBkAns,52833
+sqlspec/core/result.py,sha256=vdPXLApkS0F2Jz-Bq7-A-bKf7g4JIs5_57nAI4O4tC0,21100
+sqlspec/core/splitter.py,sha256=cb2P1B0q5vvALHi3SEJ7VdbRHH2GWsftrmJi9PYMbeE,28089
+sqlspec/core/statement.py,sha256=S_fkqYkG9v2KCsznCBAJIDkR2ASKwub5Y3dNlRYkLec,24231
+sqlspec/driver/__init__.py,sha256=QVpDRQGd1GreIP199en6qDbq-cZJcEF5go68DINagUk,569
+sqlspec/driver/_async.py,sha256=0P3VlBYXfythkh9c1wCE3oZMwQa0NtbW_6SljPxpcAw,18858
+sqlspec/driver/_common.py,sha256=go-SZRYlx_qEh8A9BGFl_BVByEdnLvtSXi6bqA6wOmU,25553
+sqlspec/driver/_sync.py,sha256=UZKCTrKh48lzRONk-eX5q6qKBLZOawODaPN_28ObACE,18586
+sqlspec/driver/mixins/__init__.py,sha256=gN4pQyJXxNy0xi91dcMJGA7DQ7TbjGjQI24SSpZc6Go,248
+sqlspec/driver/mixins/_result_tools.py,sha256=wI8bGfDrGrG8nexPWiJdvgy1yi1_z643yKPsgMzBqiY,6915
+sqlspec/driver/mixins/_sql_translator.py,sha256=zpaJsS0_-zVvm3u2yCQ8VkILrZ7j28925SyZe1ChvlM,1491
+sqlspec/extensions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sqlspec/extensions/aiosql/__init__.py,sha256=-9cefc9pYPf9vCgALoB-y1DtmcgRjKe2azfl6RIarAA,414
+sqlspec/extensions/aiosql/adapter.py,sha256=CXkNZaZq86ZhfYFGx4IFbkHmbIFQKMd9CS6Q2jkMCok,16009
+sqlspec/extensions/litestar/__init__.py,sha256=tOmQ7RHSWOot7p30gk0efxxuP0OCq1opyyZqNmQY7FE,320
+sqlspec/extensions/litestar/_utils.py,sha256=o-FuUj1_WkDrLxQxiP6hXDak66XfyRP3QLyEVKrIRjI,1954
+sqlspec/extensions/litestar/cli.py,sha256=NYjPIWe0jESBgS0AG-3bdpcC5YM4RmZeeeTeFQK5Yqo,1362
+sqlspec/extensions/litestar/config.py,sha256=3UI_vhtbupCLsf1nhUgUpRlCoUS5c0GsAjWvegT0c3c,4462
+sqlspec/extensions/litestar/handlers.py,sha256=3LreU8rZvuHaJnKlN09ttu4wSorWJedsuKgeLT-cOEc,9993
+sqlspec/extensions/litestar/plugin.py,sha256=lojdLit4Uop6fpJWGXgpWrbGUBgCWR_khOK4CdRYpKM,5615
+sqlspec/extensions/litestar/providers.py,sha256=5LRb5JvRV_XZdNOKkdaIy3j5x-dFCcAi1ea1pgwuapI,18882
+sqlspec/migrations/__init__.py,sha256=n9y2yLQb02zMz36bPXnrInsSIMLunwHvO7e8UvCYBJc,1061
+sqlspec/migrations/base.py,sha256=D5-k0m4SnmVU0QRRPcMCvMAyEoAaceYYmD8JprSozDA,13251
+sqlspec/migrations/commands.py,sha256=LUAOOJcfolv75ZRV33zjdZ9kocyJSJet5hEZhUVIemU,18490
+sqlspec/migrations/loaders.py,sha256=wz-VUd1fEtOg-b1yWnLTTNtHkZkrjw-4o5hEqwUxbS0,12952
+sqlspec/migrations/runner.py,sha256=OxzindA2EzjCAVqI59LxfWC02Ful-aPsOYY9yUV7ypA,7330
+sqlspec/migrations/tracker.py,sha256=-gfHgvcYo1uBdr9FKrIxaU_0v734K5gpdLMuoaX29nQ,5013
+sqlspec/migrations/utils.py,sha256=gWnCOdr8pwfkgG-FSUJgRz4q9TlCgOXY_B7n59NrgVA,3746
+sqlspec/storage/__init__.py,sha256=bLrUfW_E41HH-Pi5kqFTPVYZiSsxlG7OZO1xP23nDSI,691
+sqlspec/storage/capabilities.py,sha256=vyousxls9jISsykgoybpNHlGWN6Hq_pKcsZ5DmKGWvU,3045
+sqlspec/storage/registry.py,sha256=NV-NdAcMdGqU5bHbXbkMuU_HRYuEgii7B4ldXnVSIo4,9529
+sqlspec/storage/backends/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sqlspec/storage/backends/base.py,sha256=KS2JRZILoH_R_xsfKtYkqQ5a1r5OOBDSE5KbibTmhGY,5730
+sqlspec/storage/backends/fsspec.py,sha256=YVr8X_q5F4Xrr9SyfUDve1dZ9E6KppuM8w6tvLwPKpc,16050
+sqlspec/storage/backends/obstore.py,sha256=8DEpiF3SCfbWx-NxoQXZ8piJOr-9f1WURYK69hCDhnQ,19864
+sqlspec/utils/__init__.py,sha256=DJiG2RGFaL9C0lLZStByzgR_GiPHfdkypAkTEGjTakw,211
+sqlspec/utils/correlation.py,sha256=2jvkAY3nkU3UxNU_9pbBR6cz3A1Q1cGG9IaWSSOIb1Q,4195
+sqlspec/utils/deprecation.py,sha256=L5ylPlkrWahXZ3q2Yta2ReFh4pA8rZapLNw9u_mOOEE,3794
+sqlspec/utils/fixtures.py,sha256=Qm2uNMaL_6l6tlij-Pm3tLwD906iFK_OXhwZqOx3WgY,1807
+sqlspec/utils/logging.py,sha256=ZEaj7J8oALrs_-p535vSUFo2tfrHkOEcL5pK2JGGFgM,3708
+sqlspec/utils/module_loader.py,sha256=m5PSN9NwOLd0ZJBuqMVYVi-vaIQMBCVd25vnM3-rv3k,2823
+sqlspec/utils/serializers.py,sha256=TKsRryRcYMnb8Z8MGkYGClIxcYvC8CW7MsrPQTJqEcY,154
+sqlspec/utils/singleton.py,sha256=SKnszJi1NPeERgX7IjVIGYAYx4XqR1E_rph3bU6olAU,1047
+sqlspec/utils/sync_tools.py,sha256=Iv1Msj1tb24iJhhulL_wcNsyjRComzAkyUNlefVci3w,7348
+sqlspec/utils/text.py,sha256=n5K0gvXvyCc8jNteNKsBOymwf_JnQ65f3lu0YaYq4Ys,2898
+sqlspec/utils/type_guards.py,sha256=AJKSQ323mNfhRbytI2e4N2cy2P6MMG7fp75foVnz328,30339
+sqlspec-0.15.0.dist-info/METADATA,sha256=GFcTBVzGgY-1_Wc-fL_9l_lt2JN1i__L4uN9uT6rOhw,16822
+sqlspec-0.15.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sqlspec-0.15.0.dist-info/entry_points.txt,sha256=G-ZqY1Nuuw3Iys7nXw23f6ILenk_Lt47VdK2mhJCWHg,53
+sqlspec-0.15.0.dist-info/licenses/LICENSE,sha256=MdujfZ6l5HuLz4mElxlu049itenOR3gnhN1_Nd3nVcM,1078
+sqlspec-0.15.0.dist-info/licenses/NOTICE,sha256=Lyir8ozXWov7CyYS4huVaOCNrtgL17P-bNV-5daLntQ,1634
+sqlspec-0.15.0.dist-info/RECORD,,

sqlspec-0.15.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+sqlspec = sqlspec.__main__:run_cli

sqlspec/driver/connection.py

@@ -1,207 +0,0 @@
-"""Consolidated connection management utilities for database drivers.
-
-This module provides centralized connection handling to avoid duplication
-across database adapter implementations.
-"""
-
-from contextlib import asynccontextmanager, contextmanager
-from typing import TYPE_CHECKING, Any, Optional, TypeVar, cast
-
-if TYPE_CHECKING:
-    from collections.abc import AsyncIterator, Iterator
-
-from sqlspec.utils.type_guards import is_async_transaction_capable, is_sync_transaction_capable
-
-__all__ = (
-    "get_connection_info",
-    "managed_connection_async",
-    "managed_connection_sync",
-    "managed_transaction_async",
-    "managed_transaction_sync",
-    "validate_pool_config",
-)
-
-
-ConnectionT = TypeVar("ConnectionT")
-PoolT = TypeVar("PoolT")
-
-
-@contextmanager
-def managed_connection_sync(config: Any, provided_connection: Optional[ConnectionT] = None) -> "Iterator[ConnectionT]":
-    """Context manager for database connections.
-
-    Args:
-        config: Database configuration with provide_connection method
-        provided_connection: Optional existing connection to use
-
-    Yields:
-        Database connection
-    """
-    if provided_connection is not None:
-        yield provided_connection
-        return
-
-    # Get connection from config
-    with config.provide_connection() as connection:
-        yield connection
-
-
-@contextmanager
-def managed_transaction_sync(connection: ConnectionT, auto_commit: bool = True) -> "Iterator[ConnectionT]":
-    """Context manager for database transactions.
-
-    Args:
-        connection: Database connection
-        auto_commit: Whether to auto-commit on success
-
-    Yields:
-        Database connection
-    """
-    # Check if connection already has autocommit enabled
-    has_autocommit = getattr(connection, "autocommit", False)
-
-    if not auto_commit or not is_sync_transaction_capable(connection) or has_autocommit:
-        yield connection
-        return
-
-    try:
-        yield cast("ConnectionT", connection)
-        cast("Any", connection).commit()
-    except Exception:
-        # Some databases (like DuckDB) throw an error if rollback is called
-        # when no transaction is active. Catch and ignore these specific errors.
-        try:
-            cast("Any", connection).rollback()
-        except Exception as rollback_error:
-            # Check if this is a "no transaction active" type error
-            error_msg = str(rollback_error).lower()
-            if "no transaction" in error_msg or "transaction context error" in error_msg:
-                # Ignore rollback errors when no transaction is active
-                pass
-            else:
-                # Re-raise other rollback errors
-                raise
-        raise
-
-
-@asynccontextmanager
-async def managed_connection_async(
-    config: Any, provided_connection: Optional[ConnectionT] = None
-) -> "AsyncIterator[ConnectionT]":
-    """Async context manager for database connections.
-
-    Args:
-        config: Database configuration with provide_connection method
-        provided_connection: Optional existing connection to use
-
-    Yields:
-        Database connection
-    """
-    if provided_connection is not None:
-        yield provided_connection
-        return
-
-    # Get connection from config
-    async with config.provide_connection() as connection:
-        yield connection
-
-
-@asynccontextmanager
-async def managed_transaction_async(connection: ConnectionT, auto_commit: bool = True) -> "AsyncIterator[ConnectionT]":
-    """Async context manager for database transactions.
-
-    Args:
-        connection: Database connection
-        auto_commit: Whether to auto-commit on success
-
-    Yields:
-        Database connection
-    """
-    # Check if connection already has autocommit enabled
-    has_autocommit = getattr(connection, "autocommit", False)
-
-    if not auto_commit or not is_async_transaction_capable(connection) or has_autocommit:
-        yield connection
-        return
-
-    try:
-        yield cast("ConnectionT", connection)
-        await cast("Any", connection).commit()
-    except Exception:
-        # Some databases (like DuckDB) throw an error if rollback is called
-        # when no transaction is active. Catch and ignore these specific errors.
-        try:
-            await cast("Any", connection).rollback()
-        except Exception as rollback_error:
-            # Check if this is a "no transaction active" type error
-            error_msg = str(rollback_error).lower()
-            if "no transaction" in error_msg or "transaction context error" in error_msg:
-                # Ignore rollback errors when no transaction is active
-                pass
-            else:
-                # Re-raise other rollback errors
-                raise
-        raise
-
-
-def get_connection_info(connection: Any) -> dict[str, Any]:
-    """Extract connection information for logging/debugging.
-
-    Args:
-        connection: Database connection object
-
-    Returns:
-        Dictionary of connection information
-    """
-    info = {"type": type(connection).__name__, "module": type(connection).__module__}
-
-    # Try to get database name
-    for attr in ("database", "dbname", "db", "catalog"):
-        value = getattr(connection, attr, None)
-        if value is not None:
-            info["database"] = value
-            break
-
-    # Try to get host information
-    for attr in ("host", "hostname", "server"):
-        value = getattr(connection, attr, None)
-        if value is not None:
-            info["host"] = value
-            break
-
-    return info
-
-
-def validate_pool_config(
-    min_size: int, max_size: int, max_idle_time: Optional[int] = None, max_lifetime: Optional[int] = None
-) -> None:
-    """Validate connection pool configuration.
-
-    Args:
-        min_size: Minimum pool size
-        max_size: Maximum pool size
-        max_idle_time: Maximum idle time in seconds
-        max_lifetime: Maximum connection lifetime in seconds
-
-    Raises:
-        ValueError: If configuration is invalid
-    """
-    if min_size < 0:
-        msg = f"min_size must be >= 0, got {min_size}"
-        raise ValueError(msg)
-
-    if max_size < 1:
-        msg = f"max_size must be >= 1, got {max_size}"
-        raise ValueError(msg)
-
-    if min_size > max_size:
-        msg = f"min_size ({min_size}) cannot be greater than max_size ({max_size})"
-        raise ValueError(msg)
-
-    if max_idle_time is not None and max_idle_time < 0:
-        msg = f"max_idle_time must be >= 0, got {max_idle_time}"
-        raise ValueError(msg)
-
-    if max_lifetime is not None and max_lifetime < 0:
-        msg = f"max_lifetime must be >= 0, got {max_lifetime}"
-        raise ValueError(msg)

sqlspec/driver/mixins/_cache.py

@@ -1,114 +0,0 @@
-"""Adapter-level caching mixin for compiled SQL and prepared statements."""
-
-from typing import TYPE_CHECKING, Any, Optional
-
-from sqlspec.statement.cache import SQLCache
-from sqlspec.statement.parameters import ParameterStyle
-
-if TYPE_CHECKING:
-    from sqlspec.statement.sql import SQL
-
-__all__ = ("AsyncAdapterCacheMixin", "SyncAdapterCacheMixin")
-
-
-class SyncAdapterCacheMixin:
-    """Mixin for adapter-level SQL compilation caching.
-
-    This mixin provides:
-    - Compiled SQL caching to avoid repeated compilation
-    - Parameter style conversion caching
-    - Prepared statement name management (for supported databases)
-
-    Integrates transparently with existing adapter execution flow.
-    """
-
-    def __init__(self, *args: Any, **kwargs: Any) -> None:
-        """Initialize adapter with caching support."""
-        super().__init__(*args, **kwargs)
-
-        # Get cache configuration from config or use defaults
-        config = getattr(self, "config", None)
-        cache_size = getattr(config, "adapter_cache_size", 500) if config else 500
-        enable_cache = getattr(config, "enable_adapter_cache", True) if config else True
-
-        # Initialize caches
-        self._compiled_cache: Optional[SQLCache] = SQLCache(max_size=cache_size) if enable_cache else None
-        self._prepared_statements: dict[str, str] = {}
-        self._prepared_counter = 0
-
-    def _get_compiled_sql(self, statement: "SQL", target_style: ParameterStyle) -> tuple[str, Any]:
-        """Get compiled SQL with caching.
-
-        Args:
-            statement: SQL statement to compile
-            target_style: Target parameter style for compilation
-
-        Returns:
-            Tuple of (compiled_sql, parameters)
-        """
-        if self._compiled_cache is None:
-            # Caching disabled
-            return statement.compile(placeholder_style=target_style)
-
-        # Generate cache key
-        cache_key = self._adapter_cache_key(statement, target_style)
-
-        # Check cache
-        cached = self._compiled_cache.get(cache_key)
-        if cached is not None:
-            return cached  # type: ignore[no-any-return]
-
-        # Compile and cache
-        result = statement.compile(placeholder_style=target_style)
-        self._compiled_cache.set(cache_key, result)
-        return result
-
-    def _adapter_cache_key(self, statement: "SQL", style: ParameterStyle) -> str:
-        """Generate adapter-specific cache key.
-
-        Args:
-            statement: SQL statement
-            style: Parameter style
-
-        Returns:
-            Cache key string
-        """
-        # Use statement's internal cache key which includes SQL hash, params, and dialect
-        base_key = statement._cache_key()
-        # Add adapter-specific context
-        return f"{self.__class__.__name__}:{style.value}:{base_key}"
-
-    def _get_or_create_prepared_statement_name(self, sql_hash: str) -> str:
-        """Get or create a prepared statement name for the given SQL.
-
-        Used by PostgreSQL and other databases that support prepared statements.
-
-        Args:
-            sql_hash: Hash of the SQL statement
-
-        Returns:
-            Prepared statement name
-        """
-        if sql_hash in self._prepared_statements:
-            return self._prepared_statements[sql_hash]
-
-        # Create new prepared statement name
-        self._prepared_counter += 1
-        stmt_name = f"sqlspec_ps_{self._prepared_counter}"
-        self._prepared_statements[sql_hash] = stmt_name
-        return stmt_name
-
-    def _clear_adapter_cache(self) -> None:
-        """Clear all adapter-level caches."""
-        if self._compiled_cache is not None:
-            self._compiled_cache.clear()
-        self._prepared_statements.clear()
-        self._prepared_counter = 0
-
-
-class AsyncAdapterCacheMixin(SyncAdapterCacheMixin):
-    """Async version of AdapterCacheMixin.
-
-    Identical to AdapterCacheMixin since caching operations are synchronous.
-    Provided for naming consistency with async adapters.
-    """

sqlspec/driver/mixins/_csv_writer.py

@@ -1,91 +0,0 @@
-"""Optimized CSV writing utilities."""
-
-import csv
-from typing import TYPE_CHECKING, Any
-
-from sqlspec.typing import PYARROW_INSTALLED
-
-if TYPE_CHECKING:
-    from sqlspec.statement.result import SQLResult
-
-__all__ = ("write_csv", "write_csv_default", "write_csv_optimized")
-
-
-def _raise_no_column_names_error() -> None:
-    """Raise error when no column names are available."""
-    msg = "No column names available"
-    raise ValueError(msg)
-
-
-def write_csv(result: "SQLResult", file: Any, **options: Any) -> None:
-    """Write result to CSV file.
-
-    Args:
-        result: SQL result to write
-        file: File-like object to write to
-        **options: CSV writer options
-    """
-    if PYARROW_INSTALLED:
-        try:
-            write_csv_optimized(result, file, **options)
-        except Exception:
-            write_csv_default(result, file, **options)
-    else:
-        write_csv_default(result, file, **options)
-
-
-def write_csv_default(result: "SQLResult", file: Any, **options: Any) -> None:
-    """Write result to CSV file using default method.
-
-    Args:
-        result: SQL result to write
-        file: File-like object to write to
-        **options: CSV writer options
-    """
-    csv_options = options.copy()
-    csv_options.pop("compression", None)
-    csv_options.pop("partition_by", None)
-
-    writer = csv.writer(file, **csv_options)
-    if result.column_names:
-        writer.writerow(result.column_names)
-    if result.data:
-        if result.data and isinstance(result.data[0], dict):
-            rows = []
-            for row_dict in result.data:
-                row_values = [row_dict.get(col) for col in result.column_names or []]
-                rows.append(row_values)
-            writer.writerows(rows)
-        else:
-            writer.writerows(result.data)
-
-
-def write_csv_optimized(result: "SQLResult", file: Any, **options: Any) -> None:
-    """Write result to CSV using PyArrow if available for better performance.
-
-    Args:
-        result: SQL result to write
-        file: File-like object to write to
-        **options: CSV writer options
-    """
-    _ = options
-    import pyarrow as pa
-    import pyarrow.csv as pa_csv
-
-    if not result.data:
-        return
-
-    if not hasattr(file, "name"):
-        msg = "PyArrow CSV writer requires a file with a 'name' attribute"
-        raise ValueError(msg)
-
-    table: Any
-    if isinstance(result.data[0], dict):
-        table = pa.Table.from_pylist(result.data)
-    elif result.column_names:
-        data_dicts = [dict(zip(result.column_names, row)) for row in result.data]
-        table = pa.Table.from_pylist(data_dicts)
-    else:
-        _raise_no_column_names_error()
-
-    pa_csv.write_csv(table, file.name)  # pyright: ignore