Merge pull request #32 from SublimeLinter/tmpdirs

Use actual filename for the temporary files

Author: kaste

linter.py

@@ -1,3 +1,8 @@
+from contextlib import contextmanager
+import os
+import tempfile
+import time
+
 from SublimeLinter.lint import PythonLinter
 from SublimeLinter.lint.linter import TransientError
 
@@ -22,6 +27,62 @@ class Pydocstyle(PythonLinter):
         '--ignore-decorators=': ''
     }
 
+    def tmpfile(self, cmd, code, suffix=None):
+        filename = (
+            self.context.get('file_name')
+            or "{}{}".format(
+                self.context['canonical_filename'][1:-1],
+                suffix or self.get_tempfile_suffix()
+            )
+        )
+        with self._make_temp_file(filename, code) as temp_filename:
+            self.context['file_on_disk'] = self.filename
+            self.context['temp_file'] = temp_filename
+            cmd = self.finalize_cmd(
+                cmd, self.context, at_value=temp_filename, auto_append=True)
+            return self._communicate(cmd)
+
+    @contextmanager
+    def _make_temp_file(self, filename, code):
+        folder_prefix = "{}-".format(self.plugin_name)
+        with tempfile.TemporaryDirectory(prefix=folder_prefix) as tmp_dir_name:
+            temp_filename = os.path.join(tmp_dir_name, filename)
+            with open(temp_filename, "w+b") as file:
+                file.write(bytes(code, "utf-8"))
+
+            try:
+                yield temp_filename
+            finally:
+                self._retry(
+                    3,
+                    "removing '{}'".format(temp_filename),
+                    lambda: os.unlink(temp_filename)
+                )
+
+    def _retry(self, times, tag, fn):
+        for i in range(times):
+            try:
+                fn()
+            except Exception:
+                if i < times - 1:
+                    self.logger.info(
+                        "{} failed: "
+                        "will retry after sleeping for {} second(s)"
+                        .format(tag, 2**i)
+                    )
+                    time.sleep(2**i)
+                else:
+                    self.logger.warning("{} failed".format(tag))
+                    raise
+            else:
+                if i > 0:
+                    self.logger.info("{} succeeded after {} retry".format(tag, i))
+                break
+
+    @property
+    def plugin_name(self):
+        return self.__class__.__module__.split(".", 1)[0]
+
     def on_stderr(self, stderr):
         # For a doc style tester, parse errors can be treated 'transient',
         # for the benefit, that we do not re-draw, but keep the errors from