Example #1
0
  def __init__( self, imageName, endPoint ):
    """
    Constructor: uses AmazonConfiguration to parse the endPoint CS configuration
    and ImageConfiguration to parse the imageName CS configuration. 
    
    :Parameters:
      **imageName** - `string`
        imageName as defined on CS:/Resources/VirtualMachines/Images
        
      **endPoint** - `string`
        endPoint as defined on CS:/Resources/VirtualMachines/CloudEndpoint 
    
    """
    # logger
    self.log       = gLogger.getSubLogger( 'AmazonImage %s: ' % imageName )
    
    self.imageName = imageName
    self.endPoint  = endPoint 
    
    # their config() method returns a dictionary with the parsed configuration
    # they also provide a validate() method to make sure it is correct 
    self.__imageConfig = ImageConfiguration( imageName, endPoint )    
    self.__amazonConfig  = AmazonConfiguration( endPoint )

    self.__cliamazon  = None
Example #2
0
class AmazonImage:
  """
  AmazonImage class.
  """

  def __init__( self, imageName, endPoint ):
    """
    Constructor: uses AmazonConfiguration to parse the endPoint CS configuration
    and ImageConfiguration to parse the imageName CS configuration. 
    
    :Parameters:
      **imageName** - `string`
        imageName as defined on CS:/Resources/VirtualMachines/Images
        
      **endPoint** - `string`
        endPoint as defined on CS:/Resources/VirtualMachines/CloudEndpoint 
    
    """
    # logger
    self.log       = gLogger.getSubLogger( 'AmazonImage %s: ' % imageName )
    
    self.imageName = imageName
    self.endPoint  = endPoint 
    
    # their config() method returns a dictionary with the parsed configuration
    # they also provide a validate() method to make sure it is correct 
    self.__imageConfig = ImageConfiguration( imageName, endPoint )    
    self.__amazonConfig  = AmazonConfiguration( endPoint )

    self.__cliamazon  = None

  def connectAmazon( self ):
    """
    Method that issues the connection with Amazon. 
    In order to do it, validates the CS configurations.
    auth secretaccesskey

    :return: S_OK | S_ERROR
    """

    # Before doing anything, make sure the configurations make sense
    # ImageConfiguration
    validImage = self.__imageConfig.validate()
    if not validImage[ 'OK' ]:
      return validImage
    # EndpointConfiguration
    validAmazon = self.__amazonConfig.validate()
    if not validAmazon[ 'OK' ]:
      return validAmazon

    # Get authentication configuration
    # not needed jet, only secretaccesskey method
    #if not auth == 'secretaccesskey':
    #  self.__errorStatus = "auth not supported (currently secretaccesskey)"
    #  self.log.error( self.__errorStatus )
    #  return
   
    # this object will connect to the Amazon server. Better keep it private.   
    self.__cliamazon = AmazonClient( endpointConfig=self.__amazonConfig.config(), imageConfig=self.__imageConfig.config() )

    # Check connection to the server
    result = self.__cliamazon.check_connection()
    if not result[ 'OK' ]:
      self.log.error( "connectAmazon" )
      self.log.error( result[ 'Message' ] )
    else:
      self.log.info( "Successful connection check" )
      
    return result
   
  def startNewInstance( self, runningPodRequirements ):
    """
    Once the connection is stablished using the `connectAmazon` method, we can boot
    nodes. To do so, the config in __imageConfig and __amazonConfig applied to
    AmazonClient initialization is applied.
    
    :return: S_OK | S_ERROR
    """
    
    self.log.info( "Booting %s / %s" % ( self.__imageConfig.config()[ 'bootImageName' ],
                                         self.__amazonConfig.config()[ 'endpointURL' ] ) )

    result = self.__cliamazon.create_VMInstance( runningPodRequirements )

    if not result[ 'OK' ]:
      self.log.error( "startNewInstance" )
      self.log.error( result[ 'Message' ] )
    return result

  def getInstanceStatus( self, uniqueId ):
    """
    Given the node ID and the current instanceName (AMI in Amazon terms), 
    returns the status. 
    
    :Parameters:
      **uniqueId** - `string`
        node ID, given by the OpenStack service       
    
    :return: S_OK | S_ERROR
    """
    
    result = self.__cliamazon.getStatus_VMInstance( uniqueId, self.imageName )
    
    if not result[ 'OK' ]:
      self.log.error( "getInstanceStatus: %s" % uniqueId )
      self.log.error( result[ 'Message' ] )
    
    return result  
    
  def stopInstance( self, uniqueId ):
    """
    Method that destroys the instance 
    
    :Parameters:
      **uniqueId** - `string`
    
    :return: S_OK | S_ERROR
    """

    result = self.__cliamazon.terminate_VMinstance( uniqueId )
    
    if not result[ 'OK' ]:
      self.log.error( "stopInstance: %s " % ( uniqueId ) )
      self.log.error( result[ 'Message' ] )
    
    return result