astroid_compat.py 586 B

123456789101112131415161718
  1. try:
  2. from astroid import nodes as astroid_node_classes
  3. # astroid_node_classes should be whichever module has the NodeNG class
  4. from astroid.nodes import NodeNG
  5. from astroid.nodes import BaseContainer
  6. except Exception:
  7. try:
  8. from astroid import node_classes as astroid_node_classes
  9. from astroid.node_classes import NodeNG
  10. from astroid.node_classes import _BaseContainer as BaseContainer
  11. except Exception: # pragma: no cover
  12. astroid_node_classes = None
  13. NodeNG = None
  14. BaseContainer = None
  15. __all__ = ["astroid_node_classes", "NodeNG", "BaseContainer"]