Replace module imp with importlib (#1738)

Azure · Apr 24, 2020 · 11d0881 · 11d0881
1 parent 62227d6
commit 11d0881
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/bin/waagent b/bin/waagent
@@ -24,9 +24,13 @@
 #
 
 import os
-import imp
 import sys
 
+if sys.version_info[0] == 2:
+    import imp
+else:
+    import importlib
+
 if __name__ == '__main__' :
     import azurelinuxagent.agent as agent
     """

diff --git a/bin/waagent2.0 b/bin/waagent2.0
@@ -1744,7 +1744,11 @@ bsd_activate_resource_disk_txt="""\
 
 import os
 import sys
-import imp
+
+if sys.version_info[0] == 2:
+    import imp
+else:
+    import importlib
 
 # waagent has no '.py' therefore create waagent module import manually.
 __name__='setupmain' #prevent waagent.__main__ from executing