brain_sqlalchemy.py 1009 B

123456789101112131415161718192021222324252627282930313233343536373839
  1. # Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
  2. # For details: https://github.com/PyCQA/astroid/blob/main/LICENSE
  3. # Copyright (c) https://github.com/PyCQA/astroid/blob/main/CONTRIBUTORS.txt
  4. from astroid.brain.helpers import register_module_extender
  5. from astroid.builder import parse
  6. from astroid.manager import AstroidManager
  7. def _session_transform():
  8. return parse(
  9. """
  10. from sqlalchemy.orm.session import Session
  11. class sessionmaker:
  12. def __init__(
  13. self,
  14. bind=None,
  15. class_=Session,
  16. autoflush=True,
  17. autocommit=False,
  18. expire_on_commit=True,
  19. info=None,
  20. **kw
  21. ):
  22. return
  23. def __call__(self, **local_kw):
  24. return Session()
  25. def configure(self, **new_kw):
  26. return
  27. return Session()
  28. """
  29. )
  30. register_module_extender(AstroidManager(), "sqlalchemy.orm.session", _session_transform)