Don't insert imports before the shebang

Wilfred · Wilfred · commit ca4dd489fa04 · 2017-08-07T19:17:22.000+01:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -4,6 +4,8 @@
   wrong buffer where it found the import.
 * `pyimport-remove-unused` now works on all buffers, even if they're
   unsaved or not visiting a file.
+* `pyimport-insert-missing` now steps over shebangs when choosing
+  where to insert new imports.
 
 ## v1.0
 
diff --git a/pyimport.el b/pyimport.el
@@ -115,7 +115,17 @@ To terminate the loop early, throw 'break."
       ;; LINE as-is.
       (save-excursion
         (goto-char (point-min))
-        (insert line "\n")))))
+        (let ((insert-pos (point)))
+          (catch 'found
+            ;; Find the first non-comment non-blank line.
+            (dotimes (_ 30)
+              (forward-line 1)
+              (when (and (not (looking-at "\n"))
+                         (not (looking-at "#"))
+                         (not (looking-at "\"")))
+                (setq insert-pos (point))
+                (throw 'found nil))))
+          (insert line "\n"))))))
 
 (defun pyimport--get-alias (import-as symbol)
   "Return the original symbol name, the aliased name, or nil, if
