allow a different source URL for python packages

[chef.git] / cookbooks / python / resources / package.rb

diff --git a/cookbooks/python/resources/package.rb b/cookbooks/python/resources/package.rb
index 76a6af7ec4a7aad7c64635b98f8ef5f81ab78df4..f794f2b5a35bd3420b45c7a5184f28309d56f0f4 100644 (file)
--- a/cookbooks/python/resources/package.rb
+++ b/cookbooks/python/resources/package.rb
@@ -1,14 +1,14 @@
 #
 #

-# Cookbook Name:: python
+# Cookbook:: python

 # Resource:: package
 #
 # Resource:: package
 #

-# Copyright 2017, OpenStreetMap Foundation
+# Copyright:: 2017, OpenStreetMap Foundation

 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
 #
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
 #

-# http://www.apache.org/licenses/LICENSE-2.0
+# https://www.apache.org/licenses/LICENSE-2.0

 #
 # Unless required by applicable law or agreed to in writing, software
 # distributed under the License is distributed on an "AS IS" BASIS,
 #
 # Unless required by applicable law or agreed to in writing, software
 # distributed under the License is distributed on an "AS IS" BASIS,


@@ -17,28 +17,63 @@
 # limitations under the License.
 #
 
 # limitations under the License.
 #
 

+unified_mode true
+

 default_action :install
 
 property :package_name, :kind_of => String, :name_property => true
 property :version, :kind_of => String
 default_action :install
 
 property :package_name, :kind_of => String, :name_property => true
 property :version, :kind_of => String

+property :python_version, :kind_of => String
+property :python_virtualenv, :kind_of => String
+property :extra_index_url, :kind_of => String

 
 action :install do
 
 action :install do

-  if version.nil?
-    execute "pip-install-#{name}" do
-      command "pip install #{new_resource.package_name}"
-      not_if "pip show #{new_resource.package_name}"
+  if new_resource.version.nil?
+    execute "pip-install-#{new_resource.package_name}" do
+      command "#{pip_command} install #{pip_extra_index} #{new_resource.package_name}"
+      not_if "#{pip_command} show #{new_resource.package_name}"
+    end
+  else
+    execute "pip-install-#{new_resource.package_name}" do
+      command "#{pip_command} install #{pip_extra_index} #{new_resource.package_name}==#{new_resource.version}"
+      not_if "#{pip_command} show #{new_resource.package_name} | fgrep -q #{new_resource.version}"
+    end
+  end
+end
+
+action :upgrade do
+  if new_resource.version.nil?
+    execute "pip-upgrade-#{new_resource.package_name}" do
+      command "#{pip_command} install #{pip_extra_index} --upgrade #{new_resource.package_name}"
+      only_if "#{pip_command} list --outdated | fgrep -q #{new_resource.package_name}"

     end
   else
     end
   else

-    execute "pip-install-#{name}" do
-      command "pip install #{new_resource.package_name}==#{new_resource.version}"
-      not_if "pip show #{new_resource.package_name} | fgrep -q #{new_resource.version}"
+    execute "pip-upgrade-#{new_resource.package_name}" do
+      command "#{pip_command} install #{pip_extra_index} --upgrade #{new_resource.package_name}==#{new_resource.version}"
+      not_if "#{pip_command} show #{new_resource.package_name} | fgrep -q #{new_resource.version}"

     end
   end
 end
 
 action :remove do
     end
   end
 end
 
 action :remove do

-  execute "pip-uninstall-#{name}" do
-    command "pip uninstall #{new_resource.package_name}"
-    only_if "pip show #{new_resource.package_name}"
+  execute "pip-uninstall-#{new_resource.package_name}" do
+    command "#{pip_command} uninstall #{new_resource.package_name}"
+    only_if "#{pip_command} show #{new_resource.package_name}"
+  end
+end
+
+action_class do
+  def pip_extra_index
+    if new_resource.extra_index_url
+      "--extra-index-url '#{new_resource.extra_index_url}'"
+    end
+  end
+
+  def pip_command
+    if new_resource.python_virtualenv
+      "#{new_resource.python_virtualenv}/bin/pip"
+    else
+      "pip#{new_resource.python_version}"
+    end

   end
 end
   end
 end