machine.rb in rancher-api-0.5.0

- old
+ new

@@ -1,21 +1,32 @@
 require 'rancher/api/models/machine/driver_config'
 
 module Rancher
   module Api
     class Machine
-      DIGITAL_OCEAN = 'digitalocean'
-      VMWARE_VSPHERE = 'vmwarevsphere'
-
       include Her::Model
+      include Helpers::Model
 
+      DIGITAL_OCEAN = 'digitalocean'.freeze
+      VMWARE_VSPHERE = 'vmwarevsphere'.freeze
+
       attributes :name, :state, :amazonec2Config, :azureConfig, :description,
                  :digitaloceanConfig, :driver, :exoscaleConfig, :externalId,
                  :labels, :openstackConfig, :packetConfig, :rackspaceConfig,
                  :removed, :softlayerConfig, :virtualboxConfig,
                  :vmwarevcloudairConfig, :vmwarevsphereConfig
 
       has_many :hosts
+
+      class << self
+        def transitioning
+          all.select { |m| m.transitioning.eql?('yes') }
+        end
+
+        def active
+          where(state: 'active')
+        end
+      end
 
       def driver_config
         case driver
         when DIGITAL_OCEAN, VMWARE_VSPHERE
           DriverConfig.new(attributes["#{driver}Config"])