Merge pull request #2 from shK3Bq4d/issue1

Preserve current text input when opening the editor

1 files changed, 2 insertions, 1 deletions

diff --git a/edit.py b/edit.py
index a44af9c..f48e0f3 100644
--- a/edit.py
+++ b/edit.py
@@ -21,7 +21,8 @@ def edit(data, buf, args):
     editor = (weechat.config_get_plugin("editor") or
               os.environ.get("EDITOR", "vim -f"))
     path = os.path.expanduser("~/.weechat/message.txt")
-    open(path, "w+")
+    with open(path, "w+") as f:
+        f.write(weechat.buffer_get_string(buf, "input"))
 
     cmd = editor.split() + [path]
     code = subprocess.Popen(cmd).wait()