[query] make local and remote tmp settable on backend

Author: ehigham

hail/python/hail/backend/backend.py

@@ -392,3 +392,13 @@ def get_flags(self, *flags) -> Mapping[str, str]:
     @abc.abstractmethod
     def requires_lowering(self):
         pass
+
+    @property
+    @abc.abstractmethod
+    def local_tmpdir(self) -> str:
+        pass
+
+    @property
+    @abc.abstractmethod
+    def remote_tmpdir(self) -> str:
+        pass

hail/python/hail/backend/py4j_backend.py

@@ -196,8 +196,8 @@ def decode_bytearray(encoded):
         self._jhc = jhc
 
         self._jbackend = self._hail_package.backend.api.Py4JBackendApi(jbackend)
-        self._jbackend.pySetLocalTmp(tmpdir)
-        self._jbackend.pySetRemoteTmp(remote_tmpdir)
+        self.local_tmpdir = tmpdir
+        self.remote_tmpdir = tmpdir
 
         self._jhttp_server = self._jbackend.pyHttpServer()
         self._backend_server_port: int = self._jhttp_server.port()
@@ -326,3 +326,21 @@ def stop(self):
         self._jhc = None
         uninstall_exception_handler()
         super().stop()
+
+    @property
+    def local_tmpdir(self) -> str:
+        return self._local_tmpdir
+
+    @local_tmpdir.setter
+    def local_tmpdir(self, tmpdir) -> str:
+        self._local_tmpdir = tmpdir
+        self._jbackend.pySetLocalTmp(tmpdir)
+
+    @property
+    def remote_tmpdir(self) -> str:
+        return self._remote_tmpdir
+
+    @remote_tmpdir.setter
+    def remote_tmpdir(self, tmpdir) -> str:
+        self._remote_tmpdir = tmpdir
+        self._jbackend.pySetRemoteTmp(tmpdir)

hail/python/hail/backend/service_backend.py

@@ -239,6 +239,7 @@ def __init__(
         self._batch_was_submitted: bool = False
         self.disable_progress_bar = disable_progress_bar
         self.batch_attributes = batch_attributes
+        self.local_tmpdir = tmp_dir()
         self.remote_tmpdir = remote_tmpdir
         self.flags: Dict[str, str] = {}
         self._registered_ir_function_names: Set[str] = set()