Add a main function to aws.py.

zalando · Nov 25, 2015 · 054223c · 054223c
1 parent d150f40
commit 054223c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/patroni/scripts/aws.py b/patroni/scripts/aws.py
@@ -65,8 +65,11 @@ def on_role_change(self, new_role):
         return self._tag_ebs(new_role) and ret
 
 
-if __name__ == '__main__':
+def main():
     if len(sys.argv) == 4 and sys.argv[1] in ('on_start', 'on_stop', 'on_role_change'):
         AWSConnection(cluster_name=sys.argv[3]).on_role_change(sys.argv[2])
     else:
         sys.exit("Usage: {0} action role name".format(sys.argv[0]))
+
+if __name__ == '__main__':
+    main()