Igor provides the parts you need to build your Frankenprogram.
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Failed to load latest commit information.



mr.igor is an extension to pyflakes that will learn where you import things from, and then automatically fill in missing imports from the place they are most often imported.

But how does it know?

"It'th a knack."

Usage: igor [--print | --reap] filename

This script will record all imports from filename in Igor's database, and then add imports at the top of the file for any names that were not imported but were found in the database.

If the --print option is specified then the rewritten file will be written to stdout. (This allows the use of igor as a filter for editors.) Otherwise the file will be modified inplace.

If the --reap option is specified then imports will be added to the database from the specified file, but nothing will be written to stdout and the file will not be modified.

Only "from x import y" style imports are tracked and inserted. Aliases ("from x import y as z") are not supported.

mr.igor stores its database in the user's home directory. The base filename is ~/.mr.igor, but note that the Python shelve module may use a different physical filename, e.g. with a .db suffix.

Usage with TextMate

Go to the TextMate Bundle Editor and add a new command with the following settings:

Current File
igor --print $TM_FILEPATH
Replace Document
Key Equivalent: ⌘I
Scope Selector

Now you can save the current file and run it through Igor using the ⌘I keyboard shortcut.

Usage with Vim

The following key mapping will configure ⌘I to run the current file through Igor and then reload the current buffer:

nmap <D-i> :!igor %<CR> <bar> :e!<CR>

Usage with Emacs

Add the following to your .emacs. The igor function runs Igor over a temporary file copy of the current buffer. The current buffer is marked as modified only if Igor adds new imports.

(defun igor ()
  "Run the current buffer through mr.igor."
  (let ((igor-exe (or (executable-find "igor")
                      (error "No command 'igor' found")))
        (tempfile (make-temp-file "igor"))
        (buffer (current-buffer))
        (lines-before (count-lines 1 (buffer-size))))
    (with-temp-file tempfile
      (insert-buffer-substring buffer))
      (shell-command (concat igor-exe " --print " tempfile) t)
      (if (zerop (compare-buffer-substrings
                  (current-buffer) 1 (buffer-size)
                  buffer 1 (buffer-size buffer)))
          (message "igor: no new imports")
        (copy-to-buffer buffer 1 (buffer-size))
        (let ((lines-after (count-lines 1 (buffer-size))))
          (message "igor: added %d imports" (- lines-after lines-before))))
      (delete-file tempfile))))

(add-hook 'python-mode-hook '(lambda () (local-set-key (kbd "C-c C-i") 'igor)))