bash autocomplete for OSF invoke tasks

.generate_invoke_completion.py

def main():
    f = open('tasks.py', 'r')
    tasks = []
    lines = f.read().split('\n')
    f.close()
    for i in range(len(lines)):
        line = lines[i]
        if '@task' in line:
            opt = lines[i + 1].replace('def ', '').split('(')[0]
            tasks.append(opt)
    templ = open('.invoke_completion_template', 'r').read()
    templ = templ.replace('__OPTS__', ' '.join(tasks))
    out = open('.invoke_completion.sh', 'wb')
    out.write(templ)
    out.close()

if __name__ == "__main__":
    main()

.invoke_completion_template

#!/bin/bash
_invoke ()
{       
  local cur prev opts
  cur="${COMP_WORDS[COMP_CWORD]}"
  prev="${COMP_WORDS[COMP_CWORD-1]}"
  opts="__OPTS__"
  
  COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
  return 0
}
complete -F _invoke invoke
complete -F _invoke inv

postactivate

#!/bin/bash
# This hook is sourced after this virtualenv is activated.
cd $OSF_HOME
python ./.generate_invoke_completion.py
source ./.invoke_completion.sh