refactor: code structure and doc

Author: containerscrew

README.md

@@ -1,7 +1,6 @@
 <p align="center" >
     <h3 align="center">cloudsnake 🐍</h3>
-    <p align="center">Some AWS CLI commands with a beautiful TUI</p>
-    <p align="center">Implement some useful commands to operate AWS. Like EC2 SSM instance connection or RDS connection using IAM authentication</p>
+    <p align="center">Wrapping some useful AWS cli commands to operate AWS like EC2 SSM instance connection or RDS connection using IAM authentication</p>
     <p align="center">Built with ❤ in Python</p>
 </p>
 

src/cloudsnake/cli/ssm.py

@@ -6,14 +6,21 @@
 from cloudsnake.sdk.ssm_parameter_store import SSMParameterStoreWrapper
 from cloudsnake.sdk.ssm_session import SSMStartSessionWrapper
 
+EC2_RUNNING_FILTER = "Name=instance-state-name,Values=running"
+
+EC2_INSTANCE_SELECTOR_QUERY = (
+    "Reservations[*].Instances[*].{"
+        "TargetId: InstanceId, "
+        "Name: Tags[?Key=='Name'].Value | [0]"
+    "}"
+)
 
 ssm = typer.Typer(
     no_args_is_help=True,
     pretty_exceptions_short=True,
     pretty_exceptions_show_locals=False,
 )
 
-
 @ssm.command(
     "start-session", help="Start session with the given target id", no_args_is_help=True
 )
@@ -30,24 +37,33 @@ def start_session(
         help="Prompt a terminal menu and select the instance you want to connect. --target flag is no longer used",
     ),
 ):
-    ssm = SSMStartSessionWrapper()
+    ssm = SSMStartSessionWrapper(
+        profile=ctx.obj.profile,
+        region=ctx.obj.region,
+    )
     ssm.create_client(ctx.obj.session)
     if with_instance_selector:
-        filters = "Name=instance-state-name,Values=running"
-        query = "Reservations[*].Instances[*].{TargetId: InstanceId, Name: Tags[?Key=='Name'].Value | [0]}"
-        ec2 = EC2InstanceWrapper(filters=filters, query=query)
+        ec2 = EC2InstanceWrapper(
+            filters=EC2_RUNNING_FILTER,
+            query=EC2_INSTANCE_SELECTOR_QUERY,
+            profile=ctx.obj.profile,
+            region=ctx.obj.region,
+        )
         ec2.create_client(ctx.obj.session)
         instances = ec2.describe_ec2_instances()
-        if len(instances) == 0:
+        if not instances:
             typer.echo("No running instances found.")
-            exit(1)
+            raise typer.Exit(1)
+
         instance_name = ctx.obj.tui.interactive_menu(
-            instances, title="Select the EC2 you want to connect"
+            instances,
+            title="Select the EC2 you want to connect"
         )
         instance_id = ctx.obj.tui.get_target_id_by_name(instances, instance_name)
-        ssm.start_session(instance_id)
-    else:
-        ssm.start_session(target)
+
+        return ssm.start_session(instance_id)
+
+    return ssm.start_session(target)
 
 
 # @ssm.command("get-parameters", help="Get parameters from parameter store")