Ability to Upload Scheduling Goals to All Plans with Mission Model ID

src/aerie_cli/aerie_client.py

@@ -100,6 +100,20 @@ def list_all_activity_plans(self) -> List[ActivityPlanRead]:
             plan = ActivityPlanRead.from_api_read(plan)
             activity_plans.append(plan)
         return activity_plans
+
+    def get_all_activity_plans_by_model(self, model_id: int) -> List[ActivityPlanRead]: 
+        get_all_plans_by_model_query = """
+        query get_all_plans_by_model_id($model_id: Int!) {
+            mission_model_by_pk(id: $model_id) {
+                plans {
+                    id
+                }
+            }
+        }
+        """
+
+        resp = self.host_session.post_to_graphql(get_all_plans_by_model_query, model_id=model_id)
+        return resp['plans']
 
     def get_all_activity_plans(self, full_args: str = None) -> list[ActivityPlanRead]:
         get_all_plans_query = """

src/aerie_cli/commands/models.py

@@ -111,4 +111,4 @@ def list():
         )
 
     console = Console()
-    console.print(table)
+    console.print(table)

src/aerie_cli/commands/plans.py

@@ -313,5 +313,4 @@ def clean():
     for activity_plan in resp:
         client.delete_plan(activity_plan.id)
 
-    typer.echo(f"All activity plans have been deleted")
-
+    typer.echo(f"All activity plans have been deleted")

src/aerie_cli/commands/scheduling.py

@@ -8,9 +8,9 @@
 def upload(
     model_id: int = typer.Option(
         ..., help="The mission model ID to associate with the scheduling goal", prompt=True
-    ),
+    ),   
     plan_id: int = typer.Option(
-        ..., help="Plan ID", prompt=True
+        help="Plan ID (optional)", default=-1, prompt=True #how to make this optional
     ),
     schedule: str = typer.Option(
         ..., help="Text file with one path on each line to a scheduling rule file, in decreasing priority order", prompt=True
@@ -29,22 +29,36 @@ def upload(
                 d = dict(zip(keys, [filename, model_id, f.read()]))
                 upload_obj.append(d)
 
-    resp = client.upload_scheduling_goals(upload_obj)
+    if(plan_id != -1):
+        #uploading to single plan
+        resp = client.upload_scheduling_goals(upload_obj)
 
-    typer.echo(f"Uploaded scheduling goals to venue.")
+        typer.echo(f"Uploaded scheduling goals to venue.")
 
-    uploaded_ids = [kv["id"] for kv in resp]
+        uploaded_ids = [kv["id"] for kv in resp]
 
-    #priority order is order of filenames in decreasing priority order
-    #will append to existing goals in specification priority order
-    specification = client.get_specification_for_plan(plan_id)
+        #priority order is order of filenames in decreasing priority order
+        #will append to existing goals in specification priority order
+        specification = client.get_specification_for_plan(plan_id)
 
-    upload_to_spec = [{"goal_id": goal_id, "specification_id": specification} for goal_id in uploaded_ids]
+        upload_to_spec = [{"goal_id": goal_id, "specification_id": specification} for goal_id in uploaded_ids]
+        client.add_goals_to_specifications(upload_to_spec)
+        typer.echo(f"Assigned goals in priority order to plan ID {plan_id}.")
+    else: 
+        #get all plan ids from model id if no plan id is provided 
+        resp = client.get_all_activity_plans_by_model(model_id)
 
-    client.add_goals_to_specifications(upload_to_spec)
+        for plan in resp: 
+            #each schedule goal needs own ID - add each goal for each plan
+            resp = client.upload_scheduling_goals(upload_obj)
+            typer.echo(f"Uploaded scheduling goals to venue.")
+            uploaded_ids = [kv["id"] for kv in resp]
 
-    typer.echo(f"Assigned goals in priority order to plan ID {plan_id}.")
+            specification = client.get_specification_for_plan(plan["id"])
+            upload_to_spec = [{"goal_id": goal_id, "specification_id": specification} for goal_id in uploaded_ids]
+            client.add_goals_to_specifications(upload_to_spec)
 
+        typer.echo(f"Assigned goals in priority order to all plans with model ID {model_id}.")
 
 @app.command()
 def delete(