def create_image(event, context):
if "body" not in event or event.get("body") is None:
return Response.error(
"Missing required parameters: account, region, id")
event = json.loads(event.get("body"))
account_name = event.get("account", None)
region = event.get("region", None)
instance_id = event.get("id", None)
image_name = event.get("image_name", None)
logs.info(
"Handler",
"start: {}, {}, {}".format(account_name, region, instance_id),
)
try:
_, instances = _build_instance_model(account_name, region)
except Exception as error:
logs.info("Handler", "Error: start setup {}".format(error.args[0]))
return Response.error(error.args[0])
try:
instances.create_image(instance_id, image_name)
except Exception as error:
logs.info("Handler", "Error: Start Instance {}".format(error.args[0]))
return Response.error(error.args[0])
return Response.success("ja precies - creating image!")
def instance_ready(event, context):
if "body" not in event or event.get("body") is None:
return Response.error(
"Missing required parameters: account, region, id")
event = json.loads(event.get("body"))
account_name = event.get("account", None)
region = event.get("region", None)
instance_id = event.get("id", None)
logs.info(
"Handler",
"start: {}, {}, {}".format(account_name, region, instance_id),
)
try:
_, instances = _build_instance_model(account_name, region)
except Exception as error:
logs.info("Handler", "Error: start setup {}".format(error.args[0]))
return Response.error(error.args[0])
try:
instances.set_ready_state(instance_id)
except Exception as error:
logs.info("Handler", "Error: set_ready_state {}".format(error.args[0]))
return Response.error(error.args[0])
return Response.success("wohooo - ready state set!")
def list_instances(event, context):
if ("queryStringParameters" not in event
or event.get("queryStringParameters") is None):
return Response.error("Missing required parameters: account, region")
account_name = event.get("queryStringParameters").get("account", None)
region = event.get("queryStringParameters").get("region", None)
owner = event.get("queryStringParameters").get("owner", None)
logs.info(
"Handler",
"list_instances: {}, {}, {}".format(account_name, region, owner),
)
filters = []
if owner:
filters.append({"Name": "tag:Owner", "Values": [owner]})
try:
_, instances = _build_instance_model(account_name, region)
except Exception as error:
logs.info("Handler",
"Error: list_instances setup {}".format(error.args[0]))
return Response.error(error.args[0])
try:
instances_list = instances.list(filters)
except Exception as error:
logs.info("Handler", "Error: list_instances {}".format(error.args[0]))
return Response.error(error.args[0])
return Response.success(instances_list)
def terminate(event, context):
if "body" not in event or event.get("body") is None:
return Response.error(
"Missing required parameters: account, region, id")
event = json.loads(event.get("body"))
account_name = event.get("account", None)
region = event.get("region", None)
instance_id = event.get("id", None)
logs.info(
"Handler",
"terminate: {}, {}, {}".format(account_name, region, instance_id),
)
try:
_, instances = _build_instance_model(account_name, region)
except Exception as error:
logs.info("Handler",
"Error: list_images setup {}".format(error.args[0]))
return Response.error(error.args[0])
try:
instances.terminate(instance_id)
except Exception as error:
logs.info("Handler", "Error: list_images {}".format(error.args[0]))
return Response.error(error.args[0])
return Response.success("geweldig - killing the instance!")
def list_securitygroups(event, context):
if ("queryStringParameters" not in event
or event.get("queryStringParameters") is None):
return Response.error("Missing required parameters: account, region")
account_name = event.get("queryStringParameters", {}).get("account", None)
region = event.get("queryStringParameters", {}).get("region", None)
logs.info("Handler",
"list_securitygroups: {}, {}".format(account_name, region))
try:
_, instances = _build_instance_model(account_name, region)
except Exception as error:
logs.info(
"Handler",
"Error: list_securitygroups setup {}".format(error.args[0]),
)
return Response.error(error.args[0])
try:
securitygroups = instances.list_security_groups()
except Exception as error:
logs.info("Handler",
"Error: list_securitygroups {}".format(error.args[0]))
return Response.error(error.args[0])
return Response.success(securitygroups)
def stop(event, context):
if "body" not in event or event.get("body") is None:
return Response.error(
"Missing required parameters: account, region, id")
event = json.loads(event.get("body"))
account_name = event.get("account", None)
region = event.get("region", None)
instance_id = event.get("id", None)
logs.info("Handler", "stop: {}, {}, {}".format(account_name, region,
instance_id))
try:
_, instances = _build_instance_model(account_name, region)
except Exception as error:
logs.info("Handler", "Error: stop setup {}".format(error.args[0]))
return Response.error(error.args[0])
try:
instances.stop(instance_id)
except Exception as error:
logs.info("Handler", "Error: stop {}".format(error.args[0]))
return Response.error(error.args[0])
return Response.success("geweldig - instance is stopping now!")
def update(event, context):
if "body" not in event or event.get("body") is None:
return Response.error(
"Missing required parameters: account, region, id")
event = json.loads(event.get("body"))
account_name = event.get("account", None)
region = event.get("region", None)
logs.info("Handler", "update: {}, {}".format(account_name, region))
try:
account, instances = _build_instance_model(account_name, region)
except Exception as error:
logs.info("Handler", "Error: update setup {}".format(error.args[0]))
return Response.error(error.args[0])
try:
entity = Instance(event, account)
instances.update(entity)
except Exception as error:
logs.info("Handler", "Error: update {}".format(error.args[0]))
return Response.error(error.args[0])
return Response.success("all right - instance attributes updated!")
def list_accounts(event, context):
settings = Settings()
astngs = []
for k, v in settings.accounts.items():
if v.get("enabled", False):
astngs.append(v)
return Response.success(astngs)
def start_tagged_instances(event, context):
current_hour = dt.datetime.today().hour
try:
settings = Settings()
except Exception as error:
logs.info("Handler", "Error getting settings {}".format(error.args[0]))
return Response.error(error.args[0])
for account_name in settings.accounts:
logs.info(
"Handler",
"Info: start_tagged_instances account {}".format(account_name),
)
account_config = settings.accounts.get(account_name)
if not account_config.get("enabled", False):
logs.info(
"Handler",
"Info: start_tagged_instances account {} not enabled, skipping"
.format(account_name),
)
continue
for region in account_config.get("regions"):
logs.info(
"Handler",
"Info: start_tagged_instances region {}".format(region),
)
try:
_, instances = _build_instance_model(account_name, region,
settings)
except Exception as error:
logs.info(
"Handler",
"Error: start_tagged_instances setup {}".format(
error.args[0]),
)
continue
pass
try:
instances.start_tagged_instances([{
"Name": "tag:StartTime",
"Values": [str(current_hour)]
}])
except Exception as error:
logs.info(
"Handler",
"Error: start_tagged_instances {}".format(error.args[0]),
)
continue
pass
return Response.success("ok")
def terminate_tagged_instances(event, context):
current_date = datetime.now().strftime("%d/%m/%Y")
try:
settings = Settings()
except Exception as error:
logs.info("Handler", "Error getting settings {}".format(error.args[0]))
return Response.error(error.args[0])
for account_name in settings.accounts:
account_config = settings.accounts.get(account_name)
if not account_config.get("enabled", False):
logs.info(
"Handler",
"Info: terminate_tagged_instances account {} not enabled, skipping"
.format(account_name),
)
continue
for region in account_config.get("regions"):
try:
_, instances = _build_instance_model(account_name, region,
settings)
except Exception as error:
logs.info(
"Handler",
"Error: terminate_tagged_instances setup {}".format(
error.args[0]),
)
continue
pass
try:
instances.terminate_tagged_instances([
{
"Name": "tag:TerminateDate",
"Values": [str(current_date)],
},
])
except Exception as error:
logs.info(
"Handler",
"Error: terminate_tagged_instances {}".format(
error.args[0]),
)
continue
pass
return Response.success("ok")
def info(event, context):
storage = Storage(Settings().config)
try:
owners_list = storage.list_all_owners()
except Exception as error:
logs.info("Handler", "Error: list_all_owners {}".format(error.args[0]))
return Response.error(error.args[0])
response = {
"users": owners_list,
"environments": ["test", "prod", "poc", "demo"],
}
return Response.success(response)
def start(event, context):
if "body" not in event or event.get("body") is None:
return Response.error(
"Missing required parameters: account, region, id")
event = json.loads(event.get("body"))
account_name = event.get("account", None)
region = event.get("region", None)
instance_id = event.get("id", None)
logs.info(
"Handler",
"start: {}, {}, {}".format(account_name, region, instance_id),
)
try:
_, instances = _build_instance_model(account_name, region)
except Exception as error:
logs.info("Handler", "Error: start setup {}".format(error.args[0]))
return Response.error(error.args[0])
try:
instances.start(instance_id)
except exception.DnsRecordsTagsMissing as error:
logs.info(
"Handler",
"DnsRecordsTagsMissing: Start instance {}".format(error.args[0]),
)
return Response.success("instance started, but ...{}".format(
error.args[0]))
except Exception as error:
logs.info("Handler", "Error: Start instance {}".format(error.args[0]))
return Response.error(error.args[0])
return Response.success("wohooo - instance started!")
def statistics(event, context):
# select pregunta, voto, count(voto) from votos group by pregunta, voto;
engine = mysql_engine()
# connection = engine.connect()
data = {}
with engine.connect() as con:
rs = con.execute(
'select pregunta, voto, count(voto) from votos group by pregunta, voto'
)
for row in rs:
if row[0] not in data:
data[row[0]] = {}
data[row[0]][row[1]] = row[2]
# data[row[0]] = ({row[1]: row[2]})
# print("{}{}{}".format(row[0], row[1], row[2]))
print(data)
response = []
for d in data:
response.append({'pregunta': d, 'resultados': data[d]})
print(response)
return Response.success(response)
