def authenticate_clients(): """ function to instantiate Clients (astakos, cyclades, compute) """ try: astakos_client = AstakosClient(AUTHENTICATION_URL, TOKEN) astakos_client.authenticate() logging.info('Successful authentication') except ClientError: logging.info('\n Failed to authenticate user token') print 'Failed to authenticate user token' try: endpoints = astakos_client.get_endpoints() cyclades_base_url = parse_astakos_endpoints(endpoints, 'cyclades_compute') cyclades_network_base_url = parse_astakos_endpoints(endpoints, 'cyclades_network') except ClientError: print('Failed to get endpoints for cyclades') try: cyclades_client = CycladesClient(cyclades_base_url, TOKEN) compute_client = ComputeClient(cyclades_base_url, TOKEN) network_client = CycladesNetworkClient(cyclades_network_base_url, TOKEN) return cyclades_client, compute_client, network_client, astakos_client except ClientError: print 'Failed to initialize Cyclades client'
def check_credentials(token, auth_url=auth_url): """Identity,Account/Astakos. Test authentication credentials""" logging.log(REPORT, ' Test the credentials') try: auth = AstakosClient(auth_url, token) auth.authenticate() except ClientError: msg = ' Authentication failed with url %s and token %s'\ % (auth_url, token) raise ClientError(msg, error_authentication) return auth
def check_credentials(auth_url, token): print(' Test the credentials') try: auth = AstakosClient(auth_url, token) auth.authenticate() except ClientError: print('Authentication failed with url %s and token %s' % ( auth_url, token)) raise print 'Authentication verified' return auth
def check_user_credentials(token, auth_url='https://accounts.okeanos.grnet.gr' '/identity/v2.0'): '''Identity,Account/Astakos. Test ~okeanos authentication credentials''' logging.info(' Test the credentials') try: auth = AstakosClient(auth_url, token) auth.authenticate() logging.info(' Authentication verified') return AUTHENTICATED except ClientError: logging.error('Authentication failed with url %s and token %s' % ( auth_url, token)) return NOT_AUTHENTICATED
def check_user_credentials(token, auth_url='https://accounts.okeanos.grnet.gr' '/identity/v2.0'): """Identity,Account/Astakos. Test ~okeanos authentication credentials""" logging.info(' Test the credentials') try: auth = AstakosClient(auth_url, token) auth.authenticate() logging.info(' Authentication verified') return AUTHENTICATED except ClientError: logging.error('Authentication failed with url %s and token %s' % ( auth_url, token)) return NOT_AUTHENTICATED
def check_credentials(token, auth_url='https://accounts.okeanos.grnet.gr' '/identity/v2.0'): '''Identity,Account/Astakos. Test authentication credentials''' logging.log(REPORT, ' Test the credentials') try: auth = AstakosClient(auth_url, token) auth.authenticate() except ClientError: logging.error('Authentication failed with url %s and token %s' % ( auth_url, token)) sys.exit(error_authentication) logging.log(REPORT, ' Authentication verified') return auth
def authenticate(self, authentication=None): """ :param authentication: :return: """ if self.__cyclades is not None: return True try: authcl = AstakosClient(authentication['URL'], authentication['TOKEN']) authcl.authenticate() self.__cyclades = CycladesClient(authcl.get_service_endpoints('compute')['publicURL'], authentication['TOKEN']) self.__network_client = CycladesNetworkClient(authcl.get_service_endpoints('network')['publicURL'], authentication['TOKEN']) except ClientError: stderr.write('Connector initialization failed') return False return True
def list_pithos_files(self): """ Method for listing pithos+ files available to the user """ auth_url = self.opts['auth_url'] token = self.opts['token'] try: auth = AstakosClient(auth_url, token) auth.authenticate() except ClientError: msg = ' Authentication error: Invalid Token' logging.error(msg) exit(error_fatal) pithos_endpoint = auth.get_endpoint_url('object-store') pithos_container = self.opts.get('pithos_container','pithos') user_id = auth.user_info['id'] pithos_client = PithosClient(pithos_endpoint,self.opts['token'], user_id, pithos_container) objects = pithos_client.list_objects() for object in objects: is_dir = 'application/directory' in object.get('content_type', object.get('content-type', '')) if not is_dir: print u"{:>12s} \"pithos:/{:s}/{:s}\"".format(bytes_to_shorthand(object['bytes']), pithos_container,object['name'])
def list_pithos_files(self): """ Method for listing pithos+ files available to the user """ auth_url = self.opts['auth_url'] token = self.opts['token'] try: auth = AstakosClient(auth_url, token) auth.authenticate() except ClientError: msg = 'Authentication error: Invalid Token' logging.error(msg) exit(error_fatal) pithos_endpoint = auth.get_endpoint_url('object-store') pithos_container = self.opts.get('pithos_container','pithos') user_id = auth.user_info['id'] pithos_client = PithosClient(pithos_endpoint,self.opts['token'], user_id, pithos_container) objects = pithos_client.list_objects() for object in objects: is_dir = 'application/directory' in object.get('content_type', object.get('content-type', '')) is_dir = 'application/folder' in object.get('content_type', object.get('content-type', '')) if not is_dir: print u"{:>12s} \"pithos:/{:s}/{:s}\"".format(bytes_to_shorthand(object['bytes']), pithos_container,object['name'])
def authenticate(self, authentication=None): """ :param authentication: :return: """ if self.__cyclades is not None: return True try: authcl = AstakosClient(authentication['URL'], authentication['TOKEN']) authcl.authenticate() self.__cyclades = CycladesClient( authcl.get_service_endpoints('compute')['publicURL'], authentication['TOKEN']) self.__network_client = CycladesNetworkClient( authcl.get_service_endpoints('network')['publicURL'], authentication['TOKEN']) except ClientError: stderr.write('Connector initialization failed') return False return True
class Clients(object): """Our kamaki clients""" auth_url = None token = None # Astakos astakos = None retry = CONNECTION_RETRY_LIMIT # Compute compute = None compute_url = None # Cyclades cyclades = None # Network network = None network_url = None # Pithos pithos = None pithos_url = None # Image image = None image_url = None def initialize_clients(self): """Initialize all the Kamaki Clients""" self.astakos = AstakosClient(self.auth_url, self.token) self.astakos.CONNECTION_RETRY_LIMIT = self.retry endpoints = self.astakos.authenticate() self.compute_url = _get_endpoint_url(endpoints, "compute") self.compute = ComputeClient(self.compute_url, self.token) self.compute.CONNECTION_RETRY_LIMIT = self.retry self.cyclades = CycladesClient(self.compute_url, self.token) self.cyclades.CONNECTION_RETRY_LIMIT = self.retry self.network_url = _get_endpoint_url(endpoints, "network") self.network = CycladesNetworkClient(self.network_url, self.token) self.network.CONNECTION_RETRY_LIMIT = self.retry self.pithos_url = _get_endpoint_url(endpoints, "object-store") self.pithos = PithosClient(self.pithos_url, self.token) self.pithos.CONNECTION_RETRY_LIMIT = self.retry self.image_url = _get_endpoint_url(endpoints, "image") self.image = ImageClient(self.image_url, self.token) self.image.CONNECTION_RETRY_LIMIT = self.retry
def check_auth_token(auth_token, auth_url=None): """ Checks the validity of a user authentication token. :param auth_token: User authentication token :param auth_url: Authentication url :return: tuple(Success status, details) """ if not auth_url: auth_url = "https://accounts.okeanos.grnet.gr/identity/v2.0" patch_certs() cl = AstakosClient(auth_url, auth_token) try: user_info = cl.authenticate() except ClientError as ex: if ex.message == 'UNAUTHORIZED': return False, ex.details else: raise return True, user_info
def main(): #authenticate to Astakos print"---------------------------------------------------" print"******* Authenticating to Astakos******************" print"---------------------------------------------------" try: my_astakos_client = AstakosClient(AUTHENTICATION_URL, TOKEN) my_accountData = my_astakos_client.authenticate() ACCOUNT_UUID = my_accountData['access']['user']['id'] print "Status: Authenticated" except ClientError: print"Failed to authenticate user token" print"\n" print"---------------------------------------------------" print"**********Getting Endpoints for pithos*************" print"---------------------------------------------------" #get endpoint url try: endpoints = my_astakos_client.get_service_endpoints('object-store') PITHOS_URL = endpoints['publicURL'] print "The public URL:", PITHOS_URL except ClientError: print "Failed to get endpoints for pithos" print"\n" print"---------------------------------------------------" print"**********Authenticating to Pithos*****************" print"---------------------------------------------------" #Initialize pithos client try: pithos = PithosClient(PITHOS_URL, TOKEN) pithos.account = ACCOUNT_UUID pithos.container = '' except ClientError: print "Failed to initialize Pithos+ client" print"\n" print"---------------------------------------------------" print"**********LIST ALL CONTAINERS IN YOUR ACCOUNT******" print"---------------------------------------------------" #list all containers try: container_list = pithos.list_containers() containers = parse_containers(container_list) ContNums = len(containers) print "The number of Containers in your account:", ContNums print "The containers are" print ','.join(containers) except ClientError: print"Error in container list" print"\n" print"---------------------------------------------------" print"******LIST OBJECTS OF A FOLDER IN A CONTAINER******" print"---------------------------------------------------" #list all containers try: print_container_objects(pithos, YOUR_CONTAINER, prefixName=YOUR_FOLDER_PATH) except ClientError: print"Error in listing folder objects" print"\n" print"---------------------------------------------------" print"**********Print objects for all containers*********" print"---------------------------------------------------" try: for i in range(len(containers)): print_container_objects(pithos, containers[i]) except ClientError as e: print('Error: %s' % e) if e.status: print('- error code: %s' % e.status) if e.details: for detail in e.details: print('- %s' % detail) # Create and set a different container than pithos print "Create a new container - my container" CONTAINER = 'my container' pithos.create_container(CONTAINER) pithos.container = CONTAINER print"\n" print"---------------------------------------------------" print"**********UPLOAD AND DOWNLOAD**********************" print"---------------------------------------------------" """ B. UPLOAD AND DOWNLOAD """ print "Upload a small file to pithos" # Upload a small file print './test/'+SMALLFILE with open('./test/'+SMALLFILE) as f: pithos.upload_object(SMALLFILE, f) print "Download a small file from pithos and store to string" print SMALLFILE FILETOSTRING = pithos.download_to_string(SMALLFILE, download_cb= create_pb('Downloading...')) print "Small file string:", FILETOSTRING #To optimize for large files, allow pithos client # to use multiple threads! pithos client will # auto-adjust the number of threads, up to a limit pithos.MAX_THREADS = 5 print "Upload a large file to pithos" # Now, large file upload will be optimized: # dd if=/dev/zero of=test/large.txt count=8 bs=1073741824 with open('./test/'+BIGFILE) as f: pithos.upload_object(BIGFILE, f, hash_cb=create_pb('Calculating hashes...'), upload_cb=create_pb('Uploading...')) print "Create my own metadata for object" tags = {} tags['mytag'] = 12 pithos.set_object_meta(BIGFILE, tags) myOwnMetadataObject = pithos.get_object_meta(BIGFILE) print "Object Metatadata", myOwnMetadataObject print "Download a large file from pithos" # Download a file (btw, MAX_THREADS are still 5) with open('./test/'+TMPFILE, 'wb+') as f: pithos.download_object(BIGFILE, f, download_cb=create_pb('Downloading...')) # HIGHLIGHTS: If parts of the file are already uploaded or downloaded, # corresponding methods will transfer only the missing parts! print"\n" print"---------------------------------------------------" print"**********CREATE A NEW CONTAINER AND MOVE OBJECT***" print"---------------------------------------------------" """ Create a new container and move object""" print "Create a new container - containerToCopy" CONTAINERNEW = 'containertocopy' pithos.create_container(CONTAINERNEW) pithos.move_object(CONTAINER, SMALLFILE, CONTAINERNEW, SMALLFILE) print"\n" print"---------------------------------------------------" print"**********DELETE AND RECOVER***********************" print"---------------------------------------------------" """ C. DELETE AND RECOVER """ # Delete a file pithos.container = CONTAINER pithos.delete_object(BIGFILE) # Recover file file_versions = pithos.get_object_versionlist(BIGFILE) print "The file versions" for data in file_versions: print "The value of id=", data[0], "date=", data[1] first_version = file_versions[0] v_id, v_date = first_version pithos.copy_object(CONTAINER, BIGFILE, CONTAINER, BIGFILE, source_version=v_id) print"\n" print"---------------------------------------------------" print"**********GET FILE DETAILS************************" print"---------------------------------------------------" objectDetails = pithos.get_object_info(BIGFILE) for obj in objectDetails: print "The value of", obj, "=", objectDetails.get(obj) print"\n" print"---------------------------------------------------" print"**********SHARING AND PUBLISHING*******************" print"---------------------------------------------------" """ D. SHARING AND PUBLISHING """ # Read permission to all pithos users pithos.set_object_sharing(FILETOPUBLISH, read_permission='*') # Publish and get public URL pithos.publish_object(FILETOPUBLISH) print "Get sharing and public information" # Get sharing and public information info = pithos.get_object_info(FILETOPUBLISH) for data in info: print "The value of", data, "=", info.get(data) sharing = info.get('x-object-sharing', {}) print sharing public = info.get('x-object-public', None) print "The public URL=", public print "Remove sharing and publishing" # Remove sharing and publishing pithos.del_object_sharing(FILETOPUBLISH) pithos.unpublish_object(FILETOPUBLISH) print "Get sharing and public information" # Get sharing and public information info = pithos.get_object_info(FILETOPUBLISH) sharing = info.get('x-object-sharing', {}) public = info.get('x-object-public', None) print "The public URL=", public
def main(): """Parse arguments, use kamaki to create cluster, setup using ssh""" (opts, args) = parse_arguments(sys.argv[1:]) global CYCLADES, TOKEN AUTHENTICATION_URL = opts.cyclades TOKEN = opts.token # Cleanup stale servers from previous runs if opts.show_stale: cleanup_servers(prefix=opts.prefix, delete_stale=opts.delete_stale) return 0 # Initialize a kamaki instance, get endpoints user = AstakosClient(AUTHENTICATION_URL, TOKEN) my_accountData = user.authenticate() endpoints = user.get_endpoints() cyclades_endpoints = user.get_endpoints('compute') cyclades_base_url = parseAstakosEndpoints(endpoints,'cyclades_compute') cyclades_network_base_url = parseAstakosEndpoints(endpoints,'cyclades_network') my_cyclades_client = CycladesClient(cyclades_base_url, TOKEN) my_compute_client = ComputeClient(cyclades_base_url, TOKEN) my_network_client = CycladesNetworkClient(cyclades_network_base_url, TOKEN) cnt = int(opts.clustersize) # calculate size of cluster into 'cnt' # Initialize nodes = [] masterName = '' # Create a file to store the root password for later use pass_fname = opts.hadoop_dir+'/bak/adminPass'+str(datetime.now())[:19].replace(' ', '') adminPass_f = open(pass_fname, 'w') myNetworks = my_network_client.list_networks(); NetWork_free = parseNetwork(myNetworks,'public'); myIp = my_network_client.create_floatingip(NetWork_free); LastIp = myIp.get("floating_ip_address") initialClusterSize = 0 server = {} if opts.extend == False: # Create master node (0th node) server = create_machine(opts, my_cyclades_client, 0) if server == {}: return else: servers = my_cyclades_client.list_servers(detail=True) cluster = [s for s in servers if s["name"].startswith(opts.prefix)] initialClusterSize = len(cluster) if initialClusterSize==0: log.info("Cluster cannot be expanded: it does not exist.") return servername = "%s-0" % (opts.prefix) masterName = servername nodes.append(server) # Create slave (worker) nodes if cnt>1 or opts.extend: startingOffset = 1 if opts.extend: startingOffset = initialClusterSize for i in xrange(startingOffset, initialClusterSize+cnt): server = {} server = create_machine(opts, my_cyclades_client, i) if server == {}: return; nodes.append(server) servername = "%s-%d" % (opts.prefix, i) # Write the root password to a file adminPass_f.write('machine = %s, password = %s\n' % (servername, server['adminPass'])) adminPass_f.close() # Setup Hadoop files and settings on all cluster nodes # Create the 'cluster' dictionary out of servers, with only Hadoop-relevant keys (name, ip, integer key) servers = my_cyclades_client.list_servers(detail=True) cluster = [s for s in my_cyclades_client.list_servers(detail=True) if s["name"].startswith(opts.prefix)] cluster = [(s["name"], s["attachments"][1]["ipv4"], int(s["name"][s["name"].find('-')+1:])) for s in cluster] cluster = sorted(cluster, key=lambda cluster: cluster[2]) # Prepare Ansible-Hadoop config files (hosts, conf/slaves) hosts = open(opts.hadoop_dir+'/hosts', 'w') hosts.write('[master]\n') for i in xrange(0, initialClusterSize+cnt): for s in cluster: if s[0] == opts.prefix+"-"+str(i): if s[0] == masterName: hosts.write(s[1]+'\n\n'+'[slaves]\n') else: hosts.write(s[1]+'\n') hosts.close() slaves = open(opts.hadoop_dir+'/conf/slaves', 'w') for s in cluster[1:]: slaves.write(s[1]+'\n') slaves.close() # Execute respective ansible playbook if (opts.extend==False): cmd = "ansible-playbook hadoop.yml -i hosts -vv --extra-vars \""+"master_ip="+cluster[0][1]+"\""+" -l master" retval = os.system(cmd) cmd = "ansible-playbook hadoop.yml -i hosts -vv --extra-vars \""+"master_ip="+cluster[0][1]+"\""+" -l slaves" retval = os.system(cmd) slave_ip_list = [] for i in xrange(1, cnt): slave_ip_list.append(cluster[i][1]) enable_ssh_login(cluster[0][1], [cluster[0][1]]) enable_ssh_login(cluster[0][1], slave_ip_list) else: hosts_latest = open(opts.hadoop_dir+'/hosts.latest', 'w') hosts_latest.write('[master]\n') hosts_latest.write(cluster[0][1]+'\n\n'+'[slaves]\n') for i in xrange(initialClusterSize, initialClusterSize+cnt): hosts_latest.write(cluster[i][1]+'\n') hosts_latest.close() cmd = "ansible-playbook hadoop.yml -i hosts.latest -vv --extra-vars \""+"master_ip="+cluster[0][1]+"\""+" -l slaves" retval = os.system(cmd) slave_ip_list = [] for i in xrange(initialClusterSize, initialClusterSize+cnt): slave_ip_list.append(cluster[i][1]) enable_ssh_login(cluster[0][1], slave_ip_list) # Update conf/slaves in master cmd = "ansible-playbook hadoop.yml -i hosts -vv --extra-vars \""+"master_ip="+cluster[0][1]+"\""+" -l master -t slaves" retval = os.system(cmd) log.info("Done.")
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE # POSSIBILITY OF SUCH DAMAGE. # # The views and conclusions contained in the software and # documentation are those of the authors and should not be # interpreted as representing official policies, either expressed # or implied, of GRNET S.A. from kamaki.clients.astakos import AstakosClient from kamaki.clients.image import ImageClient AUTHENTICATION_URL = "https://astakos.example.com/identity/v2.0" TOKEN = "User-Token" astakos = AstakosClient(AUTHENTICATION_URL, TOKEN) user = astakos.authenticate() uuid = user["access"]["user"]["id"] service_type = ImageClient.service_type endpoint = astakos.get_endpoint_url(service_type) image = ImageClient(endpoint, TOKEN) # Get all images owned/registered by me images = filter(lambda img: img["owner"] == uuid, image.list_public()) print "My images:\n" for i in images: print "\t{name} ({location})".format(name=i["name"], location=i["location"])
class CloudClient(object): clients_map_override = { 'cyclades': {'endpoint': 'compute'} } clients_override = { 'network': 'cyclades_network' } def __init__(self, config, cloud=None, debug=False): _setup_logging(debug, debug) self.config = kamaki_config.Config(config) self.cloud = cloud if not self.cloud: self.cloud = self.config.get("global", "default_cloud") self.ignore_ssl = \ self.config.get("global", "ignore_ssl").lower() == "on" self.ca_certs = \ self.config.get("global", "ca_certs") https.patch_ignore_ssl(self.ignore_ssl) if self.ca_certs is not None: https.patch_with_certs(self.ca_certs) self.auth_url = self.config.get_cloud(self.cloud, "url") self.token = self.config.get_cloud(self.cloud, "token") self.auth_client = AstakosClient(self.auth_url, self.token) self.endpoints = None # lazyness def fill_endpoints(self): self.endpoints = self.auth_client.authenticate() def get_cli(self, name): name = self.clients_override.get(name, name) if not self.endpoints: self.fill_endpoints() ns = name.split("_")[0].lower() normalized = name.replace("_", " ").title().replace(" ", "") import_path = self.clients_map_override.get(name, {}).get('module', None) if not import_path: import_path = 'kamaki.clients.%s.%s' % (ns, normalized + 'Client') module_name = ".".join(import_path.split(".")[:-1]) cli_name = import_path.split(".")[-1] module = importlib.import_module(module_name) endpoint_type = self.clients_map_override.get( name, {}).get('endpoint', ns) try: catalog = parse_endpoints(self.endpoints, ep_type=endpoint_type) except NoEndpoints as e: if "_" not in name: raise e endpoint_type = name.split("_")[1] catalog = parse_endpoints(self.endpoints, ep_type=endpoint_type) endpoint = catalog[0]['endpoints'][0]['publicURL'] return module.__dict__[cli_name](endpoint, self.token) def __getattr__(self, name): if name not in self.__dict__: return self.get_cli(name) return object.__getattribute__(self, name)
class SynnefoCI(object): """SynnefoCI python class""" def __init__(self, config_file=None, build_id=None, cloud=None): """ Initialize SynnefoCI python class Setup logger, local_dir, config and kamaki """ # Setup logger self.logger = logging.getLogger('synnefo-ci') self.logger.setLevel(logging.DEBUG) handler1 = logging.StreamHandler(sys.stdout) handler1.setLevel(logging.DEBUG) handler1.addFilter(_InfoFilter()) handler1.setFormatter(_MyFormatter()) handler2 = logging.StreamHandler(sys.stderr) handler2.setLevel(logging.WARNING) handler2.setFormatter(_MyFormatter()) self.logger.addHandler(handler1) self.logger.addHandler(handler2) # Get our local dir self.ci_dir = os.path.dirname(os.path.abspath(__file__)) self.repo_dir = os.path.dirname(self.ci_dir) # Read config file if config_file is None: config_file = os.path.join(self.ci_dir, DEFAULT_CONFIG_FILE) config_file = os.path.abspath(config_file) self.config = ConfigParser() self.config.optionxform = str self.config.read(config_file) # Read temporary_config file self.temp_config_file = \ os.path.expanduser(self.config.get('Global', 'temporary_config')) self.temp_config = ConfigParser() self.temp_config.optionxform = str self.temp_config.read(self.temp_config_file) self.build_id = build_id if build_id is not None: self.logger.info("Will use \"%s\" as build id" % _green(self.build_id)) # Set kamaki cloud if cloud is not None: self.kamaki_cloud = cloud elif self.config.has_option("Deployment", "kamaki_cloud"): kamaki_cloud = self.config.get("Deployment", "kamaki_cloud") if kamaki_cloud == "": self.kamaki_cloud = None else: self.kamaki_cloud = None # Initialize variables self.fabric_installed = False self.kamaki_installed = False self.cyclades_client = None self.network_client = None self.compute_client = None self.image_client = None self.astakos_client = None def setup_kamaki(self): """Initialize kamaki Setup cyclades_client, image_client and compute_client """ config = kamaki_config.Config() if self.kamaki_cloud is None: try: self.kamaki_cloud = config.get("global", "default_cloud") except AttributeError: # Compatibility with kamaki version <=0.10 self.kamaki_cloud = config.get("global", "default_cloud") self.logger.info("Setup kamaki client, using cloud '%s'.." % self.kamaki_cloud) auth_url = config.get_cloud(self.kamaki_cloud, "url") self.logger.debug("Authentication URL is %s" % _green(auth_url)) token = config.get_cloud(self.kamaki_cloud, "token") #self.logger.debug("Token is %s" % _green(token)) self.astakos_client = AstakosClient(auth_url, token) endpoints = self.astakos_client.authenticate() cyclades_url = get_endpoint_url(endpoints, "compute") self.logger.debug("Cyclades API url is %s" % _green(cyclades_url)) self.cyclades_client = CycladesClient(cyclades_url, token) self.cyclades_client.CONNECTION_RETRY_LIMIT = 2 network_url = get_endpoint_url(endpoints, "network") self.logger.debug("Network API url is %s" % _green(network_url)) self.network_client = CycladesNetworkClient(network_url, token) self.network_client.CONNECTION_RETRY_LIMIT = 2 image_url = get_endpoint_url(endpoints, "image") self.logger.debug("Images API url is %s" % _green(image_url)) self.image_client = ImageClient(cyclades_url, token) self.image_client.CONNECTION_RETRY_LIMIT = 2 compute_url = get_endpoint_url(endpoints, "compute") self.logger.debug("Compute API url is %s" % _green(compute_url)) self.compute_client = ComputeClient(compute_url, token) self.compute_client.CONNECTION_RETRY_LIMIT = 2 def _wait_transition(self, server_id, current_status, new_status): """Wait for server to go from current_status to new_status""" self.logger.debug("Waiting for server to become %s" % new_status) timeout = self.config.getint('Global', 'build_timeout') sleep_time = 5 while True: server = self.cyclades_client.get_server_details(server_id) if server['status'] == new_status: return server elif timeout < 0: self.logger.error( "Waiting for server to become %s timed out" % new_status) self.destroy_server(False) sys.exit(1) elif server['status'] == current_status: # Sleep for #n secs and continue timeout = timeout - sleep_time time.sleep(sleep_time) else: self.logger.error( "Server failed with status %s" % server['status']) self.destroy_server(False) sys.exit(1) @_check_kamaki def destroy_server(self, wait=True): """Destroy slave server""" server_id = int(self.read_temp_config('server_id')) fips = [f for f in self.network_client.list_floatingips() if str(f['instance_id']) == str(server_id)] self.logger.info("Destoying server with id %s " % server_id) self.cyclades_client.delete_server(server_id) if wait: self._wait_transition(server_id, "ACTIVE", "DELETED") for fip in fips: self.logger.info("Destroying floating ip %s", fip['floating_ip_address']) self.network_client.delete_floatingip(fip['id']) def _create_floating_ip(self): """Create a new floating ip""" networks = self.network_client.list_networks(detail=True) pub_nets = [n for n in networks if n['SNF:floating_ip_pool'] and n['public']] for pub_net in pub_nets: # Try until we find a public network that is not full try: fip = self.network_client.create_floatingip(pub_net['id']) except ClientError as err: self.logger.warning("%s: %s", err.message, err.details) continue self.logger.debug("Floating IP %s with id %s created", fip['floating_ip_address'], fip['id']) return fip self.logger.error("No mor IP addresses available") sys.exit(1) def _create_port(self, floating_ip): """Create a new port for our floating IP""" net_id = floating_ip['floating_network_id'] self.logger.debug("Creating a new port to network with id %s", net_id) fixed_ips = [{'ip_address': floating_ip['floating_ip_address']}] port = self.network_client.create_port( net_id, device_id=None, fixed_ips=fixed_ips) return port @_check_kamaki # Too many local variables. pylint: disable-msg=R0914 def create_server(self, image=None, flavor=None, ssh_keys=None, server_name=None): """Create slave server""" self.logger.info("Create a new server..") # Find a build_id to use self._create_new_build_id() # Find an image to use image_id = self._find_image(image) # Find a flavor to use flavor_id = self._find_flavor(flavor) # Create Server networks = [] if self.config.get("Deployment", "allocate_floating_ip") == "True": fip = self._create_floating_ip() port = self._create_port(fip) networks.append({'port': port['id']}) private_networks = self.config.get('Deployment', 'private_networks') if private_networks: private_networks = [p.strip() for p in private_networks.split(",")] networks.extend([{"uuid": uuid} for uuid in private_networks]) if server_name is None: server_name = self.config.get("Deployment", "server_name") server_name = "%s(BID: %s)" % (server_name, self.build_id) server = self.cyclades_client.create_server( server_name, flavor_id, image_id, networks=networks) server_id = server['id'] self.write_temp_config('server_id', server_id) self.logger.debug("Server got id %s" % _green(server_id)) server_user = server['metadata']['users'] self.write_temp_config('server_user', server_user) self.logger.debug("Server's admin user is %s" % _green(server_user)) server_passwd = server['adminPass'] self.write_temp_config('server_passwd', server_passwd) server = self._wait_transition(server_id, "BUILD", "ACTIVE") self._get_server_ip_and_port(server, private_networks) self._copy_ssh_keys(ssh_keys) # Setup Firewall self.setup_fabric() self.logger.info("Setup firewall") accept_ssh_from = self.config.get('Global', 'accept_ssh_from') if accept_ssh_from != "": self.logger.debug("Block ssh except from %s" % accept_ssh_from) cmd = """ local_ip=$(/sbin/ifconfig eth0 | grep 'inet addr:' | \ cut -d':' -f2 | cut -d' ' -f1) iptables -A INPUT -s localhost -j ACCEPT iptables -A INPUT -s $local_ip -j ACCEPT iptables -A INPUT -s {0} -p tcp --dport 22 -j ACCEPT iptables -A INPUT -p tcp --dport 22 -j DROP """.format(accept_ssh_from) _run(cmd, False) # Setup apt, download packages self.logger.debug("Setup apt. Install x2goserver and firefox") cmd = """ echo 'APT::Install-Suggests "false";' >> /etc/apt/apt.conf echo 'precedence ::ffff:0:0/96 100' >> /etc/gai.conf apt-get update apt-get install curl --yes --force-yes echo -e "\n\n{0}" >> /etc/apt/sources.list # Synnefo repo's key curl https://dev.grnet.gr/files/apt-grnetdev.pub | apt-key add - # X2GO Key apt-key adv --recv-keys --keyserver keys.gnupg.net E1F958385BFE2B6E apt-get install x2go-keyring --yes --force-yes apt-get update apt-get install x2goserver x2goserver-xsession \ iceweasel --yes --force-yes # xterm published application echo '[Desktop Entry]' > /usr/share/applications/xterm.desktop echo 'Name=XTerm' >> /usr/share/applications/xterm.desktop echo 'Comment=standard terminal emulator for the X window system' >> \ /usr/share/applications/xterm.desktop echo 'Exec=xterm' >> /usr/share/applications/xterm.desktop echo 'Terminal=false' >> /usr/share/applications/xterm.desktop echo 'Type=Application' >> /usr/share/applications/xterm.desktop echo 'Encoding=UTF-8' >> /usr/share/applications/xterm.desktop echo 'Icon=xterm-color_48x48' >> /usr/share/applications/xterm.desktop echo 'Categories=System;TerminalEmulator;' >> \ /usr/share/applications/xterm.desktop """.format(self.config.get('Global', 'apt_repo')) _run(cmd, False) def _find_flavor(self, flavor=None): """Find a suitable flavor to use Search by name (reg expression) or by id """ # Get a list of flavors from config file flavors = self.config.get('Deployment', 'flavors').split(",") if flavor is not None: # If we have a flavor_name to use, add it to our list flavors.insert(0, flavor) list_flavors = self.compute_client.list_flavors() for flv in flavors: flv_type, flv_value = parse_typed_option(option="flavor", value=flv) if flv_type == "name": # Filter flavors by name self.logger.debug( "Trying to find a flavor with name \"%s\"" % flv_value) list_flvs = \ [f for f in list_flavors if re.search(flv_value, f['name'], flags=re.I) is not None] elif flv_type == "id": # Filter flavors by id self.logger.debug( "Trying to find a flavor with id \"%s\"" % flv_value) list_flvs = \ [f for f in list_flavors if str(f['id']) == flv_value] else: self.logger.error("Unrecognized flavor type %s" % flv_type) # Check if we found one if list_flvs: self.logger.debug("Will use \"%s\" with id \"%s\"" % (_green(list_flvs[0]['name']), _green(list_flvs[0]['id']))) return list_flvs[0]['id'] self.logger.error("No matching flavor found.. aborting") sys.exit(1) def _find_image(self, image=None): """Find a suitable image to use In case of search by name, the image has to belong to one of the `DEFAULT_SYSTEM_IMAGES_UUID' users. In case of search by id it only has to exist. """ # Get a list of images from config file images = self.config.get('Deployment', 'images').split(",") if image is not None: # If we have an image from command line, add it to our list images.insert(0, image) auth = self.astakos_client.authenticate() user_uuid = auth["access"]["token"]["tenant"]["id"] list_images = self.image_client.list_public(detail=True)['images'] for img in images: img_type, img_value = parse_typed_option(option="image", value=img) if img_type == "name": # Filter images by name self.logger.debug( "Trying to find an image with name \"%s\"" % img_value) accepted_uuids = DEFAULT_SYSTEM_IMAGES_UUID + [user_uuid] list_imgs = \ [i for i in list_images if i['user_id'] in accepted_uuids and re.search(img_value, i['name'], flags=re.I) is not None] elif img_type == "id": # Filter images by id self.logger.debug( "Trying to find an image with id \"%s\"" % img_value) list_imgs = \ [i for i in list_images if i['id'].lower() == img_value.lower()] else: self.logger.error("Unrecognized image type %s" % img_type) sys.exit(1) # Check if we found one if list_imgs: self.logger.debug("Will use \"%s\" with id \"%s\"" % (_green(list_imgs[0]['name']), _green(list_imgs[0]['id']))) return list_imgs[0]['id'] # We didn't found one self.logger.error("No matching image found.. aborting") sys.exit(1) def _get_server_ip_and_port(self, server, private_networks): """Compute server's IPv4 and ssh port number""" self.logger.info("Get server connection details..") if private_networks: # Choose the networks that belong to private_networks networks = [n for n in server['attachments'] if n['network_id'] in private_networks] else: # Choose the networks that are public networks = [n for n in server['attachments'] if self.network_client. get_network_details(n['network_id'])['public']] # Choose the networks with IPv4 networks = [n for n in networks if n['ipv4']] # Use the first network as IPv4 server_ip = networks[0]['ipv4'] if (".okeanos.io" in self.cyclades_client.base_url or ".demo.synnefo.org" in self.cyclades_client.base_url): tmp1 = int(server_ip.split(".")[2]) tmp2 = int(server_ip.split(".")[3]) server_ip = "gate.okeanos.io" server_port = 10000 + tmp1 * 256 + tmp2 else: server_port = 22 self.write_temp_config('server_ip', server_ip) self.logger.debug("Server's IPv4 is %s" % _green(server_ip)) self.write_temp_config('server_port', server_port) self.logger.debug("Server's ssh port is %s" % _green(server_port)) ssh_command = "ssh -p %s %s@%s" \ % (server_port, server['metadata']['users'], server_ip) self.logger.debug("Access server using \"%s\"" % (_green(ssh_command))) @_check_fabric def _copy_ssh_keys(self, ssh_keys): """Upload/Install ssh keys to server""" self.logger.debug("Check for authentication keys to use") if ssh_keys is None: ssh_keys = self.config.get("Deployment", "ssh_keys") if ssh_keys != "": ssh_keys = os.path.expanduser(ssh_keys) self.logger.debug("Will use \"%s\" authentication keys file" % _green(ssh_keys)) keyfile = '/tmp/%s.pub' % fabric.env.user _run('mkdir -p ~/.ssh && chmod 700 ~/.ssh', False) if ssh_keys.startswith("http://") or \ ssh_keys.startswith("https://") or \ ssh_keys.startswith("ftp://"): cmd = """ apt-get update apt-get install wget --yes --force-yes wget {0} -O {1} --no-check-certificate """.format(ssh_keys, keyfile) _run(cmd, False) elif os.path.exists(ssh_keys): _put(ssh_keys, keyfile) else: self.logger.debug("No ssh keys found") return _run('cat %s >> ~/.ssh/authorized_keys' % keyfile, False) _run('rm %s' % keyfile, False) self.logger.debug("Uploaded ssh authorized keys") else: self.logger.debug("No ssh keys found") def _create_new_build_id(self): """Find a uniq build_id to use""" with filelocker.lock("%s.lock" % self.temp_config_file, filelocker.LOCK_EX): # Read temp_config again to get any new entries self.temp_config.read(self.temp_config_file) # Find a uniq build_id to use if self.build_id is None: ids = self.temp_config.sections() if ids: max_id = int(max(self.temp_config.sections(), key=int)) self.build_id = max_id + 1 else: self.build_id = 1 self.logger.debug("Will use \"%s\" as build id" % _green(self.build_id)) # Create a new section try: self.temp_config.add_section(str(self.build_id)) except DuplicateSectionError: msg = ("Build id \"%s\" already in use. " + "Please use a uniq one or cleanup \"%s\" file.\n") \ % (self.build_id, self.temp_config_file) self.logger.error(msg) sys.exit(1) creation_time = \ time.strftime("%a, %d %b %Y %X", time.localtime()) self.temp_config.set(str(self.build_id), "created", str(creation_time)) # Write changes back to temp config file with open(self.temp_config_file, 'wb') as tcf: self.temp_config.write(tcf) def write_temp_config(self, option, value): """Write changes back to config file""" # Acquire the lock to write to temp_config_file with filelocker.lock("%s.lock" % self.temp_config_file, filelocker.LOCK_EX): # Read temp_config again to get any new entries self.temp_config.read(self.temp_config_file) self.temp_config.set(str(self.build_id), option, str(value)) curr_time = time.strftime("%a, %d %b %Y %X", time.localtime()) self.temp_config.set(str(self.build_id), "modified", curr_time) # Write changes back to temp config file with open(self.temp_config_file, 'wb') as tcf: self.temp_config.write(tcf) def read_temp_config(self, option): """Read from temporary_config file""" # If build_id is None use the latest one if self.build_id is None: ids = self.temp_config.sections() if ids: self.build_id = int(ids[-1]) else: self.logger.error("No sections in temporary config file") sys.exit(1) self.logger.debug("Will use \"%s\" as build id" % _green(self.build_id)) # Read specified option return self.temp_config.get(str(self.build_id), option) def setup_fabric(self): """Setup fabric environment""" self.logger.info("Setup fabric parameters..") fabric.env.user = self.read_temp_config('server_user') fabric.env.host_string = self.read_temp_config('server_ip') fabric.env.port = int(self.read_temp_config('server_port')) fabric.env.password = self.read_temp_config('server_passwd') fabric.env.connection_attempts = 10 fabric.env.shell = "/bin/bash -c" fabric.env.disable_known_hosts = True fabric.env.output_prefix = None def _check_hash_sum(self, localfile, remotefile): """Check hash sums of two files""" self.logger.debug("Check hash sum for local file %s" % localfile) hash1 = os.popen("sha256sum %s" % localfile).read().split(' ')[0] self.logger.debug("Local file has sha256 hash %s" % hash1) self.logger.debug("Check hash sum for remote file %s" % remotefile) hash2 = _run("sha256sum %s" % remotefile, False) hash2 = hash2.split(' ')[0] self.logger.debug("Remote file has sha256 hash %s" % hash2) if hash1 != hash2: self.logger.error("Hashes differ.. aborting") sys.exit(1) @_check_fabric def clone_repo(self, local_repo=False): """Clone Synnefo repo from slave server""" self.logger.info("Configure repositories on remote server..") self.logger.debug("Install/Setup git") cmd = """ apt-get install git --yes --force-yes git config --global user.name {0} git config --global user.email {1} """.format(self.config.get('Global', 'git_config_name'), self.config.get('Global', 'git_config_mail')) _run(cmd, False) # Clone synnefo_repo synnefo_branch = self.clone_synnefo_repo(local_repo=local_repo) # Clone pithos-web-client self.clone_pithos_webclient_repo(synnefo_branch) @_check_fabric def clone_synnefo_repo(self, local_repo=False): """Clone Synnefo repo to remote server""" # Find synnefo_repo and synnefo_branch to use synnefo_repo = self.config.get('Global', 'synnefo_repo') synnefo_branch = self.config.get("Global", "synnefo_branch") if synnefo_branch == "": synnefo_branch = \ subprocess.Popen( ["git", "rev-parse", "--abbrev-ref", "HEAD"], stdout=subprocess.PIPE).communicate()[0].strip() if synnefo_branch == "HEAD": synnefo_branch = \ subprocess.Popen( ["git", "rev-parse", "--short", "HEAD"], stdout=subprocess.PIPE).communicate()[0].strip() self.logger.debug("Will use branch \"%s\"" % _green(synnefo_branch)) if local_repo or synnefo_repo == "": # Use local_repo self.logger.debug("Push local repo to server") # Firstly create the remote repo _run("git init synnefo", False) # Then push our local repo over ssh # We have to pass some arguments to ssh command # namely to disable host checking. (temp_ssh_file_handle, temp_ssh_file) = tempfile.mkstemp() os.close(temp_ssh_file_handle) # XXX: git push doesn't read the password cmd = """ echo 'exec ssh -o "StrictHostKeyChecking no" \ -o "UserKnownHostsFile /dev/null" \ -q "$@"' > {4} chmod u+x {4} export GIT_SSH="{4}" echo "{0}" | git push --quiet --mirror ssh://{1}@{2}:{3}/~/synnefo rm -f {4} """.format(fabric.env.password, fabric.env.user, fabric.env.host_string, fabric.env.port, temp_ssh_file) os.system(cmd) else: # Clone Synnefo from remote repo self.logger.debug("Clone synnefo from %s" % synnefo_repo) self._git_clone(synnefo_repo) # Checkout the desired synnefo_branch self.logger.debug("Checkout \"%s\" branch/commit" % synnefo_branch) cmd = """ cd synnefo for branch in `git branch -a | grep remotes | grep -v HEAD`; do git branch --track ${branch##*/} $branch done git checkout %s """ % (synnefo_branch) _run(cmd, False) return synnefo_branch @_check_fabric def clone_pithos_webclient_repo(self, synnefo_branch): """Clone Pithos WebClient repo to remote server""" # Find pithos_webclient_repo and pithos_webclient_branch to use pithos_webclient_repo = \ self.config.get('Global', 'pithos_webclient_repo') pithos_webclient_branch = \ self.config.get('Global', 'pithos_webclient_branch') # Clone pithos-webclient from remote repo self.logger.debug("Clone pithos-webclient from %s" % pithos_webclient_repo) self._git_clone(pithos_webclient_repo) # Track all pithos-webclient branches cmd = """ cd pithos-web-client for branch in `git branch -a | grep remotes | grep -v HEAD`; do git branch --track ${branch##*/} $branch > /dev/null 2>&1 done git --no-pager branch --no-color """ webclient_branches = _run(cmd, False) webclient_branches = webclient_branches.split() # If we have pithos_webclient_branch in config file use this one # else try to use the same branch as synnefo_branch # else use an appropriate one. if pithos_webclient_branch == "": if synnefo_branch in webclient_branches: pithos_webclient_branch = synnefo_branch else: # If synnefo_branch starts with one of # 'master', 'hotfix'; use the master branch if synnefo_branch.startswith('master') or \ synnefo_branch.startswith('hotfix'): pithos_webclient_branch = "master" # If synnefo_branch starts with one of # 'develop', 'feature'; use the develop branch elif synnefo_branch.startswith('develop') or \ synnefo_branch.startswith('feature'): pithos_webclient_branch = "develop" else: self.logger.warning( "Cannot determine which pithos-web-client branch to " "use based on \"%s\" synnefo branch. " "Will use develop." % synnefo_branch) pithos_webclient_branch = "develop" # Checkout branch self.logger.debug("Checkout \"%s\" branch" % _green(pithos_webclient_branch)) cmd = """ cd pithos-web-client git checkout {0} """.format(pithos_webclient_branch) _run(cmd, False) def _git_clone(self, repo): """Clone repo to remote server Currently clonning from code.grnet.gr can fail unexpectedly. So retry!! """ cloned = False for i in range(1, 11): try: _run("git clone %s" % repo, False) cloned = True break except BaseException: self.logger.warning("Clonning failed.. retrying %s/10" % i) if not cloned: self.logger.error("Can not clone repo.") sys.exit(1) @_check_fabric def build_packages(self): """Build packages needed by Synnefo software""" self.logger.info("Install development packages") cmd = """ apt-get update apt-get install zlib1g-dev dpkg-dev debhelper git-buildpackage \ python-dev python-all python-pip ant --yes --force-yes pip install -U devflow """ _run(cmd, False) # Patch pydist bug if self.config.get('Global', 'patch_pydist') == "True": self.logger.debug("Patch pydist.py module") cmd = r""" sed -r -i 's/(\(\?P<name>\[A-Za-z\]\[A-Za-z0-9_\.)/\1\\\-/' \ /usr/share/python/debpython/pydist.py """ _run(cmd, False) # Build synnefo packages self.build_synnefo() # Build pithos-web-client packages self.build_pithos_webclient() @_check_fabric def build_synnefo(self): """Build Synnefo packages""" self.logger.info("Build Synnefo packages..") cmd = """ devflow-autopkg snapshot -b ~/synnefo_build-area --no-sign """ with fabric.cd("synnefo"): _run(cmd, True) # Install snf-deploy package self.logger.debug("Install snf-deploy package") cmd = """ dpkg -i snf-deploy*.deb apt-get -f install --yes --force-yes """ with fabric.cd("synnefo_build-area"): with fabric.settings(warn_only=True): _run(cmd, True) # Setup synnefo packages for snf-deploy self.logger.debug("Copy synnefo debs to snf-deploy packages dir") cmd = """ cp ~/synnefo_build-area/*.deb /var/lib/snf-deploy/packages/ """ _run(cmd, False) @_check_fabric def build_pithos_webclient(self): """Build pithos-web-client packages""" self.logger.info("Build pithos-web-client packages..") cmd = """ devflow-autopkg snapshot -b ~/webclient_build-area --no-sign """ with fabric.cd("pithos-web-client"): _run(cmd, True) # Setup pithos-web-client packages for snf-deploy self.logger.debug("Copy webclient debs to snf-deploy packages dir") cmd = """ cp ~/webclient_build-area/*.deb /var/lib/snf-deploy/packages/ """ _run(cmd, False) @_check_fabric def build_documentation(self): """Build Synnefo documentation""" self.logger.info("Build Synnefo documentation..") _run("pip install -U Sphinx", False) with fabric.cd("synnefo"): _run("devflow-update-version; " "./ci/make_docs.sh synnefo_documentation", False) def fetch_documentation(self, dest=None): """Fetch Synnefo documentation""" self.logger.info("Fetch Synnefo documentation..") if dest is None: dest = "synnefo_documentation" dest = os.path.abspath(dest) if not os.path.exists(dest): os.makedirs(dest) self.fetch_compressed("synnefo/synnefo_documentation", dest) self.logger.info("Downloaded documentation to %s" % _green(dest)) @_check_fabric def deploy_synnefo(self, schema=None): """Deploy Synnefo using snf-deploy""" self.logger.info("Deploy Synnefo..") if schema is None: schema = self.config.get('Global', 'schema') self.logger.debug("Will use \"%s\" schema" % _green(schema)) schema_dir = os.path.join(self.ci_dir, "schemas/%s" % schema) if not (os.path.exists(schema_dir) and os.path.isdir(schema_dir)): raise ValueError("Unknown schema: %s" % schema) self.logger.debug("Upload schema files to server") _put(os.path.join(schema_dir, "*"), "/etc/snf-deploy/") self.logger.debug("Change password in nodes.conf file") cmd = """ sed -i 's/^password =.*/password = {0}/' /etc/snf-deploy/nodes.conf """.format(fabric.env.password) _run(cmd, False) self.logger.debug("Run snf-deploy") cmd = """ snf-deploy keygen --force snf-deploy --disable-colors --autoconf all """ _run(cmd, True) @_check_fabric def unit_test(self): """Run Synnefo unit test suite""" self.logger.info("Run Synnefo unit test suite") component = self.config.get('Unit Tests', 'component') self.logger.debug("Install needed packages") cmd = """ pip install -U mock pip install -U factory_boy pip install -U nose """ _run(cmd, False) self.logger.debug("Upload tests.sh file") unit_tests_file = os.path.join(self.ci_dir, "tests.sh") _put(unit_tests_file, ".") self.logger.debug("Run unit tests") cmd = """ bash tests.sh {0} """.format(component) _run(cmd, True) @_check_fabric def run_burnin(self): """Run burnin functional test suite""" self.logger.info("Run Burnin functional test suite") cmd = """ auth_url=$(grep -e '^url =' .kamakirc | cut -d' ' -f3) token=$(grep -e '^token =' .kamakirc | cut -d' ' -f3) images_user=$(kamaki image list -l | grep owner | \ cut -d':' -f2 | tr -d ' ') snf-burnin --auth-url=$auth_url --token=$token {0} BurninExitStatus=$? exit $BurninExitStatus """.format(self.config.get('Burnin', 'cmd_options')) _run(cmd, True) @_check_fabric def fetch_compressed(self, src, dest=None): """Create a tarball and fetch it locally""" self.logger.debug("Creating tarball of %s" % src) basename = os.path.basename(src) tar_file = basename + ".tgz" cmd = "tar czf %s %s" % (tar_file, src) _run(cmd, False) if not os.path.exists(dest): os.makedirs(dest) tmp_dir = tempfile.mkdtemp() fabric.get(tar_file, tmp_dir) dest_file = os.path.join(tmp_dir, tar_file) self._check_hash_sum(dest_file, tar_file) self.logger.debug("Untar packages file %s" % dest_file) cmd = """ cd %s tar xzf %s cp -r %s/* %s rm -r %s """ % (tmp_dir, tar_file, src, dest, tmp_dir) os.system(cmd) self.logger.info("Downloaded %s to %s" % (src, _green(dest))) @_check_fabric def fetch_packages(self, dest=None): """Fetch Synnefo packages""" if dest is None: dest = self.config.get('Global', 'pkgs_dir') dest = os.path.abspath(os.path.expanduser(dest)) if not os.path.exists(dest): os.makedirs(dest) self.fetch_compressed("synnefo_build-area", dest) self.fetch_compressed("webclient_build-area", dest) self.logger.info("Downloaded debian packages to %s" % _green(dest)) def x2go_plugin(self, dest=None): """Produce an html page which will use the x2goplugin Arguments: dest -- The file where to save the page (String) """ output_str = """ <!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN"> <html> <head> <title>X2Go SynnefoCI Service</title> </head> <body onload="checkPlugin()"> <div id="x2goplugin"> <object src="location" type="application/x2go" name="x2goplugin" palette="background" height="100%" hspace="0" vspace="0" width="100%" x2goconfig=" session=X2Go-SynnefoCI-Session server={0} user={1} sshport={2} published=true autologin=true "> </object> </div> </body> </html> """.format(self.read_temp_config('server_ip'), self.read_temp_config('server_user'), self.read_temp_config('server_port')) if dest is None: dest = self.config.get('Global', 'x2go_plugin_file') self.logger.info("Writting x2go plugin html file to %s" % dest) fid = open(dest, 'w') fid.write(output_str) fid.close()
class Astakos(livetest.Generic): def setUp(self): self.cloud = 'cloud.%s' % self['testcloud'] self.client = AstakosClient( self[self.cloud, 'url'], self[self.cloud, 'token']) with open(self['astakos', 'details']) as f: self._astakos_details = eval(f.read()) def test_authenticate(self): self._test_0010_authenticate() def _test_0010_authenticate(self): r = self.client.authenticate() self.assert_dicts_are_equal(r, self._astakos_details) def test_get_services(self): self._test_0020_get_services() def _test_0020_get_services(self): for args in (tuple(), (self[self.cloud, 'token'],)): r = self.client.get_services(*args) services = self._astakos_details['access']['serviceCatalog'] self.assertEqual(len(services), len(r)) for i, service in enumerate(services): self.assert_dicts_are_equal(r[i], service) self.assertRaises(ClientError, self.client.get_services, 'wrong_token') def test_get_service_details(self): self._test_0020_get_service_details() def _test_0020_get_service_details(self): parsed_services = dict() for args in product( self._astakos_details['access']['serviceCatalog'], ([tuple(), (self[self.cloud, 'token'],)])): service = args[0] if service['type'] in parsed_services: continue r = self.client.get_service_details(service['type'], *args[1]) self.assert_dicts_are_equal(r, service) parsed_services[service['type']] = True self.assertRaises( ClientError, self.client.get_service_details, 'wrong_token') def test_get_service_endpoints(self): self._test_0020_get_service_endpoints() def _test_0020_get_service_endpoints(self): parsed_services = dict() for args in product( self._astakos_details['access']['serviceCatalog'], ([], [self[self.cloud, 'token']])): service = args[0] if service['type'] in parsed_services: continue for endpoint, with_id in product( service['endpoints'], (True, False)): vid = endpoint['versionId'] if ( with_id and endpoint['versionId']) else None end_args = [service['type'], vid] + args[1] r = self.client.get_service_endpoints(*end_args) self.assert_dicts_are_equal(r, endpoint) parsed_services[service['type']] = True self.assertRaises( ClientError, self.client.get_service_endpoints, 'wrong_token') def test_user_info(self): self._test_0020_user_info() def _test_0020_user_info(self): self.assertTrue(set([ 'roles', 'name', 'id']).issubset(self.client.user_info().keys())) def test_get(self): self._test_0020_get() def _test_0020_get(self): for term in ('id', 'name'): self.assertEqual( self.client.term(term, self[self.cloud, 'token']), self['astakos', term] or '') def test_list_users(self): self.client.authenticate() self._test_0020_list_users() def _test_0020_list_users(self): terms = set(['name', 'id']) uuid = 0 for r in self.client.list_users(): self.assertTrue(terms.issubset(r.keys())) self.assertTrue(uuid != r['id'] if uuid else True) uuid = r['id']
def main(): """Entry point for icaas-monitord""" args = get_args() manifest = read_manifest(args.manifest) if 'status' not in manifest['service']: sys.stderr.write('"status" is missing from the service section of the ' 'manifest') sys.exit(3) if 'insecure' in manifest and manifest['insecure'].lower() == 'true': verify = False else: verify = True report = Report(manifest['service']['status'], verify=verify, log=sys.stderr) def missing_key(key, section): """missing key message""" return "`%s' is missing from the `%s' section of the manifest" % \ (key, section) # Validate the manifest for key in 'url', 'token', 'log', 'status': if key not in manifest['service']: report.error(missing_key(key, 'service')) sys.exit(3) for key in 'url', 'name', 'object': if key not in manifest['image']: report.error(missing_key(key, 'image')) sys.exit(3) service = manifest['service'] try: container, logname = service['log'].split('/', 1) except ValueError: report.error('Incorrect format for log entry in manifest file') # Use the systems certificates https.patch_with_certs(CERTS) account = AstakosClient(service['url'], service['token']) try: account.authenticate() except AstakosClientError as err: report.error("Astakos: %s" % err) sys.exit(3) pithos = PithosClient( account.get_service_endpoints('object-store')['publicURL'], account.token, account.user_info['id'], container) if args.daemonize: daemon_context = daemon.DaemonContext(stdin=sys.stdin, stdout=sys.stdout, stderr=sys.stderr) daemon_context.open() with open(PID, 'w') as pid: pid.write("%d\n" % os.getpid()) try: # Export manifest to environment variables for section in manifest: for key, value in manifest[section].items(): name = "ICAAS_%s_%s" % (section.upper(), key.upper()) os.environ[name] = value # Use SIGHUP to unblock from the sleep if necessary signal.signal(signal.SIGHUP, lambda x, y: None) if 'ICAAS_MONITOR_SIGSTOP' in os.environ: # Tell service supervisor that we are ready. syslog.syslog( syslog.LOG_NOTICE, "Stopping with SIGSTOP as the " "environment variable ICAAS_MONITOR_SIGSTOP is defined") os.kill(os.getpid(), signal.SIGSTOP) del os.environ['ICAAS_MONITOR_SIGSTOP'] if do_main_loop(args.interval, pithos, logname): report.success() else: report.error("Image creation failed. Check the log for more info") finally: os.unlink(PID)
class AstakosClient(TestCase): cached = False def assert_dicts_are_equal(self, d1, d2): for k, v in d1.items(): self.assertTrue(k in d2) if isinstance(v, dict): self.assert_dicts_are_equal(v, d2[k]) else: self.assertEqual(unicode(v), unicode(d2[k])) def setUp(self): self.url = 'https://astakos.example.com' self.token = 'ast@k0sT0k3n==' from kamaki.clients.astakos import AstakosClient as AC self.client = AC(self.url, self.token) def tearDown(self): FR.json = example @patch('%s.LoggedAstakosClient.__init__' % astakos_pkg, return_value=None) @patch( '%s.LoggedAstakosClient.get_endpoints' % astakos_pkg, return_value=example) def _authenticate(self, get_endpoints, sac): r = self.client.authenticate() self.assertEqual( sac.mock_calls[-1], call(self.token, self.url, logger=getLogger('astakosclient'))) self.assertEqual(get_endpoints.mock_calls[-1], call()) return r def test_authenticate(self): r = self._authenticate() self.assert_dicts_are_equal(r, example) uuid = example['access']['user']['id'] self.assert_dicts_are_equal(self.client._uuids, {self.token: uuid}) self.assert_dicts_are_equal(self.client._cache, {uuid: r}) from astakosclient import AstakosClient as SAC self.assertTrue(isinstance(self.client._astakos[uuid], SAC)) self.assert_dicts_are_equal(self.client._uuids2usernames, {uuid: {}}) self.assert_dicts_are_equal(self.client._usernames2uuids, {uuid: {}}) def test_get_client(self): if not self.cached: self._authenticate() from astakosclient import AstakosClient as SNFAC self.assertTrue(self.client.get_client(), SNFAC) def test_get_token(self): self._authenticate() uuid = self.client._uuids.values()[0] self.assertEqual(self.client.get_token(uuid), self.token) def test_get_services(self): if not self.cached: self._authenticate() slist = self.client.get_services() self.assertEqual(slist, example['access']['serviceCatalog']) def test_get_service_details(self): if not self.cached: self._authenticate() stype = '#FAIL' self.assertRaises(ClientError, self.client.get_service_details, stype) stype = 'compute' expected = [s for s in example['access']['serviceCatalog'] if ( s['type'] == stype)] self.assert_dicts_are_equal( self.client.get_service_details(stype), expected[0]) def test_get_service_endpoints(self): if not self.cached: self._authenticate() stype, version = 'compute', 'V2' self.assertRaises( ClientError, self.client.get_service_endpoints, stype) expected = [s for s in example['access']['serviceCatalog'] if ( s['type'] == stype)] expected = [e for e in expected[0]['endpoints'] if ( e['versionId'] == version.lower())] self.assert_dicts_are_equal( self.client.get_service_endpoints(stype, version), expected[0]) def test_user_info(self): if not self.cached: self._authenticate() self.assertTrue(set(example['access']['user'].keys()).issubset( self.client.user_info().keys())) def test_item(self): if not self.cached: self._authenticate() for term, val in example['access']['user'].items(): self.assertEqual(self.client.term(term, self.token), val) self.assertTrue( example['access']['user']['email'][0] in self.client.term('email')) def test_list_users(self): if not self.cached: self._authenticate() FR.json = example self._authenticate() r = self.client.list_users() self.assertTrue(len(r) == 1) self.assertEqual(r[0]['auth_token'], self.token) @patch( '%s.LoggedAstakosClient.get_usernames' % astakos_pkg, return_value={42: 'username42', 43: 'username43'}) def test_uuids2usernames(self, get_usernames): from astakosclient import AstakosClientException self.assertRaises( AstakosClientException, self.client.uuids2usernames, [42, 43]) with patch( '%s.LoggedAstakosClient.__init__' % astakos_pkg, return_value=None) as sac: with patch( '%s.LoggedAstakosClient.get_endpoints' % astakos_pkg, return_value=example) as get_endpoints: r = self.client.uuids2usernames([42, 43]) self.assert_dicts_are_equal( r, {42: 'username42', 43: 'username43'}) self.assertEqual(sac.mock_calls[-1], call( self.token, self.url, logger=getLogger('astakosclient'))) self.assertEqual(get_endpoints.mock_calls[-1], call()) self.assertEqual(get_usernames.mock_calls[-1], call([42, 43])) @patch( '%s.LoggedAstakosClient.get_uuids' % astakos_pkg, return_value={'username42': 42, 'username43': 43}) def test_usernames2uuids(self, get_uuids): from astakosclient import AstakosClientException self.assertRaises( AstakosClientException, self.client.usernames2uuids, ['u1', 'u2']) with patch( '%s.LoggedAstakosClient.__init__' % astakos_pkg, return_value=None) as sac: with patch( '%s.LoggedAstakosClient.get_endpoints' % astakos_pkg, return_value=example) as get_endpoints: r = self.client.usernames2uuids(['u1', 'u2']) self.assert_dicts_are_equal( r, {'username42': 42, 'username43': 43}) self.assertEqual(sac.mock_calls[-1], call( self.token, self.url, logger=getLogger('astakosclient'))) self.assertEqual(get_endpoints.mock_calls[-1], call()) self.assertEqual(get_uuids.mock_calls[-1], call(['u1', 'u2']))
def main(): """Parse arguments, use kamaki to create cluster, setup using ansible playbooks""" (opts, args) = parse_arguments(sys.argv[1:]) global CYCLADES, TOKEN, my_vnat_network, my_network_client AUTHENTICATION_URL = opts.cyclades TOKEN = opts.token # Cleanup stale servers from previous runs if opts.show_stale: cleanup_servers(prefix=opts.prefix, delete_stale=opts.delete_stale) return 0 # Initialize a kamaki instance, get endpoints user = AstakosClient(AUTHENTICATION_URL, TOKEN) my_accountData = user.authenticate() endpoints = user.get_endpoints() cyclades_endpoints = user.get_endpoints('compute') cyclades_base_url = parseAstakosEndpoints(endpoints, 'cyclades_compute') cyclades_network_base_url = parseAstakosEndpoints(endpoints, 'cyclades_network') my_cyclades_client = CycladesClient(cyclades_base_url, TOKEN) my_compute_client = ComputeClient(cyclades_base_url, TOKEN) my_network_client = CycladesNetworkClient(cyclades_network_base_url, TOKEN) my_vnat_network = {} # check if 'Hadoop' vnat is created... hadoop_vnat_created = False my_network_dict = my_network_client.list_networks() for n in my_network_dict: if n['name'] == 'Hadoop': hadoop_vnat_created = True my_vnat_network = n # ...else create it if hadoop_vnat_created == False: log.info("Creating vNAT") my_vnat_network = my_network_client.create_network(type='MAC_FILTERED', name='Hadoop'); my_subnet = my_network_client.create_subnet(network_id=my_vnat_network['id'], cidr='192.168.0.0/24'); cnt = int(opts.clustersize) # calculate size of cluster into 'cnt' # Initialize nodes = [] masterName = '' # Create a file to store the root password for later use if not os.path.exists(opts.hadoop_dir+'/bak'): os.makedirs(opts.hadoop_dir+'/bak') pass_fname = opts.hadoop_dir+'/bak/adminPass'+str(datetime.now())[:19].replace(' ', '') adminPass_f = open(pass_fname, 'w') initialClusterSize = 0 server = {} if opts.extend == False: # Create master node (0th node) server = create_machine(opts, my_cyclades_client, 0) if server == {}: return else: servers = my_cyclades_client.list_servers(detail=True) cluster = [s for s in servers if s["name"].startswith(opts.prefix)] initialClusterSize = len(cluster) if initialClusterSize==0: log.info("Cluster cannot be expanded: it does not exist.") return servername = "%s-0" % (opts.prefix) masterName = servername nodes.append(server) # Create slave (worker) nodes if cnt>1 or opts.extend: startingOffset = 1 if opts.extend: startingOffset = initialClusterSize for i in xrange(startingOffset, initialClusterSize+cnt): server = {} server = create_machine(opts, my_cyclades_client, i) if server == {}: return; nodes.append(server) servername = "%s-%d" % (opts.prefix, i) # Write the root password to a file adminPass_f.write('machine = %s, password = %s\n' % (servername, server['adminPass'])) adminPass_f.close() # Setup Hadoop files and settings on all cluster nodes # Create the 'cluster' dictionary out of servers, with only Hadoop-relevant keys (name, ip, integer key) servers = my_cyclades_client.list_servers(detail=True) cluster = [s for s in my_cyclades_client.list_servers(detail=True) if s["name"].startswith(opts.prefix)] cluster0 = [(s["name"], s["attachments"], int(s["name"][s["name"].find('-')+1:])) for s in cluster] cluster0 = sorted(cluster0, key=lambda cluster0: cluster0[2]) cluster = [(cluster0[0][0], cluster0[0][1][2]["ipv4"], cluster0[0][2])] # master IP, different index cluster2 = [(s[0], s[1][1]['ipv4'], int(s[2])) for s in cluster0[1:]] # slave IPs cluster += cluster2 # Prepare Ansible-Hadoop config files (hosts, conf/slaves. vnat/etchosts) hosts = open(opts.hadoop_dir+'/hosts', 'w') hosts.write('[master]\n') etchosts = open(opts.hadoop_dir+'/vnat/etchosts', 'w') for i in xrange(0, initialClusterSize+cnt): for s in cluster: if s[0] == opts.prefix+"-"+str(i): if s[0] == masterName: hosts.write(s[1]+'\n\n'+'[slaves]\n') else: hosts.write(s[1]+'\n') etchosts.write(s[1]+'\t'+s[0]+'\n') hosts.close() etchosts.close() slaves = open(opts.hadoop_dir+'/vnat/slaves', 'w') for s in cluster[1:]: slaves.write(s[0]+'\n') slaves.close() # Execute respective ansible playbook if (opts.extend==False): cmd = "ansible-playbook hadoop_vnat.yml -i hosts -vv --extra-vars \""+"is_master=True, master_node="+cluster[0][0]+" master_ip="+cluster[0][1]+"\""+" -l master" print cmd retval = os.system(cmd) cmd = "ansible-playbook hadoop_vnat.yml -i hosts -vv --extra-vars \""+"is_slave=True, master_node="+cluster[0][0]+" master_ip="+cluster[0][1]+"\""+" -l slaves" print cmd retval = os.system(cmd) slave_ip_list = [] for i in xrange(1, cnt): slave_ip_list.append(cluster[i][0]) enable_ssh_login(cluster[0][1], [cluster[0][0]]) enable_ssh_login(cluster[0][1], slave_ip_list) else: hosts_latest = open(opts.hadoop_dir+'/hosts.latest', 'w') hosts_latest.write('[master]\n') hosts_latest.write(cluster[0][1]+'\n\n'+'[slaves]\n') for i in xrange(initialClusterSize, initialClusterSize+cnt): hosts_latest.write(cluster[i][1]+'\n') hosts_latest.close() # update etc/hosts in all nodes - TODO: de-duplicate entries cmd = "ansible-playbook hadoop_vnat.yml -i hosts -vv --extra-vars \""+"is_master=True, master_ip="+cluster[0][1]+"\""+" -t etchosts" print cmd retval = os.system(cmd) cmd = "ansible-playbook hadoop_vnat.yml -i hosts.latest -vv --extra-vars \""+"is_slave=True, master_node="+cluster[0][0]+" master_ip="+cluster[0][1]+"\""+" -l slaves" print cmd retval = os.system(cmd) slave_ip_list = [] for i in xrange(initialClusterSize, initialClusterSize+cnt): slave_ip_list.append(cluster[i][0]) print "slave_ip_list=", slave_ip_list enable_ssh_login(cluster[0][1], slave_ip_list) # Update conf/slaves in master cmd = "ansible-playbook hadoop_vnat.yml -i hosts -vv --extra-vars \""+"is_master=True, master_ip="+cluster[0][1]+"\""+" -l master -t slaves" print cmd retval = os.system(cmd) log.info("Done.")
class SynnefoCI(object): """SynnefoCI python class""" def __init__(self, config_file=None, build_id=None, cloud=None): """ Initialize SynnefoCI python class Setup logger, local_dir, config and kamaki """ # Setup logger self.logger = logging.getLogger('synnefo-ci') self.logger.setLevel(logging.DEBUG) handler1 = logging.StreamHandler(sys.stdout) handler1.setLevel(logging.DEBUG) handler1.addFilter(_InfoFilter()) handler1.setFormatter(_MyFormatter()) handler2 = logging.StreamHandler(sys.stderr) handler2.setLevel(logging.WARNING) handler2.setFormatter(_MyFormatter()) self.logger.addHandler(handler1) self.logger.addHandler(handler2) # Get our local dir self.ci_dir = os.path.dirname(os.path.abspath(__file__)) self.repo_dir = os.path.dirname(self.ci_dir) # Read config file if config_file is None: config_file = os.path.join(self.ci_dir, DEFAULT_CONFIG_FILE) config_file = os.path.abspath(config_file) self.config = ConfigParser() self.config.optionxform = str self.config.read(config_file) # Read temporary_config file self.temp_config_file = \ os.path.expanduser(self.config.get('Global', 'temporary_config')) self.temp_config = ConfigParser() self.temp_config.optionxform = str self.temp_config.read(self.temp_config_file) self.build_id = build_id if build_id is not None: self.logger.info("Will use \"%s\" as build id" % _green(self.build_id)) # Set kamaki cloud if cloud is not None: self.kamaki_cloud = cloud elif self.config.has_option("Deployment", "kamaki_cloud"): kamaki_cloud = self.config.get("Deployment", "kamaki_cloud") if kamaki_cloud == "": self.kamaki_cloud = None else: self.kamaki_cloud = None # Initialize variables self.fabric_installed = False self.kamaki_installed = False self.cyclades_client = None self.network_client = None self.compute_client = None self.image_client = None self.astakos_client = None def setup_kamaki(self): """Initialize kamaki Setup cyclades_client, image_client and compute_client """ config = kamaki_config.Config() if self.kamaki_cloud is None: try: self.kamaki_cloud = config.get("global", "default_cloud") except AttributeError: # Compatibility with kamaki version <=0.10 self.kamaki_cloud = config.get("global", "default_cloud") self.logger.info("Setup kamaki client, using cloud '%s'.." % self.kamaki_cloud) auth_url = config.get_cloud(self.kamaki_cloud, "url") self.logger.debug("Authentication URL is %s" % _green(auth_url)) token = config.get_cloud(self.kamaki_cloud, "token") #self.logger.debug("Token is %s" % _green(token)) self.astakos_client = AstakosClient(auth_url, token) endpoints = self.astakos_client.authenticate() cyclades_url = get_endpoint_url(endpoints, "compute") self.logger.debug("Cyclades API url is %s" % _green(cyclades_url)) self.cyclades_client = CycladesClient(cyclades_url, token) self.cyclades_client.CONNECTION_RETRY_LIMIT = 2 network_url = get_endpoint_url(endpoints, "network") self.logger.debug("Network API url is %s" % _green(network_url)) self.network_client = CycladesNetworkClient(network_url, token) self.network_client.CONNECTION_RETRY_LIMIT = 2 image_url = get_endpoint_url(endpoints, "image") self.logger.debug("Images API url is %s" % _green(image_url)) self.image_client = ImageClient(cyclades_url, token) self.image_client.CONNECTION_RETRY_LIMIT = 2 compute_url = get_endpoint_url(endpoints, "compute") self.logger.debug("Compute API url is %s" % _green(compute_url)) self.compute_client = ComputeClient(compute_url, token) self.compute_client.CONNECTION_RETRY_LIMIT = 2 def _wait_transition(self, server_id, current_status, new_status): """Wait for server to go from current_status to new_status""" self.logger.debug("Waiting for server to become %s" % new_status) timeout = self.config.getint('Global', 'build_timeout') sleep_time = 5 while True: server = self.cyclades_client.get_server_details(server_id) if server['status'] == new_status: return server elif timeout < 0: self.logger.error("Waiting for server to become %s timed out" % new_status) self.destroy_server(False) sys.exit(1) elif server['status'] == current_status: # Sleep for #n secs and continue timeout = timeout - sleep_time time.sleep(sleep_time) else: self.logger.error("Server failed with status %s" % server['status']) self.destroy_server(False) sys.exit(1) @_check_kamaki def destroy_server(self, wait=True): """Destroy slave server""" server_id = int(self.read_temp_config('server_id')) fips = [ f for f in self.network_client.list_floatingips() if str(f['instance_id']) == str(server_id) ] self.logger.info("Destoying server with id %s " % server_id) self.cyclades_client.delete_server(server_id) if wait: self._wait_transition(server_id, "ACTIVE", "DELETED") for fip in fips: self.logger.info("Destroying floating ip %s", fip['floating_ip_address']) self.network_client.delete_floatingip(fip['id']) def _create_floating_ip(self): """Create a new floating ip""" networks = self.network_client.list_networks(detail=True) pub_nets = [ n for n in networks if n['SNF:floating_ip_pool'] and n['public'] ] for pub_net in pub_nets: # Try until we find a public network that is not full try: fip = self.network_client.create_floatingip(pub_net['id']) except ClientError as err: self.logger.warning("%s: %s", err.message, err.details) continue self.logger.debug("Floating IP %s with id %s created", fip['floating_ip_address'], fip['id']) return fip self.logger.error("No mor IP addresses available") sys.exit(1) def _create_port(self, floating_ip): """Create a new port for our floating IP""" net_id = floating_ip['floating_network_id'] self.logger.debug("Creating a new port to network with id %s", net_id) fixed_ips = [{'ip_address': floating_ip['floating_ip_address']}] port = self.network_client.create_port(net_id, device_id=None, fixed_ips=fixed_ips) return port @_check_kamaki # Too many local variables. pylint: disable-msg=R0914 def create_server(self, image=None, flavor=None, ssh_keys=None, server_name=None): """Create slave server""" self.logger.info("Create a new server..") # Find a build_id to use self._create_new_build_id() # Find an image to use image_id = self._find_image(image) # Find a flavor to use flavor_id = self._find_flavor(flavor) # Create Server networks = [] if self.config.get("Deployment", "allocate_floating_ip") == "True": fip = self._create_floating_ip() port = self._create_port(fip) networks.append({'port': port['id']}) private_networks = self.config.get('Deployment', 'private_networks') if private_networks: private_networks = [p.strip() for p in private_networks.split(",")] networks.extend([{"uuid": uuid} for uuid in private_networks]) if server_name is None: server_name = self.config.get("Deployment", "server_name") server_name = "%s(BID: %s)" % (server_name, self.build_id) server = self.cyclades_client.create_server(server_name, flavor_id, image_id, networks=networks) server_id = server['id'] self.write_temp_config('server_id', server_id) self.logger.debug("Server got id %s" % _green(server_id)) server_user = server['metadata']['users'] self.write_temp_config('server_user', server_user) self.logger.debug("Server's admin user is %s" % _green(server_user)) server_passwd = server['adminPass'] self.write_temp_config('server_passwd', server_passwd) server = self._wait_transition(server_id, "BUILD", "ACTIVE") self._get_server_ip_and_port(server, private_networks) self._copy_ssh_keys(ssh_keys) # Setup Firewall self.setup_fabric() self.logger.info("Setup firewall") accept_ssh_from = self.config.get('Global', 'accept_ssh_from') if accept_ssh_from != "": self.logger.debug("Block ssh except from %s" % accept_ssh_from) cmd = """ local_ip=$(/sbin/ifconfig eth0 | grep 'inet addr:' | \ cut -d':' -f2 | cut -d' ' -f1) iptables -A INPUT -s localhost -j ACCEPT iptables -A INPUT -s $local_ip -j ACCEPT iptables -A INPUT -s {0} -p tcp --dport 22 -j ACCEPT iptables -A INPUT -p tcp --dport 22 -j DROP """.format(accept_ssh_from) _run(cmd, False) # Setup apt, download packages self.logger.debug("Setup apt. Install x2goserver and firefox") cmd = """ echo 'APT::Install-Suggests "false";' >> /etc/apt/apt.conf echo 'precedence ::ffff:0:0/96 100' >> /etc/gai.conf apt-get update apt-get install curl --yes --force-yes echo -e "\n\n{0}" >> /etc/apt/sources.list # Synnefo repo's key curl https://dev.grnet.gr/files/apt-grnetdev.pub | apt-key add - # X2GO Key apt-key adv --recv-keys --keyserver keys.gnupg.net E1F958385BFE2B6E apt-get install x2go-keyring --yes --force-yes apt-get update apt-get install x2goserver x2goserver-xsession \ iceweasel --yes --force-yes # xterm published application echo '[Desktop Entry]' > /usr/share/applications/xterm.desktop echo 'Name=XTerm' >> /usr/share/applications/xterm.desktop echo 'Comment=standard terminal emulator for the X window system' >> \ /usr/share/applications/xterm.desktop echo 'Exec=xterm' >> /usr/share/applications/xterm.desktop echo 'Terminal=false' >> /usr/share/applications/xterm.desktop echo 'Type=Application' >> /usr/share/applications/xterm.desktop echo 'Encoding=UTF-8' >> /usr/share/applications/xterm.desktop echo 'Icon=xterm-color_48x48' >> /usr/share/applications/xterm.desktop echo 'Categories=System;TerminalEmulator;' >> \ /usr/share/applications/xterm.desktop """.format(self.config.get('Global', 'apt_repo')) _run(cmd, False) def _find_flavor(self, flavor=None): """Find a suitable flavor to use Search by name (reg expression) or by id """ # Get a list of flavors from config file flavors = self.config.get('Deployment', 'flavors').split(",") if flavor is not None: # If we have a flavor_name to use, add it to our list flavors.insert(0, flavor) list_flavors = self.compute_client.list_flavors() for flv in flavors: flv_type, flv_value = parse_typed_option(option="flavor", value=flv) if flv_type == "name": # Filter flavors by name self.logger.debug("Trying to find a flavor with name \"%s\"" % flv_value) list_flvs = \ [f for f in list_flavors if re.search(flv_value, f['name'], flags=re.I) is not None] elif flv_type == "id": # Filter flavors by id self.logger.debug("Trying to find a flavor with id \"%s\"" % flv_value) list_flvs = \ [f for f in list_flavors if str(f['id']) == flv_value] else: self.logger.error("Unrecognized flavor type %s" % flv_type) # Check if we found one if list_flvs: self.logger.debug( "Will use \"%s\" with id \"%s\"" % (_green(list_flvs[0]['name']), _green(list_flvs[0]['id']))) return list_flvs[0]['id'] self.logger.error("No matching flavor found.. aborting") sys.exit(1) def _find_image(self, image=None): """Find a suitable image to use In case of search by name, the image has to belong to one of the `DEFAULT_SYSTEM_IMAGES_UUID' users. In case of search by id it only has to exist. """ # Get a list of images from config file images = self.config.get('Deployment', 'images').split(",") if image is not None: # If we have an image from command line, add it to our list images.insert(0, image) auth = self.astakos_client.authenticate() user_uuid = auth["access"]["token"]["tenant"]["id"] list_images = self.image_client.list_public(detail=True)['images'] for img in images: img_type, img_value = parse_typed_option(option="image", value=img) if img_type == "name": # Filter images by name self.logger.debug("Trying to find an image with name \"%s\"" % img_value) accepted_uuids = DEFAULT_SYSTEM_IMAGES_UUID + [user_uuid] list_imgs = \ [i for i in list_images if i['user_id'] in accepted_uuids and re.search(img_value, i['name'], flags=re.I) is not None] elif img_type == "id": # Filter images by id self.logger.debug("Trying to find an image with id \"%s\"" % img_value) list_imgs = \ [i for i in list_images if i['id'].lower() == img_value.lower()] else: self.logger.error("Unrecognized image type %s" % img_type) sys.exit(1) # Check if we found one if list_imgs: self.logger.debug( "Will use \"%s\" with id \"%s\"" % (_green(list_imgs[0]['name']), _green(list_imgs[0]['id']))) return list_imgs[0]['id'] # We didn't found one self.logger.error("No matching image found.. aborting") sys.exit(1) def _get_server_ip_and_port(self, server, private_networks): """Compute server's IPv4 and ssh port number""" self.logger.info("Get server connection details..") if private_networks: # Choose the networks that belong to private_networks networks = [ n for n in server['attachments'] if n['network_id'] in private_networks ] else: # Choose the networks that are public networks = [ n for n in server['attachments'] if self.network_client.get_network_details(n['network_id']) ['public'] ] # Choose the networks with IPv4 networks = [n for n in networks if n['ipv4']] # Use the first network as IPv4 server_ip = networks[0]['ipv4'] if (".okeanos.io" in self.cyclades_client.base_url or ".demo.synnefo.org" in self.cyclades_client.base_url): tmp1 = int(server_ip.split(".")[2]) tmp2 = int(server_ip.split(".")[3]) server_ip = "gate.okeanos.io" server_port = 10000 + tmp1 * 256 + tmp2 else: server_port = 22 self.write_temp_config('server_ip', server_ip) self.logger.debug("Server's IPv4 is %s" % _green(server_ip)) self.write_temp_config('server_port', server_port) self.logger.debug("Server's ssh port is %s" % _green(server_port)) ssh_command = "ssh -p %s %s@%s" \ % (server_port, server['metadata']['users'], server_ip) self.logger.debug("Access server using \"%s\"" % (_green(ssh_command))) @_check_fabric def _copy_ssh_keys(self, ssh_keys): """Upload/Install ssh keys to server""" self.logger.debug("Check for authentication keys to use") if ssh_keys is None: ssh_keys = self.config.get("Deployment", "ssh_keys") if ssh_keys != "": ssh_keys = os.path.expanduser(ssh_keys) self.logger.debug("Will use \"%s\" authentication keys file" % _green(ssh_keys)) keyfile = '/tmp/%s.pub' % fabric.env.user _run('mkdir -p ~/.ssh && chmod 700 ~/.ssh', False) if ssh_keys.startswith("http://") or \ ssh_keys.startswith("https://") or \ ssh_keys.startswith("ftp://"): cmd = """ apt-get update apt-get install wget --yes --force-yes wget {0} -O {1} --no-check-certificate """.format(ssh_keys, keyfile) _run(cmd, False) elif os.path.exists(ssh_keys): _put(ssh_keys, keyfile) else: self.logger.debug("No ssh keys found") return _run('cat %s >> ~/.ssh/authorized_keys' % keyfile, False) _run('rm %s' % keyfile, False) self.logger.debug("Uploaded ssh authorized keys") else: self.logger.debug("No ssh keys found") def _create_new_build_id(self): """Find a uniq build_id to use""" with filelocker.lock("%s.lock" % self.temp_config_file, filelocker.LOCK_EX): # Read temp_config again to get any new entries self.temp_config.read(self.temp_config_file) # Find a uniq build_id to use if self.build_id is None: ids = self.temp_config.sections() if ids: max_id = int(max(self.temp_config.sections(), key=int)) self.build_id = max_id + 1 else: self.build_id = 1 self.logger.debug("Will use \"%s\" as build id" % _green(self.build_id)) # Create a new section try: self.temp_config.add_section(str(self.build_id)) except DuplicateSectionError: msg = ("Build id \"%s\" already in use. " + "Please use a uniq one or cleanup \"%s\" file.\n") \ % (self.build_id, self.temp_config_file) self.logger.error(msg) sys.exit(1) creation_time = \ time.strftime("%a, %d %b %Y %X", time.localtime()) self.temp_config.set(str(self.build_id), "created", str(creation_time)) # Write changes back to temp config file with open(self.temp_config_file, 'wb') as tcf: self.temp_config.write(tcf) def write_temp_config(self, option, value): """Write changes back to config file""" # Acquire the lock to write to temp_config_file with filelocker.lock("%s.lock" % self.temp_config_file, filelocker.LOCK_EX): # Read temp_config again to get any new entries self.temp_config.read(self.temp_config_file) self.temp_config.set(str(self.build_id), option, str(value)) curr_time = time.strftime("%a, %d %b %Y %X", time.localtime()) self.temp_config.set(str(self.build_id), "modified", curr_time) # Write changes back to temp config file with open(self.temp_config_file, 'wb') as tcf: self.temp_config.write(tcf) def read_temp_config(self, option): """Read from temporary_config file""" # If build_id is None use the latest one if self.build_id is None: ids = self.temp_config.sections() if ids: self.build_id = int(ids[-1]) else: self.logger.error("No sections in temporary config file") sys.exit(1) self.logger.debug("Will use \"%s\" as build id" % _green(self.build_id)) # Read specified option return self.temp_config.get(str(self.build_id), option) def setup_fabric(self): """Setup fabric environment""" self.logger.info("Setup fabric parameters..") fabric.env.user = self.read_temp_config('server_user') fabric.env.host_string = self.read_temp_config('server_ip') fabric.env.port = int(self.read_temp_config('server_port')) fabric.env.password = self.read_temp_config('server_passwd') fabric.env.connection_attempts = 10 fabric.env.shell = "/bin/bash -c" fabric.env.disable_known_hosts = True fabric.env.output_prefix = None def _check_hash_sum(self, localfile, remotefile): """Check hash sums of two files""" self.logger.debug("Check hash sum for local file %s" % localfile) hash1 = os.popen("sha256sum %s" % localfile).read().split(' ')[0] self.logger.debug("Local file has sha256 hash %s" % hash1) self.logger.debug("Check hash sum for remote file %s" % remotefile) hash2 = _run("sha256sum %s" % remotefile, False) hash2 = hash2.split(' ')[0] self.logger.debug("Remote file has sha256 hash %s" % hash2) if hash1 != hash2: self.logger.error("Hashes differ.. aborting") sys.exit(1) @_check_fabric def clone_repo(self, local_repo=False): """Clone Synnefo repo from slave server""" self.logger.info("Configure repositories on remote server..") self.logger.debug("Install/Setup git") cmd = """ apt-get install git --yes --force-yes git config --global user.name {0} git config --global user.email {1} """.format(self.config.get('Global', 'git_config_name'), self.config.get('Global', 'git_config_mail')) _run(cmd, False) # Clone synnefo_repo synnefo_branch = self.clone_synnefo_repo(local_repo=local_repo) # Clone pithos-web-client self.clone_pithos_webclient_repo(synnefo_branch) @_check_fabric def clone_synnefo_repo(self, local_repo=False): """Clone Synnefo repo to remote server""" # Find synnefo_repo and synnefo_branch to use synnefo_repo = self.config.get('Global', 'synnefo_repo') synnefo_branch = self.config.get("Global", "synnefo_branch") if synnefo_branch == "": synnefo_branch = \ subprocess.Popen( ["git", "rev-parse", "--abbrev-ref", "HEAD"], stdout=subprocess.PIPE).communicate()[0].strip() if synnefo_branch == "HEAD": synnefo_branch = \ subprocess.Popen( ["git", "rev-parse", "--short", "HEAD"], stdout=subprocess.PIPE).communicate()[0].strip() self.logger.debug("Will use branch \"%s\"" % _green(synnefo_branch)) if local_repo or synnefo_repo == "": # Use local_repo self.logger.debug("Push local repo to server") # Firstly create the remote repo _run("git init synnefo", False) # Then push our local repo over ssh # We have to pass some arguments to ssh command # namely to disable host checking. (temp_ssh_file_handle, temp_ssh_file) = tempfile.mkstemp() os.close(temp_ssh_file_handle) # XXX: git push doesn't read the password cmd = """ echo 'exec ssh -o "StrictHostKeyChecking no" \ -o "UserKnownHostsFile /dev/null" \ -q "$@"' > {4} chmod u+x {4} export GIT_SSH="{4}" echo "{0}" | git push --quiet --mirror ssh://{1}@{2}:{3}/~/synnefo rm -f {4} """.format(fabric.env.password, fabric.env.user, fabric.env.host_string, fabric.env.port, temp_ssh_file) os.system(cmd) else: # Clone Synnefo from remote repo self.logger.debug("Clone synnefo from %s" % synnefo_repo) self._git_clone(synnefo_repo) # Checkout the desired synnefo_branch self.logger.debug("Checkout \"%s\" branch/commit" % synnefo_branch) cmd = """ cd synnefo for branch in `git branch -a | grep remotes | grep -v HEAD`; do git branch --track ${branch##*/} $branch done git checkout %s """ % (synnefo_branch) _run(cmd, False) return synnefo_branch @_check_fabric def clone_pithos_webclient_repo(self, synnefo_branch): """Clone Pithos WebClient repo to remote server""" # Find pithos_webclient_repo and pithos_webclient_branch to use pithos_webclient_repo = \ self.config.get('Global', 'pithos_webclient_repo') pithos_webclient_branch = \ self.config.get('Global', 'pithos_webclient_branch') # Clone pithos-webclient from remote repo self.logger.debug("Clone pithos-webclient from %s" % pithos_webclient_repo) self._git_clone(pithos_webclient_repo) # Track all pithos-webclient branches cmd = """ cd pithos-web-client for branch in `git branch -a | grep remotes | grep -v HEAD`; do git branch --track ${branch##*/} $branch > /dev/null 2>&1 done git --no-pager branch --no-color """ webclient_branches = _run(cmd, False) webclient_branches = webclient_branches.split() # If we have pithos_webclient_branch in config file use this one # else try to use the same branch as synnefo_branch # else use an appropriate one. if pithos_webclient_branch == "": if synnefo_branch in webclient_branches: pithos_webclient_branch = synnefo_branch else: # If synnefo_branch starts with one of # 'master', 'hotfix'; use the master branch if synnefo_branch.startswith('master') or \ synnefo_branch.startswith('hotfix'): pithos_webclient_branch = "master" # If synnefo_branch starts with one of # 'develop', 'feature'; use the develop branch elif synnefo_branch.startswith('develop') or \ synnefo_branch.startswith('feature'): pithos_webclient_branch = "develop" else: self.logger.warning( "Cannot determine which pithos-web-client branch to " "use based on \"%s\" synnefo branch. " "Will use develop." % synnefo_branch) pithos_webclient_branch = "develop" # Checkout branch self.logger.debug("Checkout \"%s\" branch" % _green(pithos_webclient_branch)) cmd = """ cd pithos-web-client git checkout {0} """.format(pithos_webclient_branch) _run(cmd, False) def _git_clone(self, repo): """Clone repo to remote server Currently clonning from code.grnet.gr can fail unexpectedly. So retry!! """ cloned = False for i in range(1, 11): try: _run("git clone %s" % repo, False) cloned = True break except BaseException: self.logger.warning("Clonning failed.. retrying %s/10" % i) if not cloned: self.logger.error("Can not clone repo.") sys.exit(1) @_check_fabric def build_packages(self): """Build packages needed by Synnefo software""" self.logger.info("Install development packages") cmd = """ apt-get update apt-get install zlib1g-dev dpkg-dev debhelper git-buildpackage \ python-dev python-all python-pip ant --yes --force-yes pip install -U devflow """ _run(cmd, False) # Patch pydist bug if self.config.get('Global', 'patch_pydist') == "True": self.logger.debug("Patch pydist.py module") cmd = r""" sed -r -i 's/(\(\?P<name>\[A-Za-z\]\[A-Za-z0-9_\.)/\1\\\-/' \ /usr/share/python/debpython/pydist.py """ _run(cmd, False) # Build synnefo packages self.build_synnefo() # Build pithos-web-client packages self.build_pithos_webclient() @_check_fabric def build_synnefo(self): """Build Synnefo packages""" self.logger.info("Build Synnefo packages..") cmd = """ devflow-autopkg snapshot -b ~/synnefo_build-area --no-sign """ with fabric.cd("synnefo"): _run(cmd, True) # Install snf-deploy package self.logger.debug("Install snf-deploy package") cmd = """ dpkg -i snf-deploy*.deb apt-get -f install --yes --force-yes """ with fabric.cd("synnefo_build-area"): with fabric.settings(warn_only=True): _run(cmd, True) # Setup synnefo packages for snf-deploy self.logger.debug("Copy synnefo debs to snf-deploy packages dir") cmd = """ cp ~/synnefo_build-area/*.deb /var/lib/snf-deploy/packages/ """ _run(cmd, False) @_check_fabric def build_pithos_webclient(self): """Build pithos-web-client packages""" self.logger.info("Build pithos-web-client packages..") cmd = """ devflow-autopkg snapshot -b ~/webclient_build-area --no-sign """ with fabric.cd("pithos-web-client"): _run(cmd, True) # Setup pithos-web-client packages for snf-deploy self.logger.debug("Copy webclient debs to snf-deploy packages dir") cmd = """ cp ~/webclient_build-area/*.deb /var/lib/snf-deploy/packages/ """ _run(cmd, False) @_check_fabric def build_documentation(self): """Build Synnefo documentation""" self.logger.info("Build Synnefo documentation..") _run("pip install -U Sphinx", False) with fabric.cd("synnefo"): _run( "devflow-update-version; " "./ci/make_docs.sh synnefo_documentation", False) def fetch_documentation(self, dest=None): """Fetch Synnefo documentation""" self.logger.info("Fetch Synnefo documentation..") if dest is None: dest = "synnefo_documentation" dest = os.path.abspath(dest) if not os.path.exists(dest): os.makedirs(dest) self.fetch_compressed("synnefo/synnefo_documentation", dest) self.logger.info("Downloaded documentation to %s" % _green(dest)) @_check_fabric def deploy_synnefo(self, schema=None): """Deploy Synnefo using snf-deploy""" self.logger.info("Deploy Synnefo..") if schema is None: schema = self.config.get('Global', 'schema') self.logger.debug("Will use \"%s\" schema" % _green(schema)) schema_dir = os.path.join(self.ci_dir, "schemas/%s" % schema) if not (os.path.exists(schema_dir) and os.path.isdir(schema_dir)): raise ValueError("Unknown schema: %s" % schema) self.logger.debug("Upload schema files to server") _put(os.path.join(schema_dir, "*"), "/etc/snf-deploy/") self.logger.debug("Change password in nodes.conf file") cmd = """ sed -i 's/^password =.*/password = {0}/' /etc/snf-deploy/nodes.conf """.format(fabric.env.password) _run(cmd, False) self.logger.debug("Run snf-deploy") cmd = """ snf-deploy keygen --force snf-deploy --disable-colors --autoconf all """ _run(cmd, True) @_check_fabric def unit_test(self): """Run Synnefo unit test suite""" self.logger.info("Run Synnefo unit test suite") component = self.config.get('Unit Tests', 'component') self.logger.debug("Install needed packages") cmd = """ pip install -U mock pip install -U factory_boy pip install -U nose """ _run(cmd, False) self.logger.debug("Upload tests.sh file") unit_tests_file = os.path.join(self.ci_dir, "tests.sh") _put(unit_tests_file, ".") self.logger.debug("Run unit tests") cmd = """ bash tests.sh {0} """.format(component) _run(cmd, True) @_check_fabric def run_burnin(self): """Run burnin functional test suite""" self.logger.info("Run Burnin functional test suite") cmd = """ auth_url=$(grep -e '^url =' .kamakirc | cut -d' ' -f3) token=$(grep -e '^token =' .kamakirc | cut -d' ' -f3) images_user=$(kamaki image list -l | grep owner | \ cut -d':' -f2 | tr -d ' ') snf-burnin --auth-url=$auth_url --token=$token {0} BurninExitStatus=$? exit $BurninExitStatus """.format(self.config.get('Burnin', 'cmd_options')) _run(cmd, True) @_check_fabric def fetch_compressed(self, src, dest=None): """Create a tarball and fetch it locally""" self.logger.debug("Creating tarball of %s" % src) basename = os.path.basename(src) tar_file = basename + ".tgz" cmd = "tar czf %s %s" % (tar_file, src) _run(cmd, False) if not os.path.exists(dest): os.makedirs(dest) tmp_dir = tempfile.mkdtemp() fabric.get(tar_file, tmp_dir) dest_file = os.path.join(tmp_dir, tar_file) self._check_hash_sum(dest_file, tar_file) self.logger.debug("Untar packages file %s" % dest_file) cmd = """ cd %s tar xzf %s cp -r %s/* %s rm -r %s """ % (tmp_dir, tar_file, src, dest, tmp_dir) os.system(cmd) self.logger.info("Downloaded %s to %s" % (src, _green(dest))) @_check_fabric def fetch_packages(self, dest=None): """Fetch Synnefo packages""" if dest is None: dest = self.config.get('Global', 'pkgs_dir') dest = os.path.abspath(os.path.expanduser(dest)) if not os.path.exists(dest): os.makedirs(dest) self.fetch_compressed("synnefo_build-area", dest) self.fetch_compressed("webclient_build-area", dest) self.logger.info("Downloaded debian packages to %s" % _green(dest)) def x2go_plugin(self, dest=None): """Produce an html page which will use the x2goplugin Arguments: dest -- The file where to save the page (String) """ output_str = """ <!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN"> <html> <head> <title>X2Go SynnefoCI Service</title> </head> <body onload="checkPlugin()"> <div id="x2goplugin"> <object src="location" type="application/x2go" name="x2goplugin" palette="background" height="100%" hspace="0" vspace="0" width="100%" x2goconfig=" session=X2Go-SynnefoCI-Session server={0} user={1} sshport={2} published=true autologin=true "> </object> </div> </body> </html> """.format(self.read_temp_config('server_ip'), self.read_temp_config('server_user'), self.read_temp_config('server_port')) if dest is None: dest = self.config.get('Global', 'x2go_plugin_file') self.logger.info("Writting x2go plugin html file to %s" % dest) fid = open(dest, 'w') fid.write(output_str) fid.close()
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED # AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE # POSSIBILITY OF SUCH DAMAGE. # # The views and conclusions contained in the software and # documentation are those of the authors and should not be # interpreted as representing official policies, either expressed # or implied, of GRNET S.A. from kamaki.clients.astakos import AstakosClient AUTHENTICATION_URL = "https://astakos.example.com/identity/v2.0" TOKEN = "User-Token" astakos = AstakosClient(AUTHENTICATION_URL, TOKEN) # Get user uuid user = astakos.authenticate() uuid = user['access']['user']['id'] # Get resources assigned on my personal (system) project all_resources = astakos.get_quotas() for project, resources in all_resources.items(): print "For project with id {project_id}".format(project_id=project) for resource, values in resources.items(): print " {resource}: {used}/{limit}".format( resource=resource, used=values["usage"], limit=values["limit"])
class SynnefoCI(object): """SynnefoCI python class""" def __init__(self, config_file=None, build_id=None, cloud=None): """ Initialize SynnefoCI python class Setup logger, local_dir, config and kamaki """ # Setup logger self.logger = logging.getLogger("synnefo-ci") self.logger.setLevel(logging.DEBUG) handler1 = logging.StreamHandler(sys.stdout) handler1.setLevel(logging.DEBUG) handler1.addFilter(_InfoFilter()) handler1.setFormatter(_MyFormatter()) handler2 = logging.StreamHandler(sys.stderr) handler2.setLevel(logging.WARNING) handler2.setFormatter(_MyFormatter()) self.logger.addHandler(handler1) self.logger.addHandler(handler2) # Get our local dir self.ci_dir = os.path.dirname(os.path.abspath(__file__)) self.repo_dir = os.path.dirname(self.ci_dir) # Read config file if config_file is None: config_file = os.path.join(self.ci_dir, DEFAULT_CONFIG_FILE) config_file = os.path.abspath(config_file) self.config = ConfigParser() self.config.optionxform = str self.config.read(config_file) # Read temporary_config file self.temp_config_file = os.path.expanduser(self.config.get("Global", "temporary_config")) self.temp_config = ConfigParser() self.temp_config.optionxform = str self.temp_config.read(self.temp_config_file) self.build_id = build_id if build_id is not None: self.logger.info('Will use "%s" as build id' % _green(self.build_id)) # Set kamaki cloud if cloud is not None: self.kamaki_cloud = cloud elif self.config.has_option("Deployment", "kamaki_cloud"): self.kamaki_cloud = self.config.get("Deployment", "kamaki_cloud") if self.kamaki_cloud == "": self.kamaki_cloud = None else: self.kamaki_cloud = None # Initialize variables self.fabric_installed = False self.kamaki_installed = False self.cyclades_client = None self.network_client = None self.compute_client = None self.image_client = None self.astakos_client = None def setup_kamaki(self): """Initialize kamaki Setup cyclades_client, image_client and compute_client """ # Patch kamaki for SSL verification _kamaki_ssl(ignore_ssl=IGNORE_SSL) config = kamaki_config.Config() if self.kamaki_cloud is None: try: self.kamaki_cloud = config.get("global", "default_cloud") except AttributeError: # Compatibility with kamaki version <=0.10 self.kamaki_cloud = config.get("global", "default_cloud") self.logger.info("Setup kamaki client, using cloud '%s'.." % self.kamaki_cloud) auth_url = config.get_cloud(self.kamaki_cloud, "url") self.logger.debug("Authentication URL is %s" % _green(auth_url)) token = config.get_cloud(self.kamaki_cloud, "token") # self.logger.debug("Token is %s" % _green(token)) self.astakos_client = AstakosClient(auth_url, token) endpoints = self.astakos_client.authenticate() cyclades_url = get_endpoint_url(endpoints, "compute") self.logger.debug("Cyclades API url is %s" % _green(cyclades_url)) self.cyclades_client = CycladesClient(cyclades_url, token) self.cyclades_client.CONNECTION_RETRY_LIMIT = 2 network_url = get_endpoint_url(endpoints, "network") self.logger.debug("Network API url is %s" % _green(network_url)) self.network_client = CycladesNetworkClient(network_url, token) self.network_client.CONNECTION_RETRY_LIMIT = 2 image_url = get_endpoint_url(endpoints, "image") self.logger.debug("Images API url is %s" % _green(image_url)) self.image_client = ImageClient(cyclades_url, token) self.image_client.CONNECTION_RETRY_LIMIT = 2 compute_url = get_endpoint_url(endpoints, "compute") self.logger.debug("Compute API url is %s" % _green(compute_url)) self.compute_client = ComputeClient(compute_url, token) self.compute_client.CONNECTION_RETRY_LIMIT = 2 __quota_cache = None def _get_available_project(self, skip_config=False, **resources): self.project_uuid = None if self.config.has_option("Deployment", "project"): self.project_uuid = self.config.get("Deployment", "project").strip() or None # user requested explicit project if self.project_uuid and not skip_config: return self.project_uuid def _filter_projects(_project): uuid, project_quota = _project can_fit = False for resource, required in resources.iteritems(): # transform dots in order to permit direct keyword # arguments to be used. # (cyclades__disk=1) -> 'cyclades.disk': 1 resource = resource.replace("__", ".") project_resource = project_quota.get(resource) if not project_resource: raise Exception("Requested resource does not exist %s" % resource) plimit, ppending, pusage, musage, mlimit, mpending = project_resource.values() pavailable = plimit - ppending - pusage mavailable = mlimit - mpending - musage can_fit = (pavailable - required) >= 0 and (mavailable - required) >= 0 if not can_fit: return None return uuid self.__quota_cache = quota = self.__quota_cache or self.astakos_client.get_quotas() projects = filter(bool, map(_filter_projects, quota.iteritems())) if not len(projects): raise Exception("No project available for %r" % resources) return projects[0] def _wait_transition(self, server_id, current_status, new_status): """Wait for server to go from current_status to new_status""" self.logger.debug("Waiting for server to become %s" % new_status) timeout = self.config.getint("Global", "build_timeout") sleep_time = 5 while True: server = self.cyclades_client.get_server_details(server_id) if server["status"] == new_status: return server elif timeout < 0: self.logger.error("Waiting for server to become %s timed out" % new_status) self.destroy_server(False) sys.exit(1) elif server["status"] == current_status: # Sleep for #n secs and continue timeout = timeout - sleep_time time.sleep(sleep_time) else: self.logger.error("Server failed with status %s" % server["status"]) self.destroy_server(False) sys.exit(1) @_check_kamaki def destroy_server(self, wait=True): """Destroy slave server""" server_id = int(self.read_temp_config("server_id")) fips = [f for f in self.network_client.list_floatingips() if str(f["instance_id"]) == str(server_id)] self.logger.info("Destoying server with id %s " % server_id) self.cyclades_client.delete_server(server_id) if wait: self._wait_transition(server_id, "ACTIVE", "DELETED") for fip in fips: self.logger.info("Destroying floating ip %s", fip["floating_ip_address"]) self.network_client.delete_floatingip(fip["id"]) # pylint: disable= no-self-use @_check_fabric def shell_connect(self): """Open shell to remote server""" fabric.open_shell("export TERM=xterm") def _create_floating_ip(self): """Create a new floating ip""" project_id = self._get_available_project(cyclades__floating_ip=1) networks = self.network_client.list_networks(detail=True) pub_nets = [n for n in networks if n["SNF:floating_ip_pool"] and n["public"]] for pub_net in pub_nets: # Try until we find a public network that is not full try: fip = self.network_client.create_floatingip(pub_net["id"], project_id=project_id) except ClientError as err: self.logger.warning("%s", str(err.message).strip()) continue self.logger.debug("Floating IP %s with id %s created", fip["floating_ip_address"], fip["id"]) return fip self.logger.error("No more IP addresses available") sys.exit(1) def _create_port(self, floating_ip): """Create a new port for our floating IP""" net_id = floating_ip["floating_network_id"] self.logger.debug("Creating a new port to network with id %s", net_id) fixed_ips = [{"ip_address": floating_ip["floating_ip_address"]}] port = self.network_client.create_port(net_id, device_id=None, fixed_ips=fixed_ips) return port @_check_kamaki # Too many local variables. pylint: disable-msg=R0914 def create_server(self, image=None, flavor=None, ssh_keys=None, server_name=None): """Create slave server""" self.logger.info("Create a new server..") # Find a build_id to use self._create_new_build_id() # Find an image to use image_id = self._find_image(image) # Find a flavor to use flavor_id = self._find_flavor(flavor) # get available project flavor = self.cyclades_client.get_flavor_details(flavor_id) quota = { "cyclades.disk": flavor["disk"] * 1024 ** 3, "cyclades.ram": flavor["ram"] * 1024 ** 2, "cyclades.cpu": flavor["vcpus"], "cyclades.vm": 1, } project_id = self._get_available_project(**quota) # Create Server networks = [] if self.config.get("Deployment", "allocate_floating_ip") == "True": fip = self._create_floating_ip() port = self._create_port(fip) networks.append({"port": port["id"]}) private_networks = self.config.get("Deployment", "private_networks") if private_networks: private_networks = [p.strip() for p in private_networks.split(",")] networks.extend([{"uuid": uuid} for uuid in private_networks]) if server_name is None: server_name = self.config.get("Deployment", "server_name") server_name = "%s(BID: %s)" % (server_name, self.build_id) server = self.cyclades_client.create_server( server_name, flavor_id, image_id, networks=networks, project_id=project_id ) server_id = server["id"] self.write_temp_config("server_id", server_id) self.logger.debug("Server got id %s" % _green(server_id)) # An image may have more than one user. Choose the first one. server_user = server["metadata"]["users"].split(" ")[0] self.write_temp_config("server_user", server_user) self.logger.debug("Server's admin user is %s" % _green(server_user)) server_passwd = server["adminPass"] self.write_temp_config("server_passwd", server_passwd) server = self._wait_transition(server_id, "BUILD", "ACTIVE") self._get_server_ip_and_port(server, private_networks) self._copy_ssh_keys(ssh_keys) # Setup Firewall self.setup_fabric() self.logger.info("Setup firewall") accept_ssh_from = self.config.get("Global", "accept_ssh_from") if accept_ssh_from != "": self.logger.debug("Block ssh except from %s" % accept_ssh_from) cmd = """ local_ip=$(/sbin/ifconfig eth0 | grep 'inet addr:' | \ cut -d':' -f2 | cut -d' ' -f1) iptables -A INPUT -s localhost -j ACCEPT iptables -A INPUT -s $local_ip -j ACCEPT iptables -A INPUT -s {0} -p tcp --dport 22 -j ACCEPT iptables -A INPUT -p tcp --dport 22 -j DROP """.format( accept_ssh_from ) _run(cmd, False) # Setup apt, download packages self.logger.debug("Setup apt") cmd = """ echo 'APT::Install-Suggests "false";' >> /etc/apt/apt.conf echo 'Package: python-gevent' >> /etc/apt/preferences.d/00-gevent echo 'Pin: release o=Debian' >> /etc/apt/preferences.d/00-gevent echo 'Pin-Priority: 990' >> /etc/apt/preferences.d/00-gevent echo 'precedence ::ffff:0:0/96 100' >> /etc/gai.conf apt-get update apt-get install -q=2 curl --yes --force-yes echo -e "{0}" >> /etc/apt/sources.list.d/synnefo.wheezy.list # Synnefo repo's key curl https://dev.grnet.gr/files/apt-grnetdev.pub | apt-key add - """.format( self.config.get("Global", "apt_repo") ) _run(cmd, False) cmd = """ # X2GO Key apt-key adv --recv-keys --keyserver keys.gnupg.net E1F958385BFE2B6E apt-get install x2go-keyring --yes --force-yes apt-get update apt-get install x2goserver x2goserver-xsession \ iceweasel --yes --force-yes # xterm published application echo '[Desktop Entry]' > /usr/share/applications/xterm.desktop echo 'Name=XTerm' >> /usr/share/applications/xterm.desktop echo 'Comment=standard terminal emulator for the X window system' >> \ /usr/share/applications/xterm.desktop echo 'Exec=xterm' >> /usr/share/applications/xterm.desktop echo 'Terminal=false' >> /usr/share/applications/xterm.desktop echo 'Type=Application' >> /usr/share/applications/xterm.desktop echo 'Encoding=UTF-8' >> /usr/share/applications/xterm.desktop echo 'Icon=xterm-color_48x48' >> /usr/share/applications/xterm.desktop echo 'Categories=System;TerminalEmulator;' >> \ /usr/share/applications/xterm.desktop """ if self.config.get("Global", "setup_x2go") == "True": self.logger.debug("Install x2goserver and firefox") _run(cmd, False) def _find_flavor(self, flavor=None): """Find a suitable flavor to use Search by name (reg expression) or by id """ def _is_true(value): """Boolean or string value that represents a bool""" if isinstance(value, bool): return value elif isinstance(value, str): return value in ["True", "true"] else: self.logger.error("Unrecognized boolean value %s" % value) return False # Get a list of flavors from config file flavors = self.config.get("Deployment", "flavors").split(",") if flavor is not None: # If we have a flavor_name to use, add it to our list flavors.insert(0, flavor) list_flavors = self.compute_client.list_flavors(detail=True) for flv in flavors: flv_type, flv_value = parse_typed_option(option="flavor", value=flv) if flv_type == "name": # Filter flavors by name self.logger.debug('Trying to find a flavor with name "%s"' % flv_value) list_flvs = [f for f in list_flavors if re.search(flv_value, f["name"], flags=re.I) is not None] elif flv_type == "id": # Filter flavors by id self.logger.debug('Trying to find a flavor with id "%s"' % flv_value) list_flvs = [f for f in list_flavors if str(f["id"]) == flv_value] else: self.logger.error("Unrecognized flavor type %s" % flv_type) # Check if we found one list_flvs = [f for f in list_flvs if _is_true(f["SNF:allow_create"])] if list_flvs: self.logger.debug( 'Will use "%s" with id "%s"' % (_green(list_flvs[0]["name"]), _green(list_flvs[0]["id"])) ) return list_flvs[0]["id"] self.logger.error("No matching flavor found.. aborting") sys.exit(1) def _find_image(self, image=None): """Find a suitable image to use In case of search by name, the image has to belong to one of the `DEFAULT_SYSTEM_IMAGES_UUID' users. In case of search by id it only has to exist. """ # Get a list of images from config file images = self.config.get("Deployment", "images").split(",") if image is not None: # If we have an image from command line, add it to our list images.insert(0, image) auth = self.astakos_client.authenticate() user_uuid = auth["access"]["token"]["tenant"]["id"] list_images = self.image_client.list_public(detail=True)["images"] for img in images: img_type, img_value = parse_typed_option(option="image", value=img) if img_type == "name": # Filter images by name self.logger.debug('Trying to find an image with name "%s"' % img_value) accepted_uuids = DEFAULT_SYSTEM_IMAGES_UUID + [user_uuid] list_imgs = [ i for i in list_images if i["user_id"] in accepted_uuids and re.search(img_value, i["name"], flags=re.I) is not None ] elif img_type == "id": # Filter images by id self.logger.debug('Trying to find an image with id "%s"' % img_value) list_imgs = [i for i in list_images if i["id"].lower() == img_value.lower()] else: self.logger.error("Unrecognized image type %s" % img_type) sys.exit(1) # Check if we found one if list_imgs: self.logger.debug( 'Will use "%s" with id "%s"' % (_green(list_imgs[0]["name"]), _green(list_imgs[0]["id"])) ) return list_imgs[0]["id"] # We didn't found one self.logger.error("No matching image found.. aborting") sys.exit(1) def _get_server_ip_and_port(self, server, private_networks): """Compute server's IPv4 and ssh port number""" self.logger.info("Get server connection details..") if private_networks: # Choose the networks that belong to private_networks networks = [n for n in server["attachments"] if n["network_id"] in private_networks] else: # Choose the networks that are public networks = [ n for n in server["attachments"] if self.network_client.get_network_details(n["network_id"])["public"] ] # Choose the networks with IPv4 networks = [n for n in networks if n["ipv4"]] # Use the first network as IPv4 server_ip = networks[0]["ipv4"] # Check if config has ssh_port option and if so, use that port. server_port = self.config.get("Deployment", "ssh_port") if not server_port: # No ssh port given. Get it from API (SNF:port_forwarding) if "22" in server["SNF:port_forwarding"]: server_ip = server["SNF:port_forwarding"]["22"]["host"] server_port = int(server["SNF:port_forwarding"]["22"]["port"]) else: server_port = 22 self.write_temp_config("server_ip", server_ip) self.logger.debug("Server's IPv4 is %s" % _green(server_ip)) self.write_temp_config("server_port", server_port) self.logger.debug("Server's ssh port is %s" % _green(server_port)) ssh_command = "ssh -p %s %s@%s" % (server_port, server["metadata"]["users"], server_ip) self.logger.debug('Access server using "%s"' % (_green(ssh_command))) @_check_fabric def _copy_ssh_keys(self, ssh_keys): """Upload/Install ssh keys to server""" self.logger.debug("Check for authentication keys to use") if ssh_keys is None: ssh_keys = self.config.get("Deployment", "ssh_keys") if ssh_keys != "": ssh_keys = os.path.expanduser(ssh_keys) self.logger.debug('Will use "%s" authentication keys file' % _green(ssh_keys)) keyfile = "/tmp/%s.pub" % fabric.env.user _run("mkdir -p ~/.ssh && chmod 700 ~/.ssh", False) if ssh_keys.startswith("http://") or ssh_keys.startswith("https://") or ssh_keys.startswith("ftp://"): cmd = """ apt-get update apt-get install wget --yes --force-yes wget {0} -O {1} --no-check-certificate """.format( ssh_keys, keyfile ) _run(cmd, False) elif os.path.exists(ssh_keys): _put(ssh_keys, keyfile) else: self.logger.debug("No ssh keys found") return _run("cat %s >> ~/.ssh/authorized_keys" % keyfile, False) _run("rm %s" % keyfile, False) self.logger.debug("Uploaded ssh authorized keys") else: self.logger.debug("No ssh keys found") def _create_new_build_id(self): """Find a uniq build_id to use""" with filelocker.lock("%s.lock" % self.temp_config_file, filelocker.LOCK_EX): # Read temp_config again to get any new entries self.temp_config.read(self.temp_config_file) # Find a uniq build_id to use if self.build_id is None: ids = self.temp_config.sections() if ids: max_id = int(max(self.temp_config.sections(), key=int)) self.build_id = max_id + 1 else: self.build_id = 1 self.logger.debug('Will use "%s" as build id' % _green(self.build_id)) # Create a new section try: self.temp_config.add_section(str(self.build_id)) except DuplicateSectionError: msg = ('Build id "%s" already in use. ' + 'Please use a uniq one or cleanup "%s" file.\n') % ( self.build_id, self.temp_config_file, ) self.logger.error(msg) sys.exit(1) creation_time = time.strftime("%a, %d %b %Y %X", time.localtime()) self.temp_config.set(str(self.build_id), "created", str(creation_time)) # Write changes back to temp config file with open(self.temp_config_file, "wb") as tcf: self.temp_config.write(tcf) def write_temp_config(self, option, value): """Write changes back to config file""" # Acquire the lock to write to temp_config_file with filelocker.lock("%s.lock" % self.temp_config_file, filelocker.LOCK_EX): # Read temp_config again to get any new entries self.temp_config.read(self.temp_config_file) self.temp_config.set(str(self.build_id), option, str(value)) curr_time = time.strftime("%a, %d %b %Y %X", time.localtime()) self.temp_config.set(str(self.build_id), "modified", curr_time) # Write changes back to temp config file with open(self.temp_config_file, "wb") as tcf: self.temp_config.write(tcf) def read_temp_config(self, option): """Read from temporary_config file""" # If build_id is None use the latest one if self.build_id is None: ids = self.temp_config.sections() if ids: self.build_id = int(ids[-1]) else: self.logger.error("No sections in temporary config file") sys.exit(1) self.logger.debug('Will use "%s" as build id' % _green(self.build_id)) # Read specified option return self.temp_config.get(str(self.build_id), option) def setup_fabric(self): """Setup fabric environment""" self.logger.info("Setup fabric parameters..") fabric.env.user = self.read_temp_config("server_user") fabric.env.host_string = self.read_temp_config("server_ip") fabric.env.port = int(self.read_temp_config("server_port")) fabric.env.password = self.read_temp_config("server_passwd") fabric.env.connection_attempts = 10 fabric.env.shell = "/bin/bash -c" fabric.env.disable_known_hosts = True fabric.env.output_prefix = None def _check_hash_sum(self, localfile, remotefile): """Check hash sums of two files""" self.logger.debug("Check hash sum for local file %s" % localfile) hash1 = os.popen("sha256sum %s" % localfile).read().split(" ")[0] self.logger.debug("Local file has sha256 hash %s" % hash1) self.logger.debug("Check hash sum for remote file %s" % remotefile) hash2 = _run("sha256sum %s" % remotefile, False) hash2 = hash2.split(" ")[0] self.logger.debug("Remote file has sha256 hash %s" % hash2) if hash1 != hash2: self.logger.error("Hashes differ.. aborting") sys.exit(1) @_check_fabric def clone_repo(self, synnefo_repo=None, synnefo_branch=None, local_repo=False, pull_request=None): """Clone Synnefo repo from slave server""" self.logger.info("Configure repositories on remote server..") self.logger.debug("Install/Setup git") cmd = """ apt-get install git --yes --force-yes git config --global user.name {0} git config --global user.email {1} """.format( self.config.get("Global", "git_config_name"), self.config.get("Global", "git_config_mail") ) _run(cmd, False) # Clone synnefo_repo synnefo_branch = self.clone_synnefo_repo( synnefo_repo=synnefo_repo, synnefo_branch=synnefo_branch, local_repo=local_repo, pull_request=pull_request ) # Clone pithos-web-client if self.config.get("Global", "build_pithos_webclient") == "True": # Clone pithos-web-client self.clone_pithos_webclient_repo(synnefo_branch) @_check_fabric def clone_synnefo_repo(self, synnefo_repo=None, synnefo_branch=None, local_repo=False, pull_request=None): """Clone Synnefo repo to remote server""" assert pull_request is None or (synnefo_branch is None and synnefo_repo is None) pull_repo = None if pull_request is not None: # Get a Github pull request and run the testsuite in # a sophisticated way. # Sophisticated means that it will not just check the remote branch # from which the pull request originated. Instead it will checkout # the branch for which the pull request is indented (e.g. # grnet:develop) and apply the pull request over it. This way it # checks the pull request against the branch this pull request # targets. m = re.search("github.com/([^/]+)/([^/]+)/pull/(\d+)", pull_request) if m is None: self.logger.error("Couldn't find a valid GitHub pull request" " URL") sys.exit(1) group = m.group(1) repo = m.group(2) pull_number = m.group(3) # Construct api url api_url = "/repos/%s/%s/pulls/%s" % (group, repo, pull_number) headers = {"User-Agent": "snf-ci"} # Get pull request info try: conn = httplib.HTTPSConnection("api.github.com") conn.request("GET", api_url, headers=headers) response = conn.getresponse() payload = json.load(response) synnefo_repo = payload["base"]["repo"]["html_url"] synnefo_branch = payload["base"]["ref"] pull_repo = (payload["head"]["repo"]["html_url"], payload["head"]["ref"]) finally: conn.close() # Find synnefo_repo and synnefo_branch to use if synnefo_repo is None: synnefo_repo = self.config.get("Global", "synnefo_repo") if synnefo_branch is None: synnefo_branch = self.config.get("Global", "synnefo_branch") if synnefo_branch == "": synnefo_branch = ( subprocess.Popen(["git", "rev-parse", "--abbrev-ref", "HEAD"], stdout=subprocess.PIPE) .communicate()[0] .strip() ) if synnefo_branch == "HEAD": synnefo_branch = ( subprocess.Popen(["git", "rev-parse", "--short", "HEAD"], stdout=subprocess.PIPE) .communicate()[0] .strip() ) self.logger.debug('Will use branch "%s"' % _green(synnefo_branch)) if local_repo or synnefo_repo == "": # Use local_repo self.logger.debug("Push local repo to server") # Firstly create the remote repo _run("git init %s/synnefo" % work_dir, False) # Create a symlink to the userdir _run("ln -s %s/synnefo ~/synnefo" % work_dir, False) # Then push our local repo over ssh # We have to pass some arguments to ssh command # namely to disable host checking. (temp_ssh_file_handle, temp_ssh_file) = tempfile.mkstemp() os.close(temp_ssh_file_handle) # XXX: git push doesn't read the password cmd = """ echo 'exec ssh -o "StrictHostKeyChecking no" \ -o "UserKnownHostsFile /dev/null" \ -q "$@"' > {5} chmod u+x {5} export GIT_SSH="{5}" echo "{0}" | git push -q --mirror ssh://{1}@{2}:{3}{4}/synnefo #echo "{0}" | git push -q --mirror ssh://{1}@{2}:{3}/~/synnefo rm -f {5} """.format( fabric.env.password, fabric.env.user, fabric.env.host_string, fabric.env.port, work_dir, temp_ssh_file ) os.system(cmd) else: # Clone Synnefo from remote repo self.logger.debug("Clone synnefo from %s" % synnefo_repo) self._git_clone(synnefo_repo, directory="%s/synnefo" % work_dir) # Checkout the desired synnefo_branch self.logger.debug('Checkout "%s" branch/commit' % synnefo_branch) cmd = """ cd %s/synnefo # Squelch the error message about pushing to master. # Keep default behaviour but hide the error message. git config receive.denyCurrentBranch refuse #cd synnefo for branch in `git branch -a | grep remotes | grep -v HEAD`; do git branch --track ${branch##*/} $branch done git checkout %s """ % ( work_dir, synnefo_branch, ) _run(cmd, False) # Apply a Github pull request if pull_repo is not None: self.logger.debug("Apply patches from pull request %s", pull_number) cmd = """ cd %s/synnefo git pull --no-edit --no-rebase {0} {1} """.format( work_dir, pull_repo[0], pull_repo[1] ) _run(cmd, False) return synnefo_branch @_check_fabric def clone_pithos_webclient_repo(self, synnefo_branch): """Clone Pithos WebClient repo to remote server""" # Find pithos_webclient_repo and pithos_webclient_branch to use pithos_webclient_repo = self.config.get("Global", "pithos_webclient_repo") pithos_webclient_branch = self.config.get("Global", "pithos_webclient_branch") # Clone pithos-webclient from remote repo self.logger.debug("Clone pithos-webclient from %s" % pithos_webclient_repo) self._git_clone(pithos_webclient_repo, directory="%s/pithos-web-client" % work_dir) # Track all pithos-webclient branches cmd = ( """ cd %s/pithos-web-client for branch in `git branch -a | grep remotes | grep -v HEAD`; do git branch --track ${branch##*/} $branch > /dev/null 2>&1 done git --no-pager branch --no-color """ % work_dir ) webclient_branches = _run(cmd, False) webclient_branches = webclient_branches.split() # If we have pithos_webclient_branch in config file use this one # else try to use the same branch as synnefo_branch # else use an appropriate one. if pithos_webclient_branch == "": if synnefo_branch in webclient_branches: pithos_webclient_branch = synnefo_branch else: # If synnefo_branch starts with one of # 'master', 'hotfix'; use the master branch if synnefo_branch.startswith("master") or synnefo_branch.startswith("hotfix"): pithos_webclient_branch = "master" # If synnefo_branch starts with one of # 'develop', 'feature'; use the develop branch elif synnefo_branch.startswith("develop") or synnefo_branch.startswith("feature"): pithos_webclient_branch = "develop" else: self.logger.warning( "Cannot determine which pithos-web-client branch to " 'use based on "%s" synnefo branch. ' "Will use develop." % synnefo_branch ) pithos_webclient_branch = "develop" # Checkout branch self.logger.debug('Checkout "%s" branch' % _green(pithos_webclient_branch)) cmd = """ cd {0}/pithos-web-client git checkout {1} """.format( work_dir, pithos_webclient_branch ) _run(cmd, False) def _git_clone(self, repo, directory=""): """Clone repo to remote server Currently clonning from code.grnet.gr can fail unexpectedly. So retry!! """ cloned = False for i in range(1, 11): try: _run("git clone %s %s" % (repo, directory), False) cloned = True break except BaseException: self.logger.warning("Clonning failed.. retrying %s/10" % i) if not cloned: self.logger.error("Can not clone repo.") sys.exit(1) @_check_fabric def build_packages(self): """Build packages needed by Synnefo software""" self.logger.info("Install development packages") cmd = """ apt-get update apt-get install zlib1g-dev dpkg-dev debhelper git-buildpackage \ python-dev python-all python-pip ant --yes --force-yes pip install -U devflow """ _run(cmd, False) # Patch pydist bug if self.config.get("Global", "patch_pydist") == "True": self.logger.debug("Patch pydist.py module") cmd = r""" sed -r -i 's/(\(\?P<name>\[A-Za-z\]\[A-Za-z0-9_\.)/\1\\\-/' \ /usr/share/python/debpython/pydist.py """ _run(cmd, False) # Build synnefo packages self.build_synnefo() # Build pithos-web-client packages if self.config.get("Global", "build_pithos_webclient") == "True": self.build_pithos_webclient() @_check_fabric def build_synnefo(self): """Build Synnefo packages""" self.logger.info("Build Synnefo packages..") cmd = ( """ devflow-autopkg snapshot -b %s/synnefo_build-area --no-sign """ % work_dir ) with fabric.cd("%s/synnefo" % work_dir): _run(cmd, True) # Install snf-deploy package self.logger.debug("Install snf-deploy package") cmd = """ dpkg -i snf-deploy*.deb apt-get -f install --yes --force-yes snf-deploy keygen """ with fabric.cd("%s/synnefo_build-area" % work_dir): with fabric.settings(warn_only=True): _run(cmd, True) # Setup synnefo packages for snf-deploy self.logger.debug("Copy synnefo debs to snf-deploy packages dir") cmd = ( """ cp %s/synnefo_build-area/*.deb /var/lib/snf-deploy/packages/ """ % work_dir ) _run(cmd, False) @_check_fabric def build_pithos_webclient(self): """Build pithos-web-client packages""" self.logger.info("Build pithos-web-client packages..") cmd = ( """ devflow-autopkg snapshot -b %s/webclient_build-area --no-sign """ % work_dir ) with fabric.cd("%s/pithos-web-client" % work_dir): _run(cmd, True) # Setup pithos-web-client packages for snf-deploy self.logger.debug("Copy webclient debs to snf-deploy packages dir") cmd = ( """ cp %s/webclient_build-area/*.deb /var/lib/snf-deploy/packages/ """ % work_dir ) _run(cmd, False) @_check_fabric def build_documentation(self): """Build Synnefo documentation""" self.logger.info("Build Synnefo documentation..") _run("pip install -U Sphinx", False) with fabric.cd("%s/synnefo" % work_dir): _run("devflow-update-version; " "./ci/make_docs.sh synnefo_documentation", False) def fetch_documentation(self, dest=None): """Fetch Synnefo documentation""" self.logger.info("Fetch Synnefo documentation..") if dest is None: dest = "synnefo_documentation" dest = os.path.abspath(dest) if not os.path.exists(dest): os.makedirs(dest) self.fetch_compressed("%s/synnefo/synnefo_documentation" % work_dir, dest) self.logger.info("Downloaded documentation to %s" % _green(dest)) @_check_fabric def deploy_synnefo(self, schema=None): """Deploy Synnefo using snf-deploy""" self.logger.info("Deploy Synnefo..") if schema is None: schema = self.config.get("Global", "schema") self.logger.debug('Will use "%s" schema' % _green(schema)) self.logger.debug("Update schema files to server") cmd = """ schema_dir="{0}/synnefo/ci/schemas/{1}" if [ -d "$schema_dir" ]; then cp "$schema_dir"/* /etc/snf-deploy/ else echo "$schema_dir" does not exist exit 1 fi """.format( work_dir, schema ) _run(cmd, False) self.logger.debug("Change password in nodes.conf file") cmd = """ sed -i 's/^password =.*/password = {0}/' /etc/snf-deploy/nodes.conf """.format( fabric.env.password ) _run(cmd, False) self.logger.debug("Run snf-deploy") cmd = """ snf-deploy --disable-colors --autoconf synnefo """ _run(cmd, True) @_check_fabric def unit_test(self): """Run Synnefo unit test suite""" self.logger.info("Run Synnefo unit test suite") component = self.config.get("Unit Tests", "component") self.logger.debug("Install needed packages") cmd = """ pip install -U funcsigs mock==1.1.2 factory_boy==2.4.1 nose coverage """ _run(cmd, False) self.logger.debug("Upload tests.sh file") unit_tests_file = os.path.join(self.ci_dir, "tests.sh") _put(unit_tests_file, ".") self.logger.debug("Run unit tests") cmd = """ bash tests.sh {0} """.format( component ) _run(cmd, True) @_check_fabric def run_burnin(self): """Run burnin functional test suite""" self.logger.info("Run Burnin functional test suite") cmd = """ auth_url=$(grep -e '^url =' .kamakirc | cut -d' ' -f3) token=$(grep -e '^token =' .kamakirc | cut -d' ' -f3) images_user=$(kamaki image list -l | grep owner | \ cut -d':' -f2 | tr -d ' ') snf-burnin --auth-url=$auth_url --token=$token {0} BurninExitStatus=$? exit $BurninExitStatus """.format( self.config.get("Burnin", "cmd_options") ) _run(cmd, True) @_check_fabric def fetch_compressed(self, src, dest=None): """Create a tarball and fetch it locally""" self.logger.debug("Creating tarball of %s" % src) basename = os.path.basename(src) tar_file = basename + ".tgz" cmd = "tar czf %s %s" % (tar_file, src) _run(cmd, False) if not os.path.exists(dest): os.makedirs(dest) tmp_dir = tempfile.mkdtemp() fabric.get(tar_file, tmp_dir) dest_file = os.path.join(tmp_dir, tar_file) self._check_hash_sum(dest_file, tar_file) self.logger.debug("Untar packages file %s" % dest_file) cmd = """ cd %s tar xzf %s cp -r %s/%s/* %s rm -r %s """ % ( tmp_dir, tar_file, tmp_dir, src, dest, tmp_dir, ) os.system(cmd) self.logger.info("Downloaded %s to %s" % (src, _green(dest))) @_check_fabric def fetch_packages(self, dest=None): """Fetch Synnefo packages""" if dest is None: dest = self.config.get("Global", "pkgs_dir") dest = os.path.abspath(os.path.expanduser(dest)) if not os.path.exists(dest): os.makedirs(dest) self.fetch_compressed("%s/synnefo_build-area" % work_dir, dest) if self.config.get("Global", "build_pithos_webclient") == "True": self.fetch_compressed("%s/webclient_build-area" % work_dir, dest) self.logger.info("Downloaded debian packages to %s" % _green(dest)) def x2go_plugin(self, dest=None): """Produce an html page which will use the x2goplugin Arguments: dest -- The file where to save the page (String) """ output_str = """ <!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN"> <html> <head> <title>X2Go SynnefoCI Service</title> </head> <body onload="checkPlugin()"> <div id="x2goplugin"> <object src="location" type="application/x2go" name="x2goplugin" palette="background" height="100%" hspace="0" vspace="0" width="100%" x2goconfig=" session=X2Go-SynnefoCI-Session server={0} user={1} sshport={2} published=true autologin=true "> </object> </div> </body> </html> """.format( self.read_temp_config("server_ip"), self.read_temp_config("server_user"), self.read_temp_config("server_port"), ) if dest is None: dest = self.config.get("Global", "x2go_plugin_file") self.logger.info("Writting x2go plugin html file to %s" % dest) fid = open(dest, "w") fid.write(output_str) fid.close()