escoffier 0.1.0 → 0.1.1

data/Rakefile

@@ -10,6 +10,8 @@ begin
     gemspec.homepage = "http://github.com/kastman/escoffier"
     gemspec.authors = ["Erik Kastman"]
     gemspec.add_dependency('bzip2-ruby')
+    gemspec.add_dependency('snmp')
+    gemspec.add_dependency('term-ansicolor')
     gemspec.add_development_dependency('rspec')
   end
   Jeweler::GemcutterTasks.new

data/VERSION

@@ -1 +1 @@
-0.1.0
+0.1.1

data/bin/loads

@@ -0,0 +1,88 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'pp'
+require 'snmp'
+require 'term/ansicolor'
+
+def hosts
+  if ARGV.size >= 1
+    hosts = ARGV
+  else
+    # Key-value pairs are hostname and number of CPUs
+    hosts = {
+      :miho => 4, 
+      :pebbles => 4,
+      :schroeder => 8,
+      :nelson => 8,
+      :jimbo => 8,
+      :kearney => 8,
+      :stella => 16}
+  end
+end
+
+def gather_data_on(hosts)
+  load_data = {}
+
+  hosts.each do |host|
+    begin
+      if host.kind_of? String
+        hostname = host
+        cpus = 0
+      elsif host.kind_of? Array
+        hostname = host.first.to_s
+        cpus = host.last 
+      end
+      # mibs = ["SNMPv2-MIB", "UCD-SNMP-MIB"]
+      mibs = ["SNMPv2-MIB"]
+      # Since the UCD-SNMP MIB is included by default in the SNMP gem, and it 
+      # does not correctly import, and we don't need dynamic lookup anyway,
+      # just list the required OID's directly.
+      SNMP::Manager.open(:Host => hostname, :Community => 'brainmap', :MibModules => mibs, :retries => 1, :timeout => 1) do |manager|
+        name = manager.get("sysName.0").varbind_list.first.value.split(".").first.to_s
+        # 1-min:  UCD-SNMP-MIB::laLoad.1 => "1.3.6.1.4.1.2021.10.1.3.1"
+        # 5-min:  UCD-SNMP-MIB::laLoad.2 => "1.3.6.1.4.1.2021.10.1.3.2"
+        # 15-min: UCD-SNMP-MIB::laLoad.3 => "1.3.6.1.4.1.2021.10.1.3.3"
+        # load_data[name] = [manager.get(["UCD-SNMP-MIB::laLoad.1", "UCD-SNMP-MIB::laLoad.2", "UCD-SNMP-MIB::laLoad.3"]).each_varbind.collect{|vb| vb.value.to_s }, cpus]
+        load_data[name] = [manager.get(["1.3.6.1.4.1.2021.10.1.3.1", "1.3.6.1.4.1.2021.10.1.3.2", "1.3.6.1.4.1.2021.10.1.3.3"]).each_varbind.collect{|vb| vb.value.to_s }, cpus]
+        # response = manager.get(["sysName.0", "UCD-SNMP-MIB::laLoad.1", "UCD-SNMP-MIB::laLoad.2", "UCD-SNMP-MIB::laLoad.3"])
+        # pp response
+      end
+    rescue SNMP::RequestTimeout => e
+      puts "Warning: #{e}"
+    rescue SocketError
+      puts "Warning: Host #{hostname} not recognized. #{e}"
+    end
+  end
+  
+  display_data(load_data)
+end
+
+def display_data(load_data)
+  display = ""
+  display << "\nLoad Averages over three intervals:\n\n"
+  display << Term::ANSIColor.bold {"\t%-15s\t%-4s\t" % ["Server (CPUs)", "% Util"] }
+  display << Term::ANSIColor.bold {"%s\t%s\t%s\n" % ["1-minute", "5-minutes", "15-minutes"] }
+  display << "\t%-15s\t%-4s\t%s\t%s\t%s\n" % ["-"*13, "-"*6, "-" * 8, "-" * 8, "-" * 8]
+  load_data.sort_by{|la| la.last.first}.reverse.each do |average_hash|
+    cpus = average_hash.last.last
+    utilization = (average_hash.last.first[1].to_f / cpus) * 100
+    case utilization
+    when 0.0...50.0
+      color = "green"
+    when 50.0..100
+      color = "yellow"
+    else
+      color = "red"
+    end
+    display << eval("Term::ANSIColor.#{color}")
+    display << "\t%-15s\t" % "#{average_hash.first} (#{average_hash.last.last})"
+    display << "%-2.2f\t" % utilization
+    average_hash.last.first.collect {|la| display << "%s\t\t" % la }
+    display << "\n"
+  end
+  display << Term::ANSIColor.clear
+end
+
+
+puts gather_data_on hosts

