add suffix option to NamedTemporaryFile for filetype syntax highlighting/detection

nackjicholson · nackjicholson · commit 70a657eabb6b · 2018-10-15T08:41:25.000-07:00
diff --git a/editor.py b/editor.py
@@ -73,15 +73,15 @@ def get_tty_filename():
     return '/dev/tty'
 
 
-def edit(filename=None, contents=None, use_tty=None):
+def edit(filename=None, contents=None, use_tty=None, suffix=None):
     editor = get_editor()
     args = [editor] + get_editor_args(os.path.basename(os.path.realpath(editor)))
 
     if use_tty is None:
         use_tty = sys.stdin.isatty() and not sys.stdout.isatty()
 
     if filename is None:
-        tmp = tempfile.NamedTemporaryFile()
+        tmp = tempfile.NamedTemporaryFile(suffix=suffix)
         filename = tmp.name
 
     if contents is not None:
