Added Dump Import Directories command

Kronuz · Kronuz · commit 4f2cea1f4a25 · 2015-01-26T09:58:00.000-06:00
diff --git a/Default.sublime-commands b/Default.sublime-commands
@@ -33,5 +33,9 @@
         "caption": "SublimeCodeIntel: Disable Live Autocompletion for Current Language",
         "command": "sublimecodeintel_disable_live_lang",
         "args": {"action": "lang-off"}
+    },
+    {
+        "caption": "SublimeCodeIntel: Dump Import Directories",
+        "command": "sublimecodeintel_dump_import_dirs"
     }
 ]
diff --git a/SublimeCodeIntel.py b/SublimeCodeIntel.py
@@ -1638,6 +1638,45 @@ def run(self, edit, block=False):
                 window.open_file(previous_location, sublime.ENCODED_POSITION)
 
 
+class SublimecodeintelDumpImportDirs(sublime_plugin.WindowCommand):
+    def run(self):
+        settings_manager.update()
+        if not codeintel_enabled():
+            return
+
+        codeintel_database_dir = os.path.expanduser(settings_manager.get('codeintel_database_dir'))
+        db_dir = os.path.join(codeintel_database_dir, 'db')
+
+        stat_dir = os.path.join(codeintel_database_dir, 'import_dir_stats')
+        if not os.path.exists(stat_dir):
+            os.makedirs(stat_dir)
+
+        def get_immediate_subdirectories(a_dir):
+            for subdirname in os.listdir(a_dir):
+                dirname = os.path.join(a_dir, subdirname)
+                if os.path.isdir(dirname):
+                    yield (dirname, subdirname)
+
+        for lib_dir in [lib_dir for lib_dir in get_immediate_subdirectories(db_dir)]:
+            if lib_dir[1] != 'stdlibs':
+                lang = lib_dir[1]
+                import_dirs = []
+                hash_map = {}
+                for lib_dir_entry in get_immediate_subdirectories(lib_dir[0]):
+                    try:
+                        pathfile_path = os.path.join(lib_dir_entry[0], 'path')
+                        path_file = open(pathfile_path)
+                        import_dir = path_file.read()
+                        import_dirs.append(import_dir)
+                        hash_map[import_dir] = lib_dir_entry[1]
+                    except:
+                        pass
+
+                stat_file = open(os.path.join(stat_dir, lang), 'w')
+                for item in sorted(import_dirs, key=str.lower):
+                    stat_file.write("%s %s\n" % (hash_map[item], item))
+
+
 class CodeintelCommand(sublime_plugin.TextCommand):
     """command to interact with codeintel"""
 

Original file line number	Diff line number	Diff line change
`@@ -33,5 +33,9 @@`
`33`	`33`	`"caption": "SublimeCodeIntel: Disable Live Autocompletion for Current Language",`
`34`	`34`	`"command": "sublimecodeintel_disable_live_lang",`
`35`	`35`	`"args": {"action": "lang-off"}`
	`36`	`+ },`
	`37`	`+ {`
	`38`	`+ "caption": "SublimeCodeIntel: Dump Import Directories",`
	`39`	`+ "command": "sublimecodeintel_dump_import_dirs"`
`36`	`40`	`}`
`37`	`41`	`]`