summaryrefslogtreecommitdiff
path: root/astroid/brain/brain_sqlalchemy.py
blob: c80eee1ed73a0d33da0d34c2c36f85b120f93fca (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import astroid


def _session_transform():
    return astroid.parse(
        """
    from sqlalchemy.orm.session import Session

    class sessionmaker:
        def __init__(
            self,
            bind=None,
            class_=Session,
            autoflush=True,
            autocommit=False,
            expire_on_commit=True,
            info=None,
            **kw
        ):
            return

        def __call__(self, **local_kw):
            return Session()

        def configure(self, **new_kw):
            return

        return Session()
    """
    )


astroid.register_module_extender(
    astroid.MANAGER, "sqlalchemy.orm.session", _session_transform
)