Merge pull request voxpupuli#6 from zoni/sharedrequirements

stankevich · stankevich · commit 34f562aa3ea8 · 2013-03-14T04:43:42.000-07:00
Allow virtualenv definitions to share the same requirements file
diff --git a/manifests/config.pp b/manifests/config.pp
@@ -5,7 +5,6 @@
   Class['python::install'] -> Python::Virtualenv <| |>
 
   Python::Virtualenv <| |> -> Python::Pip <| |>
-  Python::Virtualenv <| |> -> Python::Requirements <| |>
 
   if $python::gunicorn {
     Class['python::install'] -> Python::Gunicorn <| |>
diff --git a/manifests/requirements.pp b/manifests/requirements.pp
@@ -4,6 +4,9 @@
 #
 # === Parameters
 #
+# [*requirements*]
+#  Path to the requirements file. Defaults to the resource name
+#
 # [*virtualenv*]
 #  virtualenv to run pip in. Default: system-wide
 #
@@ -23,12 +26,11 @@
 # Ashley Penney
 #
 define python::requirements (
-  $virtualenv = 'system',
-  $proxy      = false
+  $requirements = $name,
+  $virtualenv   = 'system',
+  $proxy        = false
 ) {
 
-  $requirements = $name
-
   $pip_env = $virtualenv ? {
     'system' => '`which pip`',
     default  => "${virtualenv}/bin/pip",
@@ -41,13 +43,17 @@
 
   $req_crc = "${requirements}.sha1"
 
-  file { $requirements:
-    ensure  => present,
-    mode    => '0644',
-    owner   => 'root',
-    group   => 'root',
-    replace => false,
-    content => '# Puppet will install and/or update pip packages listed here',
+   # This will ensure multiple python::virtualenv definitions can share the
+   # the same requirements file.
+   if !defined(File[$requirements]) {
+	file { $requirements:
+      ensure  => present,
+      mode    => '0644',
+      owner   => 'root',
+      group   => 'root',
+      replace => false,
+      content => '# Puppet will install and/or update pip packages listed here',
+    }
   }
 
   # SHA1 checksum to detect changes
diff --git a/manifests/virtualenv.pp b/manifests/virtualenv.pp
@@ -76,12 +76,18 @@
     }
 
     if $requirements {
-      Exec["python_virtualenv_${venv_dir}"]
-      -> Python::Requirements[$requirements]
+      exec { "python_requirements_initial_install_${requirements}_${venv_dir}":
+        command     => "${venv_dir}/bin/pip install ${proxy_flag} --requirement ${requirements}",
+        refreshonly => true,
+        timeout     => 1800,
+        subscribe   => Exec["python_virtualenv_${venv_dir}"],
+      }
 
-      python::requirements { $requirements:
-        virtualenv => $venv_dir,
-        proxy      => $proxy,
+      python::requirements { "${requirements}_${venv_dir}":
+        requirements => $requirements,
+        virtualenv   => $venv_dir,
+        proxy        => $proxy,
+        require      => Exec["python_virtualenv_${venv_dir}"],
       }
     }
 
