#compdef python

_python () {
  local -a reply
  local ret=1

  if [ $CURRENT -eq 2 ]; then
    _files && ret=0
  else
    reply=( $(auto_argcomplete ${words[2]}) )
    _describe -t reply "options" reply && ret=0
  fi
}
compdef _python python
