Allow for other python language packages.

bh · Oct 14, 2017 · 147b479 · 147b479
1 parent 3015bf4
commit 147b479
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/lib/python-isort.coffee b/lib/python-isort.coffee
@@ -6,10 +6,9 @@ module.exports =
 class PythonIsort
 
   checkForPythonContext: ->
-    editor = atom.workspace.getActiveTextEditor()
-    if not editor?
-      return false
-    return editor.getGrammar().scopeName == 'source.python'
+    grammar = atom.workspace.getActiveTextEditor()?.getGrammar?()
+    if grammar? and ~grammar.name.indexOf('Python')
+        return true
 
   removeStatusbarItem: =>
     @statusBarTile?.destroy()