{% from "python_env/map.jinja" import python_env with context %}

{% if grains['travis'] == True and grains['kernel'] == 'Linux' %}
  {% if "python2" in grains['pip'] %}
    {% python_env.update({ 'pip_bin_env': '/opt/python/3.5.2/bin/pip3'}) %}
  {% else %}
    {% python_env.update({'pip_bin_env': grains['pip']}) %}
  {% endif %}
{% endif %}

pip-install:
  pkg.installed:
    - name: {{ python_env.pip }}

virtualenv-install:
  pip.installed:
    - name: {{ python_env.virtualenv }}
    - user: {{ grains.user }}
    - bin_env: {{ python_env.pip_bin_env }}
    - require: 
      - pkg: pip-install

virtualenvwrapper-install:
  pip.installed:
    - name: {{ python_env.virtualenvwrapper }}
    - user: {{ grains.user }}
    - bin_env: {{ python_env.pip_bin_env }}
    - require: 
      - pip: virtualenv-install

zsh-virtualenvwrapper:
  file.append:
    - name: {{ grains.homedir }}/.zshrc
    - text:
      - "\n"
      - "# Adding virtualenvwrapper python binary path"
      - "export VIRTUALENVWRAPPER_PYTHON={{ python_env.python_path }}"
      - ""
      - "# Sourcing virtualenv"
      - "source {{ grains.homedir }}/.local/bin/virtualenvwrapper.sh"