schema: add ColumnElement to Column.server_default (#95)

Fixed #81

Author: MaicoTimmerman

sqlalchemy-stubs/sql/schema.pyi

@@ -25,6 +25,7 @@ from .base import DialectKWArgs
 from .base import SchemaEventTarget
 from .elements import ClauseElement
 from .elements import ColumnClause
+from .elements import ColumnElement
 from .elements import TextClause
 from .selectable import TableClause
 from .. import util
@@ -125,7 +126,9 @@ class Column(DialectKWArgs, SchemaItem, ColumnClause[_TE]):
     primary_key: bool = ...
     nullable: bool = ...
     default: Optional[Any] = ...
-    server_default: Optional[Union[FetchedValue, str, TextClause]] = ...
+    server_default: Optional[
+        Union[FetchedValue, str, TextClause, ColumnElement[_TE]]
+    ] = ...
     server_onupdate: Optional[FetchedValue] = ...
     index: Optional[bool] = ...
     unique: Optional[bool] = ...
@@ -152,7 +155,9 @@ class Column(DialectKWArgs, SchemaItem, ColumnClause[_TE]):
         nullable: bool = ...,
         onupdate: Optional[Any] = ...,
         primary_key: bool = ...,
-        server_default: Optional[Union[FetchedValue, str, TextClause]] = ...,
+        server_default: Optional[
+            Union[FetchedValue, str, TextClause, ColumnElement[Any]]
+        ] = ...,
         server_onupdate: Optional[FetchedValue] = ...,
         quote: Optional[bool] = ...,
         unique: Optional[bool] = ...,
@@ -173,7 +178,9 @@ class Column(DialectKWArgs, SchemaItem, ColumnClause[_TE]):
         nullable: bool = ...,
         onupdate: Optional[Any] = ...,
         primary_key: bool = ...,
-        server_default: Optional[Union[FetchedValue, str, TextClause]] = ...,
+        server_default: Optional[
+            Union[FetchedValue, str, TextClause, ColumnElement[Any]]
+        ] = ...,
         server_onupdate: Optional[FetchedValue] = ...,
         quote: Optional[bool] = ...,
         unique: Optional[bool] = ...,
@@ -196,7 +203,9 @@ class Column(DialectKWArgs, SchemaItem, ColumnClause[_TE]):
         nullable: bool = ...,
         onupdate: Optional[Any] = ...,
         primary_key: bool = ...,
-        server_default: Optional[Union[FetchedValue, str, TextClause]] = ...,
+        server_default: Optional[
+            Union[FetchedValue, str, TextClause, ColumnElement[_TE]]
+        ] = ...,
         server_onupdate: Optional[FetchedValue] = ...,
         quote: Optional[bool] = ...,
         unique: Optional[bool] = ...,
@@ -218,7 +227,9 @@ class Column(DialectKWArgs, SchemaItem, ColumnClause[_TE]):
         nullable: bool = ...,
         onupdate: Optional[Any] = ...,
         primary_key: bool = ...,
-        server_default: Optional[Union[FetchedValue, str, TextClause]] = ...,
+        server_default: Optional[
+            Union[FetchedValue, str, TextClause, ColumnElement[_TE]]
+        ] = ...,
         server_onupdate: Optional[FetchedValue] = ...,
         quote: Optional[bool] = ...,
         unique: Optional[bool] = ...,

test/files/column_server_default_ticket_81.py

@@ -0,0 +1,19 @@
+from sqlalchemy import Boolean
+from sqlalchemy import Column
+from sqlalchemy import DateTime
+from sqlalchemy import true
+from sqlalchemy.orm import registry
+from sqlalchemy.sql import functions as func
+
+reg: registry = registry()
+
+
+@reg.mapped
+class A:
+    __tablename__ = "a"
+
+    b = Column(Boolean, nullable=False, server_default=true())
+    c = Column(DateTime, server_default=func.now(), nullable=False)
+
+    # EXPECTED: error: Cannot infer type argument 1 of "Column"
+    d = Column(Boolean, server_default=func.now(), nullable=False)