Python в Sublime Text 3: oslist dir; список спецификаций c только расширения - PullRequest
Новая
       138

Python в Sublime Text 3: oslist dir; список спецификаций c только расширения

0 голосов
/ 07 августа 2020

Я пытаюсь внести небольшую поправку в этот плагин Sublime Text 3 , поэтому это мой первый набег на Python.

Плагин предоставляет список файлов Markdown, когда команда [[ вводится с клавиатуры. Дело в том, что в списке появляются другие файлы, которые я не хочу отображать!

Насколько я понял, основная часть команды требует os.listdir(directory). Я знаю, что при попытке перечислить только определенные c расширения с помощью os.listdir ранее запрашивались, например, здесь и здесь , но я просто не понимаю, как go о реализации этих ответов в коде, который у меня есть. Как и другие, я пытаюсь понять, как показывать файлы только .md с помощью os.listdir, но за исключением того, что я хочу сделать это в рамках этого плагина.

Полный код плагина доступен здесь .

Ниже приводится отрывок, который включает в себя то, что, как мне кажется, является наиболее важными деталями моей дилеммы. 

class GetWikiLinkCommand(sublime_plugin.TextCommand):
    def on_done(self, selection):
        if selection == -1:
            self.view.run_command(
                "insert_wiki_link", {"args":
                {'text': '[['}})
            return
        self.view.run_command(
            "insert_wiki_link", {"args":
            {'text': '[['+self.modified_files[selection]+']]'}})

    def run(self, edit):
        settings = sublime.load_settings('MyWiki.sublime-settings')
        directory = settings.get('wiki_directory')
        directory = os.path.expanduser(directory)
        extension = settings.get('wiki_extension')

        self.outputText = '[['
        self.files = os.listdir(directory) 
        self.modified_files = [item.replace(extension,"") for item in self.files]
        self.view.window().show_quick_panel(self.modified_files, self.on_done)

Спасибо !!!

1 Ответ

2 голосов
/ 07 августа 2020

Вероятно, самое простое решение - изменить эту строку: 

self.files = os.listdir(directory)

на 

self.files = [f for f in os.listdir(directory) if f.endswith('.md')]

В качестве альтернативы, если вы добавите glob и os.path в оператор импорта, вы мог бы сделать: 

self.files = [os.path.basename(p) for p in glob.glob(os.path.join(directory, '*.md'))]

, что может быть более производительным, если glob может фильтровать файлы быстрее - не уверен в этом.

...