blob: d2352ce04bed71639482d6129127700751df4ebe (
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
|
from astroid.brain.helpers import register_module_extender
from astroid.builder import parse
from astroid.manager import AstroidManager
def _session_transform():
return 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()
"""
)
register_module_extender(AstroidManager(), "sqlalchemy.orm.session", _session_transform)
|