From 78f0b16543541ce20b42c0ab97878b8b57b47d17 Mon Sep 17 00:00:00 2001
From: Oleksii Kliukin <oleksii.kliukin@zalando.de>
Date: Fri, 10 Jul 2015 10:57:24 +0200
Subject: [PATCH] Enable aws.py to run as a stand-alone program.

---
 helpers/aws.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/helpers/aws.py b/helpers/aws.py
index 26c65e89..02ddd7fe 100644
--- a/helpers/aws.py
+++ b/helpers/aws.py
@@ -1,6 +1,7 @@
 import logging
 import requests
 from requests.exceptions import RequestException
+import sys
 import boto.ec2
 
 logger = logging.getLogger(__name__)
@@ -67,3 +68,15 @@ class AWSConnection:
     def on_role_change(self, new_role):
         ret = self._tag_ec2(new_role)
         return self._tag_ebs(new_role) and ret
+
+
+if __name__ == '__main__':
+    if len(sys.argv) != 3:
+        print ("Usage: {0} action role name".format(sys.argv[0]))
+        return 1
+    action, role, name = sys.argv[1:]
+    if action in ('on_start', 'on_stop', 'on_role_change'):
+        aws = gAWSConnection({'cluster_name': name})
+        aws.on_role_change(role)
+        return 0
+    return 2