Column wrapping functions not handled correctly

[sqlalchemy]

Column wrapping functions not handled correctly

[zzzeek]

Yes I think ideally your function would have to use a generic like that, then I could pick up the "T" part.

however great news! there's a workaround you can use for now, which is the _mypy_mapped_attrs attribute, this passes the SQLAlchemy part of this (the --strict thing above is still separate):

class Foo(Base):
    bar: int = MyColumn(Integer)

    _mypy_mapped_attrs = [bar]



foo = Foo(bar=1)

[alxdb]

ah okay, I see the issue now. Would there be a way to annotate the wrapper function with a generic type? something like def MyColumn(col_type: T, *args, **kwargs) -> Column[T]:?

PS: Thanks so much for your work on this great project 💯 this isn't a serious issue for me atm, so I'm happy to wait a while for this to be addressed

[zzzeek]

hey there -

i can try to improve this and perhaps have your Column considered to be reutrning "Any", but I can't derive that this Column is against Integer because your function doesn't indicate this:

from sqlalchemy import Column, Integer, orm

def MyColumn(*args, **kwargs) -> Column:
    return Column(*args, **kwargs)

x = MyColumn(Integer)

$ mypy test3.py --strict
test3.py:3: error: Function is missing a type annotation for one or more arguments
test3.py:3: error: Missing type parameters for generic type "Column"

I'm pretty backlogged now but will try to get it to work at least as:

class Foo(Base):
    bar: int = MyColumn(Integer)

[peteris-zealid]

Another solution is to do

class MyInteger(Integer):
    pass

class Foo(Base):
    bar: int = Column(MyInteger())

And for isinstance checks you can use isinstance(col.type, MyInteger)