Fix #85 generic migration state (#86)

Author: twarnock

step/lambdas/copy_image.py

@@ -8,7 +8,7 @@
 from typing import Any, Dict
 
 import boto3
-from servicenowstate import ServiceNowState, ServiceNowStateHandler
+from migrationstate import MigrationState, MigrationStateHandler
 
 print("Loading function copy_image")
 
@@ -59,7 +59,7 @@ def lambda_handler(event: Dict[str, Any], context: Any) -> str:
         print(e)
         return ""
 
-    ServiceNowStateHandler().update_state(
+    MigrationStateHandler().update_state(
         state="IMAGE_COPYING", machine_name=instance_name
     )
     return new_image.get("ImageId", "")

step/lambdas/create_image.py

@@ -8,7 +8,7 @@
 from typing import Any, Dict
 
 import boto3
-from servicenowstate import ServiceNowStateHandler
+from migrationstate import MigrationStateHandler
 
 print("Loading function create_image")
 
@@ -42,6 +42,6 @@ def lambda_handler(event: Dict[str, Any], context: Any) -> str:
 
     instance.create_tags(Tags=[{"Key": "CloneStatus", "Value": "IMAGE_CREATED"}])
 
-    ServiceNowStateHandler().update_state(state="IMAGE_CREATING", machine_name=name)
+    MigrationStateHandler().update_state(state="IMAGE_CREATING", machine_name=name)
 
     return ec2_image.image_id

step/lambdas/find_instance.py

@@ -8,7 +8,7 @@
 from typing import Any, Dict
 
 import boto3
-from servicenowstate import ServiceNowStateHandler
+from migrationstate import MigrationStateHandler
 
 print("Loading function find_instance")
 
@@ -72,7 +72,7 @@ def lambda_handler(event: Dict[str, Any], context: Any) -> str:
         print(e)
         event_dict["instance_id"] = "not-found"
 
-    ServiceNowStateHandler().update_state(
+    MigrationStateHandler().update_state(
         state="INSTANCE_LAUNCHED", machine_name=event_dict.get("name")
     )
     return event_dict

step/lambdas/get_copy_status.py

@@ -8,7 +8,7 @@
 from typing import Any, Dict
 
 import boto3
-from servicenowstate import ServiceNowStateHandler
+from migrationstate import MigrationStateHandler
 
 print("Loading function get_copy_status")
 
@@ -39,7 +39,7 @@ def lambda_handler(event: Dict[str, Any], context: Any) -> str:
 
     state = ami_state["Images"][0]["State"]
     if state == "available":
-        ServiceNowStateHandler().update_state(
+        MigrationStateHandler().update_state(
             state="IMAGE_COPIED", machine_name=instance_name
         )
 

step/lambdas/get_image_status.py

@@ -7,7 +7,7 @@
 from typing import Any, Dict
 
 import boto3
-from servicenowstate import ServiceNowStateHandler
+from migrationstate import MigrationStateHandler
 
 print("Loading function get_image_status")
 
@@ -33,7 +33,7 @@ def lambda_handler(event: Dict[str, Any], context: Any) -> str:
 
     state = ami_state["Images"][0]["State"]
     if state == "available":
-        ServiceNowStateHandler().update_state(
+        MigrationStateHandler().update_state(
             state="IMAGE_CREATED", machine_name=instance_name
         )
 

step/lambdas/get_instance_status.py

@@ -7,7 +7,7 @@
 from typing import Any, Dict
 
 import boto3
-from servicenowstate import ServiceNowStateHandler
+from migrationstate import MigrationStateHandler
 
 print("Loading function get_instance_status")
 
@@ -43,7 +43,7 @@ def lambda_handler(event: Dict[str, Any], context: Any) -> str:
                 state = "system_failed"
 
     if state == "running":
-        ServiceNowStateHandler().update_state(
+        MigrationStateHandler().update_state(
             state="INSTANCE_READY", machine_name=instance_name
         )
 

step/lambdas/image_cleanup.py

@@ -8,7 +8,7 @@
 from typing import Any, Dict, List
 
 import boto3
-from servicenowstate import ServiceNowStateHandler
+from migrationstate import MigrationStateHandler
 
 print("Loading function image_cleanup")
 
@@ -46,7 +46,7 @@ def lambda_handler(event: Dict[str, Any], context: Any) -> bool:
         print(f"Failed.  AMI may not exist.\n{str(e)}")
         return False
 
-    ServiceNowStateHandler().update_state(
+    MigrationStateHandler().update_state(
         state="IMAGE_READY", machine_name=instance_name
     )
 

step/lambdas/servicenowstate.py step/lambdas/migrationstate.pystep/lambdas/servicenowstate.py renamed to step/lambdas/migrationstate.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-"""Define ServiceNow events."""
+"""Define Migration events."""
 from __future__ import annotations
 
 import json
@@ -10,7 +10,7 @@
 import boto3
 
 
-class ServiceNowStateHandler:
+class MigrationStateHandler:
     sqs = None
 
     def update_state(self, state: str, machine_name: str, **kwargs) -> bool:
@@ -19,22 +19,22 @@ def update_state(self, state: str, machine_name: str, **kwargs) -> bool:
 
         print("Event queue: " + os.environ.get("event_queue"))
         queue_url = os.environ.get("event_queue")
-        state_obj = ServiceNowState(state, machine_name, **kwargs)
+        state_obj = MigrationState(state, machine_name, **kwargs)
         print(
             sqs.send_message(
                 QueueUrl=queue_url, MessageBody=json.dumps(state_obj.state_dict)
             )
         )
 
 
-class ServiceNowException(Exception):
-    """Define the structure of a ServiceNow exception."""
+class MigrationException(Exception):
+    """Define the structure of a Migration exception."""
 
     pass
 
 
-class ServiceNowState:
-    """Define the structure of a ServiceNow state."""
+class MigrationState:
+    """Define the structure of a Migration state."""
 
     state_dict = {}
 
@@ -66,7 +66,7 @@ class ServiceNowState:
     def __init__(self, state: str, machine_name: str = "NA", **kwargs) -> None:
         """Initialize the State."""
         if state not in self.STATES:
-            raise ServiceNowException(f"State: {state} is unrecognized!")
+            raise MigrationException(f"State: {state} is unrecognized!")
 
         self.machine_name: str = machine_name.upper()
         self.state: str = state

step/lambdas/share_image.py

@@ -7,7 +7,7 @@
 from typing import Any, Dict
 
 import boto3
-from servicenowstate import ServiceNowStateHandler
+from migrationstate import MigrationStateHandler
 
 print("Loading function share_image")
 
@@ -59,7 +59,7 @@ def lambda_handler(event: Dict[str, Any], context: Any) -> bool:
                 print(e)
                 return False
 
-    ServiceNowStateHandler().update_state(
+    MigrationStateHandler().update_state(
         state="IMAGE_SHARED", machine_name=instance_name
     )
     return True

step/lambdas/split_image.py

@@ -8,7 +8,7 @@
 from typing import Any, Dict, List
 
 import boto3
-from servicenowstate import ServiceNowStateHandler
+from migrationstate import MigrationStateHandler
 
 print("Loading function split_image")
 
@@ -90,7 +90,7 @@ def lambda_handler(event: Dict[str, Any], context: Any) -> str:
         print(e)
         return ""
 
-    ServiceNowStateHandler().update_state(
+    MigrationStateHandler().update_state(
         state="IMAGE_SPLIT", machine_name=instance_name
     )