diff --git a/third_party/2.7/sqlalchemy/engine/__init__.pyi b/third_party/2.7/sqlalchemy/engine/__init__.pyi index 18ceed96a..49eca6d67 100644 --- a/third_party/2.7/sqlalchemy/engine/__init__.pyi +++ b/third_party/2.7/sqlalchemy/engine/__init__.pyi @@ -2,5 +2,10 @@ # # NOTE: This dynamically typed stub was automatically generated by stubgen. +from .base import Connection as Connection +from .base import Engine as Engine +from .base import RowProxy as RowProxy +from .base import Transaction as Transaction + def create_engine(*args, **kwargs): ... def engine_from_config(configuration, prefix=..., **kwargs): ... diff --git a/third_party/2.7/sqlalchemy/engine/base.pyi b/third_party/2.7/sqlalchemy/engine/base.pyi index 3b8981398..71f6e196a 100644 --- a/third_party/2.7/sqlalchemy/engine/base.pyi +++ b/third_party/2.7/sqlalchemy/engine/base.pyi @@ -1,3 +1,21 @@ +from typing import Any, List, Tuple + # Dummy until I figure out something better. class Connectable: pass + +class Connection: + def begin(self): ... + def execute(self, object, *multiparams, **params): ... + +class Engine(object): ... + +class RowProxy: + def items(self) -> List[Tuple[Any, Any]]: ... + def keys(self) -> List[Any]: ... + def values(self) -> List[Any]: ... + def __getitem__(self, key: str): ... + +class Transaction: + def commit(self): ... + def rollback(self): ... diff --git a/third_party/2.7/sqlalchemy/sql/schema.pyi b/third_party/2.7/sqlalchemy/sql/schema.pyi index 403795e0d..d40404df0 100644 --- a/third_party/2.7/sqlalchemy/sql/schema.pyi +++ b/third_party/2.7/sqlalchemy/sql/schema.pyi @@ -20,7 +20,13 @@ class SchemaItem(SchemaEventTarget, visitors.Visitable): def __repr__(self): ... -class Table(DialectKWArgs, SchemaItem, TableClause): ... +class Table(DialectKWArgs, SchemaItem, TableClause): + def __init__(self, name, metadata, *args, **kwargs): ... + def insert(self, *args, **kwargs): ... + def select(self, *args, **kwargs): ... + def update(self, *args, **kwargs): ... + def c(self, *args, **kwargs): ... + class Column(SchemaItem, ColumnClause): primary_key = ... # type: Any