data/bin/mise

@@ -11,7 +11,7 @@ require 'escoffier'
 def parse_options
   options = Hash.new
   parser = OptionParser.new do |opts|
-    opts.banner = "Usage: #{__FILE__} [options] source_paths destination"
+    opts.banner = "Usage: #{File.basename(__FILE__)} [options] source_paths destination"
 
     # opts.on('-o', '--output OUTPUT_DIRECTORY', "Output Destination Directory")     do |dir| 
     #   options[:output] = dir

data/bin/normalize_directory.rb

@@ -49,7 +49,7 @@ def normalize!(options)
   directories = ARGV.to_a
   directories.each do |directory|
     unless File.directory?(directory)
-      puts "Error: Cannot find directory #{dir}."
+      puts "Error: Cannot find directory #{directory}."
       next
     end
     normalizer = Normalizer.new(directory, options)
@@ -61,8 +61,8 @@ if File.basename(__FILE__) == File.basename($PROGRAM_NAME)
   $LOG = Logger.new(STDOUT)
   
   options = parse_options
-  if options[:spec]
-    directories = 
-  end
+  # if options[:spec]
+  #   directories = 
+  # end
   normalize!(options)
 end

data/escoffier.gemspec

@@ -5,14 +5,14 @@
 
 Gem::Specification.new do |s|
   s.name = %q{escoffier}
-  s.version = "0.1.0"
+  s.version = "0.1.1"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Erik Kastman"]
-  s.date = %q{2010-05-11}
+  s.date = %q{2010-09-20}
   s.description = %q{Administrative prep tasks, from mother sauces to hotel pans.}
   s.email = %q{ekk@medicine.wisc.edu}
-  s.executables = ["normalize_directory.rb", "mise~", "mise"]
+  s.executables = ["normalize_directory.rb", "mise~", "mise", "loads"]
   s.extra_rdoc_files = [
     "README.rdoc"
   ]
@@ -22,6 +22,7 @@ Gem::Specification.new do |s|
      "README.rdoc",
      "Rakefile",
      "VERSION",
+     "bin/loads",
      "bin/mise",
      "bin/normalize_directory.rb",
      "escoffier.gemspec",
@@ -55,13 +56,19 @@ Gem::Specification.new do |s|
 
     if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
       s.add_runtime_dependency(%q<bzip2-ruby>, [">= 0"])
+      s.add_runtime_dependency(%q<snmp>, [">= 0"])
+      s.add_runtime_dependency(%q<term-ansicolor>, [">= 0"])
       s.add_development_dependency(%q<rspec>, [">= 0"])
     else
       s.add_dependency(%q<bzip2-ruby>, [">= 0"])
+      s.add_dependency(%q<snmp>, [">= 0"])
+      s.add_dependency(%q<term-ansicolor>, [">= 0"])
       s.add_dependency(%q<rspec>, [">= 0"])
     end
   else
     s.add_dependency(%q<bzip2-ruby>, [">= 0"])
+    s.add_dependency(%q<snmp>, [">= 0"])
+    s.add_dependency(%q<term-ansicolor>, [">= 0"])
     s.add_dependency(%q<rspec>, [">= 0"])
   end
 end

data/lib/escoffier.rb

