Merge pull request voxpupuli#148 from esendex/master

shivapoudel · shivapoudel · commit 35810f9ecb85 · 2014-11-15T00:58:03.000+05:45
Amended facter pupperversion check to cope with Enterprise string
diff --git a/lib/facter/pip_version.rb b/lib/facter/pip_version.rb
@@ -2,7 +2,15 @@
 # Works with pip loaded and without, pip installed using pip  and package installed
 require 'puppet'
 require 'rubygems'
-if Gem::Version.new(Facter.value(:puppetversion)) >= Gem::Version.new('3.6')
+
+facter_puppet_version = Facter.value(:puppetversion)
+facter_is_pe = Facter.value(:is_pe)
+
+if facter_is_pe
+    facter_puppet_version = facter_puppet_version.to_s.split(' ')[0]
+end
+
+if Gem::Version.new(facter_puppet_version) >= Gem::Version.new('3.6')
   pkg = Puppet::Type.type(:package).new(:name => 'python-pip', :allow_virtual => 'false')
 else
   pkg = Puppet::Type.type(:package).new(:name => 'python-pip')
diff --git a/lib/facter/python_version.rb b/lib/facter/python_version.rb
@@ -2,7 +2,15 @@
 # In lists default python and system python versions
 require 'puppet'
 require 'rubygems'
-if Gem::Version.new(Facter.value(:puppetversion)) >= Gem::Version.new('3.6')
+
+facter_puppet_version = Facter.value(:puppetversion)
+facter_is_pe = Facter.value(:is_pe)
+
+if facter_is_pe
+    facter_puppet_version = facter_puppet_version.to_s.split(' ')[0]
+end
+
+if Gem::Version.new(facter_puppet_version) >= Gem::Version.new('3.6')
   pkg = Puppet::Type.type(:package).new(:name => 'python', :allow_virtual => 'false')
 else
   pkg = Puppet::Type.type(:package).new(:name => 'python')
diff --git a/lib/facter/virtualenv_version.rb b/lib/facter/virtualenv_version.rb
@@ -2,7 +2,15 @@
 # Works with virualenv loaded and without, pip installed and package installed
 require 'puppet'
 require 'rubygems'
-if Gem::Version.new(Facter.value(:puppetversion)) >= Gem::Version.new('3.6')
+
+facter_puppet_version = Facter.value(:puppetversion)
+facter_is_pe = Facter.value(:is_pe)
+
+if facter_is_pe
+    facter_puppet_version = facter_puppet_version.to_s.split(' ')[0]
+end
+
+if Gem::Version.new(facter_puppet_version) >= Gem::Version.new('3.6')
   pkg = Puppet::Type.type(:package).new(:name => 'virtualenv', :allow_virtual => 'false')
 else
   pkg = Puppet::Type.type(:package).new(:name => 'virtualenv')
