Replace ipaplatform's symlinks with a meta importer

Signed-off-by: Christian Heimes <cheimes@redhat.com>
Reviewed-By: Petr Spacek <pspacek@redhat.com>

1 files changed, 19 insertions, 1 deletions

diff --git a/pylint_plugins.py b/pylint_plugins.py
index bf357737c..6273d2c84 100644
--- a/pylint_plugins.py
+++ b/pylint_plugins.py
@@ -6,9 +6,11 @@ from __future__ import print_function
 
 import copy
 import sys
+import textwrap
 
-from astroid import MANAGER
+from astroid import MANAGER, register_module_extender
 from astroid import scoped_nodes
+from astroid.builder import AstroidBuilder
 
 
 def register(linter):
@@ -255,3 +257,19 @@ def fix_ipa_classes(cls):
         fake_class(cls, ipa_class_members[class_name_with_module])
 
 MANAGER.register_transform(scoped_nodes.Class, fix_ipa_classes)
+
+
+def ipaplatform_transform():
+    """Module aliases for IpaPlatformImporter
+    """
+    return AstroidBuilder(MANAGER).string_build(textwrap.dedent(
+        """
+        from ipaplatform.base import constants
+        from ipaplatform.base import paths
+        from ipaplatform.base import services
+        from ipaplatform.base import tasks
+        """
+    ))
+
+
+register_module_extender(MANAGER, 'ipaplatform', ipaplatform_transform)