sqlspec 0.62.1__py3-none-any.whl → 0.62.2__py3-none-any.whl

sqlspec/builder/_ddl.py

@@ -688,7 +688,7 @@ class _SingleObjectDropBuilder(DDLBuilder, _IfExistsDDLMixin, _CascadeRestrictDD
         self._require(self._name, f"{self._object_label} name must be set for DROP {self._drop_kind}.")
         return exp.Drop(
             kind=self._drop_kind,
-            this=self._build_drop_this(),
+            tables=[self._build_drop_this()],
             exists=self._if_exists,
             cascade=self._cascade,
             **self._drop_expression_args(),
@@ -1464,10 +1464,10 @@ class AlterTable(DDLBuilder, _IfExistsDDLMixin):
             return build_column_expression(op.column_definition)
 
         if op_type == "DROP COLUMN":
-            return exp.Drop(this=exp.to_identifier(op.column_name), kind="COLUMN", exists=True)
+            return exp.Drop(tables=[exp.to_identifier(op.column_name)], kind="COLUMN", exists=True)
 
         if op_type == "DROP COLUMN CASCADE":
-            return exp.Drop(this=exp.to_identifier(op.column_name), kind="COLUMN", cascade=True, exists=True)
+            return exp.Drop(tables=[exp.to_identifier(op.column_name)], kind="COLUMN", cascade=True, exists=True)
 
         if op_type == "ALTER COLUMN TYPE":
             if not op.new_type:
@@ -1488,10 +1488,12 @@ class AlterTable(DDLBuilder, _IfExistsDDLMixin):
             return exp.AddConstraint(expressions=[constraint_expr])
 
         if op_type == "DROP CONSTRAINT":
-            return exp.Drop(this=exp.to_identifier(op.constraint_name), kind="CONSTRAINT", exists=True)
+            return exp.Drop(tables=[exp.to_identifier(op.constraint_name)], kind="CONSTRAINT", exists=True)
 
         if op_type == "DROP CONSTRAINT CASCADE":
-            return exp.Drop(this=exp.to_identifier(op.constraint_name), kind="CONSTRAINT", cascade=True, exists=True)
+            return exp.Drop(
+                tables=[exp.to_identifier(op.constraint_name)], kind="CONSTRAINT", cascade=True, exists=True
+            )
 
         if op_type == "ALTER COLUMN SET NOT NULL":
             return exp.AlterColumn(this=exp.to_identifier(op.column_name), allow_null=False)

sqlspec/extensions/litestar/plugin.py

@@ -833,12 +833,21 @@ class SQLSpecPlugin(InitPluginProtocol, CLIPlugin):
         Args:
             name: A ``session_key``, ``connection_key``, or ``pool_key``.
 
+        Raises:
+            KeyError: If ``name`` is not a known dependency key.
+
         Returns:
             The matching plugin state.
         """
-        if any(state.annotation is None for state in self._plugin_configs):
-            self._raise_plugin_not_registered()
-        return self._get_plugin_state(name)
+        if isinstance(name, str) and any(
+            name in {state.connection_key, state.pool_key, state.session_key} for state in self._plugin_configs
+        ):
+            if any(state.annotation is None for state in self._plugin_configs):
+                self._raise_plugin_not_registered()
+            return self._get_plugin_state(name)
+        self._raise_config_not_found(name)
+        msg = "unreachable"
+        raise AssertionError(msg)
 
     def _get_plugin_state(
         self, key: "str | DatabaseConfigProtocol[Any, Any, Any] | type[DatabaseConfigProtocol[Any, Any, Any]]"

{sqlspec-0.62.1.dist-info → sqlspec-0.62.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: sqlspec
-Version: 0.62.1
+Version: 0.62.2
 Summary: SQL Experiments in Python
 Project-URL: Discord, https://discord.gg/litestar
 Project-URL: Issue, https://github.com/litestar-org/sqlspec/issues/

{sqlspec-0.62.1.dist-info → sqlspec-0.62.2.dist-info}/RECORD

@@ -274,7 +274,7 @@ sqlspec/adapters/sqlite/litestar/store.py,sha256=d8Uk2YdbbkpIWKhLmVWue16fXVRfALx
 sqlspec/builder/__init__.py,sha256=S-yrPxGah-kKg0OtDE05qkB8nIRsVlNmtsP5dEchKW8,4380
 sqlspec/builder/_base.py,sha256=JPhbA_NyHsvB1cCvkyX4qYmfYKIR5-Hb3Xt3h4y3ACA,39774
 sqlspec/builder/_column.py,sha256=_Do3Xt7wkk2uEmdNKIIx3uMgEjdeXhGMmQbMbIefQ1Y,16312
-sqlspec/builder/_ddl.py,sha256=ll33voSWitHYFKt7pgrlZc3_p2hpIpUimkdEEryemCA,58056
+sqlspec/builder/_ddl.py,sha256=bNpLdHYo9z_N5U518dR2ua2Se8sFgD2pR8a8OAFQk_M,58106
 sqlspec/builder/_delete.py,sha256=--KFsRs1XUA3TE1PZNd5YQgmyCJGaA4GvEGgiUho80U,2614
 sqlspec/builder/_dml.py,sha256=1uQJ9jcXXHTQcfe00voNmEIxLvoaRAjZgScLZF_jO4w,15373
 sqlspec/builder/_explain.py,sha256=S7wDHWeraZDft_hJZm6ajgJiwIzRo6E5-yIXThhTsH4,18813
@@ -583,7 +583,7 @@ sqlspec/extensions/litestar/channels.py,sha256=W-EKVEmXOPrdrcsNh1a-FzIWGcdEE6V24
 sqlspec/extensions/litestar/cli.py,sha256=6ZHIWNT0RPzgr3LOKVAIEk9hUcyJTMUKgnpjkBrdliA,3476
 sqlspec/extensions/litestar/config.py,sha256=EvHRJjAdGuJkbRj0Ke7QsU6eXef_okLLPm0-atD4uXM,3597
 sqlspec/extensions/litestar/handlers.py,sha256=VTtBoCfl8zxtEe9N1_aZrFA4ILrEbP_2FjFuFg-9FF4,11162
-sqlspec/extensions/litestar/plugin.py,sha256=4-dQc_x85_Eoi_buSZ7oqFVeIlkfRy9ND8K0uz1YbPg,46142
+sqlspec/extensions/litestar/plugin.py,sha256=t_e1MHDOx-8Hzn9s7Ua7elcWh68Ove8GVED0xaN6a_8,46506
 sqlspec/extensions/litestar/providers.py,sha256=ofpZf9EJbXVmVyqKkAM2ef1HM_4bEzpAT7EvK1Ejf7k,41361
 sqlspec/extensions/litestar/store.py,sha256=4b9QC4s7R7Xh_6WDbrGKyES__XZuIhtpWw_6uUTDHBg,9996
 sqlspec/extensions/litestar/migrations/0001_create_session_table.py,sha256=60HehSre1tXoHaOBx_75Hha-7smWP61RYrRVBYnIo8w,4206
@@ -665,8 +665,8 @@ sqlspec/utils/serializers/__init__.py,sha256=X99t8YMHVGmKwrqPxumnSXu4JMJL8SpKoU7
 sqlspec/utils/serializers/_json.py,sha256=5PBan5kHoDAGqbOqT33SdEK17VZzNjzZKneWI_BnDzY,18136
 sqlspec/utils/serializers/_numpy.py,sha256=_LllCtVn_gyLrRG2naHM7Unjtpot86BPTvGPlZuAGRw,1742
 sqlspec/utils/serializers/_schema.py,sha256=kN8lW8GJa08yMi_qN_x8A7XpO7V6aAaoYqKq72rxaCk,11416
-sqlspec-0.62.1.dist-info/METADATA,sha256=IR46oDOTIdV42G8U5l54Gttwyw-V4qv0N5bZAX48dNs,9066
-sqlspec-0.62.1.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
-sqlspec-0.62.1.dist-info/entry_points.txt,sha256=T_1g2gTpkUsl6YC-KSCoNS5r7CPG2kn6wyp1WyMvpqU,404
-sqlspec-0.62.1.dist-info/licenses/LICENSE,sha256=MdujfZ6l5HuLz4mElxlu049itenOR3gnhN1_Nd3nVcM,1078
-sqlspec-0.62.1.dist-info/RECORD,,
+sqlspec-0.62.2.dist-info/METADATA,sha256=pdDXwLK4BTqODd8EBgo8AAaRs7cRPDUh4sRvnZYACyY,9066
+sqlspec-0.62.2.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
+sqlspec-0.62.2.dist-info/entry_points.txt,sha256=T_1g2gTpkUsl6YC-KSCoNS5r7CPG2kn6wyp1WyMvpqU,404
+sqlspec-0.62.2.dist-info/licenses/LICENSE,sha256=MdujfZ6l5HuLz4mElxlu049itenOR3gnhN1_Nd3nVcM,1078
+sqlspec-0.62.2.dist-info/RECORD,,