From e46d308ab49fab51fcbde45724b0e4847610fecd Mon Sep 17 00:00:00 2001 From: Herbert Ho Date: Thu, 9 Jun 2016 17:58:19 -0700 Subject: [PATCH] fleshing out some more stubs for sqlalchemy (#282) --- third_party/2.7/sqlalchemy/engine/__init__.pyi | 5 +++++ third_party/2.7/sqlalchemy/engine/base.pyi | 18 ++++++++++++++++++ third_party/2.7/sqlalchemy/sql/schema.pyi | 8 +++++++- 3 files changed, 30 insertions(+), 1 deletion(-) 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