Put cursor in last line if old position is invalid (#641)

diff --git a/plugin/black.vim b/plugin/black.vim
index 34f30c532f006d0ea0f0411a59170c60b3666f16..f0b55b2307fc84312394934b08dfadfe3c7c2a65 100644 (file)
--- a/plugin/black.vim
+++ b/plugin/black.vim
@@ -114,7 +114,10 @@ def Black():
   else:
     cursor = vim.current.window.cursor
     vim.current.buffer[:] = new_buffer_str.split('\n')[:-1]
   else:
     cursor = vim.current.window.cursor
     vim.current.buffer[:] = new_buffer_str.split('\n')[:-1]

-    vim.current.window.cursor = cursor
+    try:
+      vim.current.window.cursor = cursor
+    except vim.error:
+      vim.current.window.cursor = (len(vim.current.buffer), 0)

     print(f'Reformatted in {time.time() - start:.4f}s.')
 
 def BlackUpgrade():
     print(f'Reformatted in {time.time() - start:.4f}s.')
 
 def BlackUpgrade():