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