Merge pull request #234 from DataDog/add_instance_reason

EC2: Add instance state reason

moto/ec2/models.py

@@ -2,6 +2,7 @@ from __future__ import unicode_literals
 import copy
 import itertools
 from collections import defaultdict
+from datetime import datetime
 
 import six
 import boto
@@ -96,6 +97,11 @@ class InstanceState(object):
         self.name = name
         self.code = code
 
+class StateReason(object):
+    def __init__(self, message="", code=""):
+        self.message = message
+        self.code = code
+
 
 class TaggedEC2Resource(object):
     def get_tags(self, *args, **kwargs):
@@ -259,6 +265,8 @@ class Instance(BotoInstance, TaggedEC2Resource):
         self.id = random_instance_id()
         self.image_id = image_id
         self._state = InstanceState("running", 16)
+        self._reason = ""
+        self._state_reason = StateReason()
         self.user_data = user_data
         self.security_groups = security_groups
         self.instance_type = kwargs.get("instance_type", "m1.small")
@@ -318,6 +326,9 @@ class Instance(BotoInstance, TaggedEC2Resource):
         self._state.name = "running"
         self._state.code = 16
 
+        self._reason = ""
+        self._state_reason = StateReason()
+
     def stop(self, *args, **kwargs):
         for nic in self.nics.values():
             nic.stop()
@@ -325,6 +336,10 @@ class Instance(BotoInstance, TaggedEC2Resource):
         self._state.name = "stopped"
         self._state.code = 80
 
+        self._reason = "User initiated ({0})".format(datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S UTC'))
+        self._state_reason = StateReason("Client.UserInitiatedShutdown: User initiated shutdown",
+                                         "Client.UserInitiatedShutdown")
+
     def terminate(self, *args, **kwargs):
         for nic in self.nics.values():
             nic.stop()
@@ -332,10 +347,17 @@ class Instance(BotoInstance, TaggedEC2Resource):
         self._state.name = "terminated"
         self._state.code = 48
 
+        self._reason = "User initiated ({0})".format(datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S UTC'))
+        self._state_reason = StateReason("Client.UserInitiatedShutdown: User initiated shutdown",
+                                         "Client.UserInitiatedShutdown")
+
     def reboot(self, *args, **kwargs):
         self._state.name = "running"
         self._state.code = 16
 
+        self._reason = ""
+        self._state_reason = StateReason()
+
     def get_tags(self):
         tags = ec2_backend.describe_tags(filters={'resource-id': [self.id]})
         return tags