sqlspec 0.13.1__py3-none-any.whl → 0.16.2__py3-none-any.whl

{sqlspec-0.13.1.dist-info → sqlspec-0.16.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlspec
-Version: 0.13.1
+Version: 0.16.2
 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.16.2.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=FIYp2u5NgIBc375Jkt2ZfA81Xs7ABcKY92A5rwWXuFE,60340
+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=iwwy7zdIBV7TcoxIYpKuTvN5fGiULQac2f4a-saxyKU,12937
+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=yz-6e-x66vrNne6z5zq4Ae0C3p0mHEEIe1Y8er-A0pg,17812
+sqlspec/builder/_column.py,sha256=46baZj403BKfGjZcMc9LtQfMLeMQ7ROPyFL64V7dDM0,13124
+sqlspec/builder/_ddl.py,sha256=A_fV4d92o2ZOhX150YMSsQDm3veQTQrwlxgLdFMBBfg,48184
+sqlspec/builder/_ddl_utils.py,sha256=1mFSNe9w5rZXA1Ud4CTuca7eibi0XayHrIPcnEgRB7s,4034
+sqlspec/builder/_delete.py,sha256=xWA5nQB3UB8kpEGXN2k5ynt4cGZ7blkNoURpI0bKoeg,2264
+sqlspec/builder/_insert.py,sha256=rARWh5olbur6oPP_3FoAuJp8irj5cRLW0mKdWEx2cqU,16896
+sqlspec/builder/_merge.py,sha256=95PLQSKA3zjk0wTZG3m817fTZpsS95PrS2qF34iLAP8,2004
+sqlspec/builder/_parsing_utils.py,sha256=LyvFJ6nCU7b-h8QFyg4cYtcc0RKVzI_sHTeOVrU1_s8,7102
+sqlspec/builder/_select.py,sha256=m5sfyuAssjlNimLLNBAeFooVIfM2FgKN1boPfdsOkaA,5785
+sqlspec/builder/_update.py,sha256=UFHM_uWVY5RnZQ6winiyjKNtBryKRAXJlXtCVQdifyw,6015
+sqlspec/builder/mixins/__init__.py,sha256=YXhAzKmQbQtne5j26SKWY8PUxwosl0RhlhLoahAdkj0,1885
+sqlspec/builder/mixins/_cte_and_set_ops.py,sha256=p5O9m_jvpaWxv1XP9Ys2DRI-qOTq30rr2EwYjAbIT8o,9088
+sqlspec/builder/mixins/_delete_operations.py,sha256=l0liajnoAfRgtWtyStuAIfxreEFRkNO4DtBwyGqAfic,1198
+sqlspec/builder/mixins/_insert_operations.py,sha256=3ZuVNAPgJG0fzOPaprwUPa0Un3NP7erHwtCg8AGZWD8,9500
+sqlspec/builder/mixins/_join_operations.py,sha256=hZKuKCGjPRX379g0DblADRXdU8mtEiEOEerxCRHuo28,7280
+sqlspec/builder/mixins/_merge_operations.py,sha256=e9QDv1s84-2F2ZAZrr7UJtKXhy3X0NDN7AZ--8mOTKw,24193
+sqlspec/builder/mixins/_order_limit_operations.py,sha256=ABPuFSqHRv7XaS9-3HNZO3Jn0QovhJrkYT158xxduns,4835
+sqlspec/builder/mixins/_pivot_operations.py,sha256=j5vdzXuEqB1Jn3Ie_QjVwSH2_OEi65oZ64bQJHd3jXo,6108
+sqlspec/builder/mixins/_select_operations.py,sha256=ArFWrEGk6ZDiwfi5gUseGTBHCzsSKWfF2EjJlJWL2Ek,25021
+sqlspec/builder/mixins/_update_operations.py,sha256=lk9VRM0KGmYhofbWChemJxSZF6I0LhrRgqMXVmXZMeU,8650
+sqlspec/builder/mixins/_where_clause.py,sha256=jqz58H9X2ZDsiQH9DeBrrQJf-Am7-x1ApJH47U2DFfo,33789
+sqlspec/core/__init__.py,sha256=rU_xGsXhqIOnBbyB2InhJknYePm5NQ2DSWdBigror4g,1775
+sqlspec/core/cache.py,sha256=cLL9bd5wn1oeMzn5E5Ym0sAemA8U4QP6B55x4L9-26I,27044
+sqlspec/core/compiler.py,sha256=pfBRrMFvrNuCwQgzKNljNTYH3imARlek3Ja5mJkHI88,14652
+sqlspec/core/filters.py,sha256=X0wRd0vNOFgeOK98ReeTyKt408GCnnmE9p45Bvur3kw,31351
+sqlspec/core/hashing.py,sha256=4KyAFWtFDMYreoBGGPQppEuMWO6_NrRYlw9Lm-qeJqo,10429
+sqlspec/core/parameters.py,sha256=yLnGt_tqqpb28xrgnq3oQ-1mcf45Pjf6uE3c0ag5JJ4,54265
+sqlspec/core/result.py,sha256=VnxiTW7m6QfQkKRA8U_WtX198HelX2TCHd4aKPmKHHo,21340
+sqlspec/core/splitter.py,sha256=cb2P1B0q5vvALHi3SEJ7VdbRHH2GWsftrmJi9PYMbeE,28089
+sqlspec/core/statement.py,sha256=HsbSu0qnfJzpL_s1kwfxtWnyuCfAq8WjYRc3YQqxkDw,25700
+sqlspec/driver/__init__.py,sha256=QVpDRQGd1GreIP199en6qDbq-cZJcEF5go68DINagUk,569
+sqlspec/driver/_async.py,sha256=29XYWjeIuIwElO3QFUtEqQZq1ss2Ulueh0UD-aX2x-E,19636
+sqlspec/driver/_common.py,sha256=bjJCBfM5YaU-98vEnEc5qFcjWvaUVEcQS6v3gGC0UKw,26313
+sqlspec/driver/_sync.py,sha256=pCKNHj46HcZYn9292FWyoWkNc6gj4ArUT8ttik4YyRQ,19408
+sqlspec/driver/mixins/__init__.py,sha256=gN4pQyJXxNy0xi91dcMJGA7DQ7TbjGjQI24SSpZc6Go,248
+sqlspec/driver/mixins/_result_tools.py,sha256=8z-W4py3BOtn3WB7ElpsVAEjGRozgHsfymTE_oXqcnw,7576
+sqlspec/driver/mixins/_sql_translator.py,sha256=jmwlocjSqj-ZMfssMva6GR8t-CRlDVwU3ec2ve0l3JE,3322
+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=X4DlAx3Ry-ccOjAQSxe8SMtyJKCFJVLTbENPU_efKuU,1356
+sqlspec/extensions/litestar/config.py,sha256=3UI_vhtbupCLsf1nhUgUpRlCoUS5c0GsAjWvegT0c3c,4462
+sqlspec/extensions/litestar/handlers.py,sha256=3LreU8rZvuHaJnKlN09ttu4wSorWJedsuKgeLT-cOEc,9993
+sqlspec/extensions/litestar/plugin.py,sha256=I0aRnL4oZPUYk7pYhZSL3yikl7ViM0kr33kVmH4W-MU,5637
+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=WRuk1ZEhb_0CRrumAdnmi-i-dV6qVd3cgJyZw8RY9QQ,7390
+sqlspec/utils/text.py,sha256=n5K0gvXvyCc8jNteNKsBOymwf_JnQ65f3lu0YaYq4Ys,2898
+sqlspec/utils/type_guards.py,sha256=9C4SRebO4JiQrMzcJZFUA0KjSU48G26RmX6lbijyjBg,30476
+sqlspec-0.16.2.dist-info/METADATA,sha256=uxIDZpVGDc5LNmEdLfdULbelze5R5WSAxXwRYXZ42ZQ,16822
+sqlspec-0.16.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sqlspec-0.16.2.dist-info/entry_points.txt,sha256=G-ZqY1Nuuw3Iys7nXw23f6ILenk_Lt47VdK2mhJCWHg,53
+sqlspec-0.16.2.dist-info/licenses/LICENSE,sha256=MdujfZ6l5HuLz4mElxlu049itenOR3gnhN1_Nd3nVcM,1078
+sqlspec-0.16.2.dist-info/licenses/NOTICE,sha256=Lyir8ozXWov7CyYS4huVaOCNrtgL17P-bNV-5daLntQ,1634
+sqlspec-0.16.2.dist-info/RECORD,,

sqlspec-0.16.2.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/_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