Merge PR #31835 into master

* refs/pull/31835/head:
	mgr/ssh: add TemporaryDirectory impl for py2 compat

Reviewed-by: Michael Fritch <mfritch@suse.com>
Reviewed-by: Sebastian Wagner <swagner@suse.com>

src/pybind/mgr/ssh/module.py

@@ -9,6 +9,7 @@ import os
 import random
 import tempfile
 import multiprocessing.pool
+import shutil
 import subprocess
 
 from ceph.deployment import inventory
@@ -37,6 +38,27 @@ DEFAULT_SSH_CONFIG = ('Host *\n'
                       'StrictHostKeyChecking no\n'
                       'UserKnownHostsFile /dev/null\n')
 
+# for py2 compat
+try:
+    from tempfile import TemporaryDirectory # py3
+except ImportError:
+    # define a minimal (but sufficient) equivalent for <= py 3.2
+    class TemporaryDirectory(object): # type: ignore
+        def __init__(self):
+            self.name = tempfile.mkdtemp()
+
+        def __enter__(self):
+            if not self.name:
+                self.name = tempfile.mkdtemp()
+            return self.name
+
+        def cleanup(self):
+            shutil.rmtree(self.name)
+
+        def __exit__(self, exc_type, exc_value, traceback):
+            self.cleanup()
+
+
 # high-level TODO:
 #  - bring over some of the protections from ceph-deploy that guard against
 #    multiple bootstrapping / initialization
@@ -355,7 +377,7 @@ class SSHOrchestrator(MgrModule, orchestrator.OrchestratorClientMixin):
     def _generate_key(self):
         if not self.ssh_pub or not self.ssh_key:
             self.log.info('Generating ssh key...')
-            tmp_dir = tempfile.TemporaryDirectory()
+            tmp_dir = TemporaryDirectory()
             path = tmp_dir.name + '/key'
             try:
                 subprocess.call([