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

psf · Mar 7, 2019 · 1580477 · 1580477
1 parent 21ab37a
commit 1580477
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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]
-    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():