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
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