Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 22 additions & 1 deletion src/aerie_cli/aerie_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -1298,6 +1298,27 @@ def get_all_activity_types(self, model_id: int) -> List[str]:
activity_types = [o["name"] for o in data]
return activity_types

def get_activity_parameters(self, activity_name:str, model_id:int):
query = """
query GetActivityParameters(
$model_id: Int!
$activity_name: String!
) {
activity_type(where: { model_id: { _eq: $model_id },
name : {_eq : $activity_name}}) {
name
model_id
parameters
}
}
"""
resp = self.aerie_host.post_to_graphql(
query,
activity_name=activity_name,
model_id=model_id,
)
return resp

def list_command_dictionaries(self) -> List[CommandDictionaryInfo]:
"""List all command dictionaries on an Aerie host

Expand Down Expand Up @@ -1583,7 +1604,7 @@ def __expand_activity_arguments(self, plan: ActivityPlanRead, full_args: str = N
activity.arguments = ApiEffectiveActivityArguments.from_dict(
resp).arguments
return plan

def upload_constraint(self, constraint):
upload_constraint_query = """
mutation CreateConstraint($constraint: constraint_insert_input!) {
Expand Down