diff options
-rw-r--r-- | yoyo/scripts/newmigration.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/yoyo/scripts/newmigration.py b/yoyo/scripts/newmigration.py index 40ca69d..fdc8ea6 100644 --- a/yoyo/scripts/newmigration.py +++ b/yoyo/scripts/newmigration.py @@ -154,12 +154,14 @@ def make_filename(config, directory, message): def create_with_editor(config, directory, migration_source): editor = utils.get_editor(config) - tmpfile = NamedTemporaryFile(dir=directory, + tmpfile = NamedTemporaryFile(mode='w', + dir=directory, prefix=tempfile_prefix, suffix='.py', + encoding='UTF-8', delete=False) try: - with io.open(tmpfile.name, 'w', encoding='UTF-8') as f: + with tmpfile as f: f.write(migration_source) editor = [part.format(tmpfile.name) for part in shlex.split(editor)] |