/
seppukku.py
67 lines (50 loc) · 1.88 KB
/
seppukku.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
#!/usr/bin/env
import requests
import json
import time, datetime
from datetime import datetime
from dateutil.relativedelta import relativedelta
import sns_sqs
metadata_server="http://169.254.169.254/latest/meta-data/"
clean_tmp_dirs = False
def get_instance_starttime() :
try:
data = requests.get(metadata_server + "local-ipv4")
last_mod = data.headers['last-modified']
return datetime.strptime(last_mod, "%a, %d %b %Y %H:%M:%S %Z")
except e:
print "Caught exception : {0}".format(e)
def kill_self(app, dry_run=False):
instance_id = "foo"
rc = app.config["ec2.conn"].terminate_instances(instance_ids=[instance_id], dry_run=dry_run)
return rc
def request_kill_from_controller(app, dry_run=False):
sqs_conn = app.config["sqs.conn"]
active = app.config["instance.tags"]["ActiveQueueName"]
active_q = sqs_conn.get_queue(active)
msg = {"job_id" : "kill_request",
"dry_run" : str(dry_run)}
job_id = "kill_request"
sns_sqs.post_message_to_active(app, active_q, json.dumps(msg), job_id)
def die_at_hour_edge (app, dry_run=False) :
# Die at 5 mins
start_time = get_instance_starttime()
current_time = datetime.now()
delta = current_time - start_time
hours = delta.total_seconds() / (60*60)
partial_hour = ( delta.total_seconds() - int(hours)*60*60 ) /60
print "Minutes from hour edge ", 60 - partial_hour
if partial_hour > 58 :
print "Time to die"
#kill_self(app, dry_run=dry_run)
request_kill_from_controller(app, dry_run=True)
else:
print "Wait till 2 mins to hour to die"
return delta
if __name__ == "__main__" :
import config_manager as cm
app = cm.load_configs("production.conf")
request_kill_from_controller(app, dry_run=False)
while True:
time.sleep(1)
die_at_hour_edge(app)