@@ -1,5 +1,13 @@
-$:.unshift(File.dirname(__FILE__)) unless
-  $:.include?(File.dirname(__FILE__)) || $:.include?(File.expand_path(File.dirname(__FILE__)))
+$LOAD_PATH.unshift(File.dirname(__FILE__)) unless
+  $LOAD_PATH.include?(File.dirname(__FILE__)) || $LOAD_PATH.include?(File.expand_path(File.dirname(__FILE__)))
+
+begin
+  require 'bzip2_ext'
+rescue LoadError
+  require 'rubygems'
+  gem 'bzip2-ruby'
+  require 'bzip2_ext'
+end
 
 require 'escoffier/core_additions'
 require 'escoffier/normalizer'

data/lib/escoffier/core_additions.rb

@@ -3,7 +3,7 @@ require 'escoffier/smepable'
 class Pathname
   include Smepable
   
-  def prep_mise_to(output_dir)
-    self.prep_mise(self.to_s, output_dir)
+  def prep_mise_to(output_dir, &block)
+    self.prep_mise(self.to_s, output_dir, &block)
   end
 end

data/lib/escoffier/normalizer.rb

@@ -5,7 +5,9 @@ require 'escoffier/compressible'
 
 class Normalizer
   COMPRESSION_REGEXES = /\.bz2$/
+  PFILE_REGEXES = [/^P\d{5}\.7$/]
   DICOM_REGEXES = [/\d{3,5}.dcm$/, /I?.\d{4}$/ ]
+  IMAGE_REGEXES = [PFILE_REGEXES, DICOM_REGEXES].flatten
   
   include Compressible
   attr_reader :entry_path
@@ -53,7 +55,7 @@ class Normalizer
   end
   
   def normalize_compression(file)
-     DICOM_REGEXES.each do |regex|
+     IMAGE_REGEXES.each do |regex|
        if file =~ regex
          file = zip(file) unless @dry_run
        end

data/lib/escoffier/smepable.rb

@@ -9,7 +9,7 @@ module Smepable
   $LOG ||= Logger.new(STDOUT)
   include Compressible
   
-  def prep_mise(input_entry, output_directory = Dir.mktmpdir)
+  def prep_mise(input_entry, output_directory = Dir.mktmpdir, &block)
     # destination_dirname = File.dirname(output_directory)
     FileUtils.mkdir_p(output_directory) unless File.exist?(output_directory)
     $LOG.info "Copying #{input_entry} to #{output_directory}"
@@ -19,7 +19,17 @@ module Smepable
     $LOG.info "Unzipping #{output_directory}"
     unzip(output_directory, :recursive => true)
     
-    return output_directory
+    if block
+      begin
+        yield output_directory
+      ensure
+        File.delete(output_directory)
+      end
+
+    else
+      return output_directory
+    end
+    
   end
   
 end

data/spec/smepable_spec.rb

@@ -1,6 +1,6 @@
 $:.unshift File.join(File.dirname(__FILE__),'..','lib')
 
-# require 'rubygems'
+require 'rubygems'
 require 'spec'
 require 'fileutils'
 require 'logger'

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 1
-  - 0
-  version: 0.1.0
+  - 1
+  version: 0.1.1
 platform: ruby
 authors: 
 - Erik Kastman
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-05-11 00:00:00 -05:00
+date: 2010-09-20 00:00:00 -05:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -30,7 +30,7 @@ dependencies:
   type: :runtime
   version_requirements: *id001
 - !ruby/object:Gem::Dependency 
-  name: rspec
+  name: snmp
   prerelease: false
   requirement: &id002 !ruby/object:Gem::Requirement 
     requirements: 
@@ -39,14 +39,39 @@ dependencies:
         segments: 
         - 0
         version: "0"
-  type: :development
+  type: :runtime
   version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: term-ansicolor
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id004
 description: Administrative prep tasks, from mother sauces to hotel pans.
 email: ekk@medicine.wisc.edu
 executables: 
 - normalize_directory.rb
 - mise~
 - mise
+- loads
 extensions: []
 
 extra_rdoc_files: 
@@ -57,6 +82,7 @@ files:
 - README.rdoc
 - Rakefile
 - VERSION
+- bin/loads
 - bin/mise
 - bin/normalize_directory.rb
 - escoffier.gemspec