diff options
Diffstat (limited to '')
| -rw-r--r-- | .vim/pydiction/pydiction.py | 263 |
1 files changed, 263 insertions, 0 deletions
diff --git a/.vim/pydiction/pydiction.py b/.vim/pydiction/pydiction.py new file mode 100644 index 0000000..ee6a622 --- /dev/null +++ b/.vim/pydiction/pydiction.py @@ -0,0 +1,263 @@ +#!/usr/bin/env python +# Last modified: July 20th, 2009 +""" + +pydiction.py 1.1 by Ryan Kulla (rkulla AT gmail DOT com). + +Description: Creates a Vim dictionary of Python module attributes for Vim's + completion feature. The created dictionary file is used by + the Vim ftplugin "python_pydiction.vim". + +Usage: pydiction.py <module> ... [-v] +Example: The following will append all the "time" and "math" modules' + attributes to a file, in the current directory, called "pydiction" + with and without the "time." and "math." prefix: + $ python pydiction.py time math + To print the output just to stdout, instead of appending to the file, + supply the -v option: + $ python pydiction.py -v time math + +License: BSD. +""" + + +__author__ = "Ryan Kulla (rkulla AT gmail DOT com)" +__version__ = "1.1" +__copyright__ = "Copyright (c) 2003-2009 Ryan Kulla" + + +import os +import sys +import types +import shutil + + +# Path/filename of the vim dictionary file to write to: +PYDICTION_DICT = r'complete-dict' +# Path/filename of the vim dictionary backup file: +PYDICTION_DICT_BACKUP = r'complete-dict.last' + +# Sentintal to test if we should only output to stdout: +STDOUT_ONLY = False + + +def get_submodules(module_name, submodules): + """Build a list of all the submodules of modules.""" + + # Try to import a given module, so we can dir() it: + try: + imported_module = my_import(module_name) + except ImportError, err: + return submodules + + mod_attrs = dir(imported_module) + + for mod_attr in mod_attrs: + if type(getattr(imported_module, mod_attr)) is types.ModuleType: + submodules.append(module_name + '.' + mod_attr) + + return submodules + + +def write_dictionary(module_name): + """Write to module attributes to the vim dictionary file.""" + + try: + imported_module = my_import(module_name) + except ImportError, err: + return + + mod_attrs = dir(imported_module) + + # Generate fully-qualified module names: + write_to.write('\n--- %(x)s module with "%(x)s." prefix ---\n' % + {'x': module_name}) + for mod_attr in mod_attrs: + if callable(getattr(imported_module, mod_attr)): + # If an attribute is callable, show an opening parentheses: + prefix_on = '%s.%s(' + else: + prefix_on = '%s.%s' + write_to.write(prefix_on % (module_name, mod_attr) + '\n') + + # Generate non-fully-qualified module names: + write_to.write('\n--- %(x)s module without "%(x)s." prefix ---\n' % + {'x': module_name}) + for mod_attr in mod_attrs: + if callable(getattr(imported_module, mod_attr)): + prefix_off = '%s(' + else: + prefix_off = '%s' + write_to.write(prefix_off % mod_attr + '\n') + + +def my_import(name): + """Make __import__ import "package.module" formatted names.""" + mod = __import__(name) + components = name.split('.') + for comp in components[1:]: + mod = getattr(mod, comp) + return mod + + +def remove_duplicates(seq, keep=()): + """ + + Remove duplicates from a sequence while perserving order. + + The optional tuple argument "keep" can be given to specificy + each string you don't want to be removed as a duplicate. + """ + seq2 = [] + seen = set(); + for i in seq: + if i in (keep): + seq2.append(i) + continue + elif i not in seen: + seq2.append(i) + seen.add(i) + return seq2 + + +def get_yesno(msg="[Y/n]?"): + """ + + Returns True if user inputs 'n', 'Y', "yes", "Yes"... + Returns False if user inputs 'n', 'N', "no", "No"... + If they enter an invalid option it tells them so and asks again. + Hitting Enter is equivalent to answering Yes. + Takes an optional message to display, defaults to "[Y/n]?". + + """ + while True: + answer = raw_input(msg) + if answer == '': + return True + elif len(answer): + answer = answer.lower()[0] + if answer == 'y': + return True + break + elif answer == 'n': + return False + break + else: + print "Invalid option. Please try again." + continue + + +def main(write_to): + """Generate a dictionary for Vim of python module attributes.""" + submodules = [] + + for module_name in sys.argv[1:]: + try: + imported_module = my_import(module_name) + except ImportError, err: + print "Couldn't import: %s. %s" % (module_name, err) + sys.argv.remove(module_name) + + cli_modules = sys.argv[1:] + + # Step through each command line argument: + for module_name in cli_modules: + print "Trying module: %s" % module_name + submodules = get_submodules(module_name, submodules) + + # Step through the current module's submodules: + for submodule_name in submodules: + submodules = get_submodules(submodule_name, submodules) + + # Add the top-level modules to the list too: + for module_name in cli_modules: + submodules.append(module_name) + + submodules.sort() + + # Step through all of the modules and submodules to create the dict file: + for submodule_name in submodules: + write_dictionary(submodule_name) + + if STDOUT_ONLY: + return + + # Close and Reopen the file for reading and remove all duplicate lines: + write_to.close() + print "Removing duplicates..." + f = open(PYDICTION_DICT, 'r') + file_lines = f.readlines() + file_lines = remove_duplicates(file_lines, ('\n')) + f.close() + + # Delete the original file: + os.unlink(PYDICTION_DICT) + + # Recreate the file, this time it won't have any duplicates lines: + f = open(PYDICTION_DICT, 'w') + for attr in file_lines: + f.write(attr) + f.close() + print "Done." + + +if __name__ == '__main__': + """Process the command line.""" + + if sys.version_info[0:2] < (2, 3): + sys.exit("You need a Python 2.x version of at least Python 2.3") + + if len(sys.argv) <= 1: + sys.exit("%s requires at least one argument. None given." % + sys.argv[0]) + + if '-v' in sys.argv: + write_to = sys.stdout + sys.argv.remove('-v') + STDOUT_ONLY = True + elif os.path.exists(PYDICTION_DICT): + # See if any of the given modules have already been pydiction'd: + f = open(PYDICTION_DICT, 'r') + file_lines = f.readlines() + for module_name in sys.argv[1:]: + for line in file_lines: + if line.find('--- %s module with' % module_name) != -1: + print '"%s" already exists in %s. Skipping...' % \ + (module_name, PYDICTION_DICT) + sys.argv.remove(module_name) + break + f.close() + + if len(sys.argv) < 2: + # Check if there's still enough command-line arguments: + sys.exit("Nothing new to do. Aborting.") + + if os.path.exists(PYDICTION_DICT_BACKUP): + answer = get_yesno('Overwrite existing backup "%s" [Y/n]? ' % \ + PYDICTION_DICT_BACKUP) + if (answer): + print "Backing up old dictionary to: %s" % \ + PYDICTION_DICT_BACKUP + try: + shutil.copyfile(PYDICTION_DICT, PYDICTION_DICT_BACKUP) + except IOError, err: + print "Couldn't back up %s. %s" % (PYDICTION_DICT, err) + else: + print "Skipping backup..." + + print 'Appending to: "%s"' % PYDICTION_DICT + else: + print "Backing up current %s to %s" % \ + (PYDICTION_DICT, PYDICTION_DICT_BACKUP) + try: + shutil.copyfile(PYDICTION_DICT, PYDICTION_DICT_BACKUP) + except IOError, err: + print "Couldn't back up %s. %s" % (PYDICTION_DICT, err) + else: + print 'Creating file: "%s"' % PYDICTION_DICT + + + if not STDOUT_ONLY: + write_to = open(PYDICTION_DICT, 'a') + + main(write_to) |