maxi speichers changes adapted to new structure

Author: GiorgioPorgio

sqlmodel/sql/_expression_select_gen.py

@@ -15,6 +15,7 @@
 
 from sqlalchemy import (
     Column,
+    Label,
 )
 from sqlalchemy.sql.elements import (
     SQLCoreOperations,
@@ -40,6 +41,7 @@
     Column,  # type: ignore
     Sequence,  # type: ignore
     Mapping,  # type: ignore
+    Label,  # type: ignore
     UUID,
     datetime,
     float,
@@ -58,6 +60,7 @@
     Column,  # type: ignore
     Sequence,  # type: ignore
     Mapping,  # type: ignore
+    Label,  # type: ignore
     UUID,
     datetime,
     float,
@@ -76,6 +79,7 @@
     Column,  # type: ignore
     Sequence,  # type: ignore
     Mapping,  # type: ignore
+    Label,  # type: ignore
     UUID,
     datetime,
     float,
@@ -94,6 +98,7 @@
     Column,  # type: ignore
     Sequence,  # type: ignore
     Mapping,  # type: ignore
+    Label,  # type: ignore
     UUID,
     datetime,
     float,
@@ -119,6 +124,16 @@ def select(__ent0: _TScalar_0) -> SelectOfScalar[_TScalar_0]:  # type: ignore
     ...
 
 
+@overload
+def select(*entities: _TCCA[_T0]) -> Select[Tuple[_T0, ...]]:  # type: ignore
+    ...
+
+
+@overload
+def select(*entities: _TScalar_0) -> Select[Tuple[_TScalar_0, ...]]:  # type: ignore
+    ...
+
+
 # Generated overloads start
 
 

sqlmodel/sql/_expression_select_gen.py.jinja2

@@ -13,6 +13,7 @@ from uuid import UUID
 
 from sqlalchemy import (
     Column,
+    Label,
 )
 from sqlalchemy.sql.elements import (
     SQLCoreOperations,
@@ -39,6 +40,7 @@ _TScalar_{{ i }} = TypeVar(
     Column,  # type: ignore
     Sequence,  # type: ignore
     Mapping,  # type: ignore
+    Label, # type: ignore
     UUID,
     datetime,
     float,
@@ -64,6 +66,16 @@ def select(__ent0: _TScalar_0) -> SelectOfScalar[_TScalar_0]:  # type: ignore
     ...
 
 
+@overload
+def select(*entities: _TCCA[_T0]) -> Select[Tuple[_T0, ...]]:  # type: ignore
+    ...
+
+
+@overload
+def select(*entities: _TScalar_0) -> Select[Tuple[_TScalar_0, ...]]:  # type: ignore
+    ...
+
+
 # Generated overloads start
 
 {% for signature in signatures %}