linux-tools/modules/vim/module.py

from IModule import IModule
import os.path


class VimModule(IModule):
    def __init__(self):
        super().__init__()

    def get_command(self):

        plugins = [
            ("1. Colorschemes", "Plug 'https://github.com/rafi/awesome-vim-colorschemes.git'"),
            ("2. Unix Operations", "Plug 'https://github.com/tpope/vim-eunuch.git'"),
            ("3. Emmet", "Plug 'https://github.com/mattn/emmet-vim.git'"),
            ("4. Meta5 Theme", "Plug 'https://github.com/christophermca/meta5.git'"),
            ("5. Nerdtree", "Plug 'https://github.com/scrooloose/nerdtree'"),
            ("6. Lightline", "Plug 'https://github.com/itchyny/lightline.vim'")
        ]

        print("Available Plugins:\n")

        for key in plugins:
            print(key[0])

        print("")

        # Select Plugins
        user_input = [int(userInput) for userInput in input("Select modules: ").split()]
        clean_plugin_list = set(user_input)

        # print out selected plugins
        for element in clean_plugin_list:
            selection = element - 1
            try:
                print(plugins[selection][0])
            except Exception:
                print("Fail")

        # Create vimrc file
        vimrc_exists = os.path.isfile("testfile")

        if vimrc_exists:
            vimrc_file = open("testfile", "a+")
            vimrc_file.write("\n" + plugins[selection][1])
            vimrc_file.close()

        else:
            vimrc_file = open("testfile", "w+")
            vimrc_file.write("\n" + "call plug#begin('~/.vim/plugged')\n")
            for element in clean_plugin_list:
                selection = element -1
                vimrc_file.write(plugins[selection][1] + "\n")
            vimrc_file.write("call plug#end()\n")
            vimrc_file.close()