Fixed some bugs in write_tmpfile.

2011-10-08 18:10:10 +00:00 · 2011-10-08 18:10:10 +00:00 · 44553732ca
commit 44553732ca
parent 6cd76a205d
1 changed files with 5 additions and 2 deletions
--- a/cli/util.py
+++ b/cli/util.py
@ -1,3 +1,6 @@
+import os
+import tempfile
+
 def make_result_dict(cursor, row):
    d = {}
    for i in xrange(len(row)):
@ -82,9 +85,9 @@ def run_editor(filename):
    else:
        exit("Error: %s: File does not exist!" % filename)

-def write_tmpfile(pfix='', content=''):
+def write_tmpfile(pfix, content, encoding='utf8'):
    file = tempfile.NamedTemporaryFile(prefix=pfix+'-', dir='/tmp', delete=False)
-    file.write(content.encode(file_encoding))
+    file.write(content.encode(encoding))
    name = file.name
    file.close()
    return name