GitHub - dropbox/sqlalchemy-stubs: Mypy plugin and stubs for SQLAlchemy (original) (raw)

Mypy plugin and stubs for SQLAlchemy

This package contains type stubs and amypy pluginto provide more precise static types and type inference forSQLAlchemy framework. SQLAlchemy uses some Python "magic" that makes having precise types for some code patterns problematic. This is why we need to accompany the stubs with mypy plugins. The final goal is to be able to get precise types for most common patterns. Currently, basic operations with models are supported. A simple example:

from sqlalchemy.ext.declarative import declarative_base from sqlalchemy import Column, Integer, String

Base = declarative_base()

class User(Base): tablename = 'users' id = Column(Integer, primary_key=True) name = Column(String)

user = User(id=42, name=42) # Error: Incompatible type for "name" of "User" # (got "int", expected "Optional[str]") user.id # Inferred type is "int" User.name # Inferred type is "Column[Optional[str]]"

Some auto-generated attributes are added to models. Simple relationships are supported but require models to be imported:

from typing import TYPE_CHECKING if TYPE_CHECKING: from models.address import Address

...

class User(Base): tablename = 'users' id = Column(Integer, primary_key=True) name = Column(String) address = relationship('Address') # OK, mypy understands string references.

The next step is to support precise types for table definitions (e.g. inferring Column[Optional[str]] for users.c.name, currently it is justColumn[Any]), and precise types for results of queries made using query()and select().

Installation

Install latest published version as:

pip install -U sqlalchemy-stubs

Important: you need to enable the plugin in your mypy config file:

To install the development version of the package:

git clone https://github.com/dropbox/sqlalchemy-stubs
cd sqlalchemy-stubs
pip install -U .

Development Setup

First, clone the repo and cd into it, like in Installation, then:

git submodule update --init --recursive
pip install -r dev-requirements.txt

Then, to run the tests, simply:

Development status

The package is currently in alpha stage. See issue trackerfor bugs and missing features. If you want to contribute, a good place to start ishelp-wanted label.

Currently, some basic use cases like inferring model field types are supported. The long term goal is to be able to infer types for more complex situations like correctly inferring columns in most compound queries.

External contributions to the project should be subject toDropbox Contributor License Agreement (CLA).