def execute(self, cmd, dryrun=None):
   """Wrapper to execute code.
   """
   if dryrun is None:
     dryrun = self.dryrun()
   deb("Project Executing code (dryrun={}): '''{}'''".format(dryrun, cmd))
   if dryrun:
     print "#DRYRUN# Wont execute this: '''{}'''".format(cmd)
     return "None(_DRYRUN_)"
   else:
       deb("PWARNING, executing: {}".format(cmd))
       ret = os.system(cmd)
       print "[PCMD] return={}".format(ret)
       return ret
   return None
Пример #2
0
def getConfigYaml(ymlfile=default_ymlfile):
  """Returns a dict with configuration.

  Also provides correct messaging in case of error/
  """
  deb("Conf file: {}".format(ymlfile))
  conf = {}
  try:
    yml = open(ymlfile, 'r')
    whole_conf = yaml.load(yml)
    deb( whole_conf)
    conf = whole_conf['production']
    deb( conf)
  except (OSError, IOError) as e:
    print """YAML not found. Maybe you forgot to do this:
    1. cp {f}.dist {f}
    2. vi {f}                 # edit as needed
    """.format(f=ymlfile)
    exit(2)
  except Exception as e:
    print "Generic exception: {}".format(e)
    exit(3)
  explaination = CheckValidity(conf)
  if explaination != '':
    print "Configuration is invalid: {}".format(explaination)
    exit(4)
  return conf
Пример #3
0
def gcutil_addinstance(project, name, description, 
  public_ip = False,
  tags = [],
  zone = None, 
  machine_type = None,            
  network = None,
  startup_script = None,
  disk = None,
  disks = [] ,
  image = None,
  metadata = {},
  service_account_scopes = None,
  persistent_boot_disk = False,
  additional_options = ''
  ):
  '''Adds an instance of gcutil (for the moment using the bash script, in the future using
  directly the API written - putacaso - in python!

  Please forgive me, this code is getting more and more unmantainable.
  '''
  print "= Project (should be an object): {} =".format(project)
  if (tags.__class__ != list):
    print "Sorry, I need a list for 'tags', not a %s" % (tags.__class__)
    exit(1)
  if (disks.__class__ != list):
    print "Sorry, I need a list for 'disks', not a %s" % (tags.__class__)
    exit(1)
  if machine_type is None:
    machine_type = project.default('machine_type')
  if image is None:
    image = project.default('image')
  if zone is None:
    zone = project.default('zone')
  if network is None:
    network = project.default('network')

  if startup_script is None:
     # Check if there is a file called DIRECT, if not, it will opt for storage and do nothing (common-startuip-script takes care of it)
     direct_file_path = './addons/{addon}.d/direct.{vmname}.sh'.format(addon=project.addon, vmname=name)
     if os.path.isfile(direct_file_path):
       #print "\n\n   File found! DEBUG: using it instead: ", direct_file_path
       startup_script = direct_file_path
     else:
       #pyellow("NOT FOUND LOCALLY: "+direct_file_path)
       startup_script = project.default('startup_script')
    
  
  # Using sets to guarantee unicity, then moving back to array
  all_tags = set(tags)
  all_tags.update(project.default('tags'))
  all_tags.update(["gwv%s" % str(lib_ver_common).replace('.','') ])
  all_tags = list(all_tags) # transofmring back to array

  date = str(datetime.now())

  if service_account_scopes is None:
    service_account_scopes = project.default('service_account_scopes')

  metadata['original-vm-name'] = name         # without prefix
  prefixed_name = project.default('vm_prefix') + name

  # Addons...
  public_ip_addon = '--external_ip_address=ephemeral' if public_ip == True else '--external_ip_address=none'
  if (public_ip.__class__ == str):
    public_ip_addon = '--external_ip_address=%s' % public_ip
  image_addon = '--image=%s' % image if image else ''
  disk_addon = ('--disk=%s ' % disk) if disk else ''
  for disk2 in disks:
    disk_addon += ('--disk=%s ' % disk2) if disk2 else ''
  
  deb(disk_addon)

  persistent_boot_disk_opts = '--persistent_boot_disk' if persistent_boot_disk else "--nopersistent_boot_disk"
  
  metadata_addon = ''
  metadata.update(project.metadata())
  for k in metadata.keys():
    metadata_addon = metadata_addon + "--metadata='{}:{}' ".format(k, metadata[k])

  addinstance_opts = """--tags='{tags}' \
  --zone='{zone}' \
  --service_account_scopes='{service_account_scopes}' \
  --machine_type='{mt}' \
  --metadata_from_file=startup-script:{startup} \
  --metadata=startup-metadata:project:{project_id}:maybeIssuesWithNumbers \
  --metadata=date_creation:'{date}' \
  {metadata_addon} \
  --network='{net}' \
  {disk_addon} \
  {public_ip_addon} \
  {image_addon} \
  {persistent_boot_disk_opts} \
  {additional_options} \
""".format(
    tags=','.join(all_tags), 
    zone=zone, 
    mt=machine_type, 
    startup=startup_script, 
    project_id=project.project_id, 
    date=date, 
    net=network, 
    disk_addon=disk_addon, 
    public_ip_addon=public_ip_addon,
    image_addon=image_addon,
    persistent_boot_disk_opts=persistent_boot_disk_opts,
    additional_options=additional_options, 
    metadata_addon=metadata_addon,
    service_account_scopes=','.join(service_account_scopes),
  )

  command = '''gcutil --project {} addinstance '{}' --description='[{}] {}' {}  '''.format(
      project.project_id, prefixed_name, project.addon, description, addinstance_opts)
  project.execute(command)
 def __del__(self):
   """Should be the destructor."""
   deb("+ ProjectInitiator Destructor: {}".format(self))
   if self.post_install:
     self.post_install()