[PROD-1293] Add 'auto_appy_recs' property to projects (#51)

Author: rmoneys

sync/__init__.py

@@ -1,4 +1,4 @@
 """Library for leveraging the power of Sync"""
-__version__ = "0.4.4"
+__version__ = "0.4.5"
 
 TIME_FORMAT = "%Y-%m-%dT%H:%M:%SZ"

sync/api/projects.py

@@ -55,6 +55,7 @@ def create_project(
     job_id: str = None,
     cluster_log_url: str = None,
     prediction_preference: Preference = Preference.ECONOMY,
+    auto_apply_recs: bool = False,
     prediction_params: dict = None,
     app_id: str = None,
 ) -> Response[dict]:
@@ -72,6 +73,8 @@ def create_project(
     :type cluster_log_url: str, optional
     :param prediction_preference: preferred prediction solution, defaults to `Preference.ECONOMY`
     :type prediction_preference: Preference, optional
+    :param auto_apply_recs: automatically apply project recommendations, defaults to False
+    :type auto_apply_recs: bool, optional
     :param prediction_params: dictionary of prediction parameters, defaults to None. Valid options are documented `here <https://developers.synccomputing.com/reference/create_project_v1_projects_post>`__
     :type prediction_params: dict, optional
     :param app_id: Apache Spark application identifier, defaults to None
@@ -88,6 +91,7 @@ def create_project(
                 "job_id": job_id,
                 "cluster_log_url": cluster_log_url,
                 "prediction_preference": prediction_preference,
+                "auto_apply_recs": auto_apply_recs,
                 "prediction_params": prediction_params,
                 "app_id": app_id,
             }
@@ -112,6 +116,7 @@ def update_project(
     cluster_log_url: str = None,
     app_id: str = None,
     prediction_preference: Preference = None,
+    auto_apply_recs: bool = None,
     prediction_params: dict = None,
 ) -> Response[dict]:
     """Updates a project's mutable properties
@@ -126,6 +131,8 @@ def update_project(
     :type app_id: str, optional
     :param prediction_preference: default preference for predictions, defaults to None
     :type prediction_preference: Preference, optional
+    :param auto_apply_recs: automatically apply project recommendations, defaults to None
+    :type auto_apply_recs: bool, optional
     :param prediction_params: dictionary of prediction parameters, defaults to None. Valid options are documented `here <https://developers.synccomputing.com/reference/update_project_v1_projects__project_id__put>`__
     :type prediction_preference: dict, optional
     :return: updated project
@@ -140,6 +147,8 @@ def update_project(
         project_update["app_id"] = app_id
     if prediction_preference:
         project_update["prediction_preference"] = prediction_preference
+    if auto_apply_recs is not None:
+        project_update["auto_apply_recs"] = auto_apply_recs
     if prediction_params:
         project_update["prediction_params"] = prediction_params
 

sync/cli/projects.py

@@ -62,12 +62,19 @@ def get(project: dict):
     type=click.Choice(Preference),
     default=CONFIG.default_prediction_preference,
 )
+@click.option(
+    "--auto-apply-recs",
+    is_flag=True,
+    default=False,
+    help="Automatically apply project recommendations",
+)
 @click.option(
     "-i", "--app-id", help="External identifier often based on the project's target application"
 )
 def create(
     name: str,
     product_code: str,
+    auto_apply_recs: bool,
     description: str = None,
     job_id: str = None,
     location: str = None,
@@ -84,6 +91,7 @@ def create(
         job_id=job_id,
         cluster_log_url=location,
         prediction_preference=preference,
+        auto_apply_recs=auto_apply_recs,
         app_id=app_id,
     )
     project = response.result
@@ -106,15 +114,24 @@ def create(
     type=click.Choice(Preference),
     default=CONFIG.default_prediction_preference,
 )
+@click.option("--auto-apply-recs", type=bool, help="Automatically apply project recommendations")
 def update(
     project_id: str,
     description: str = None,
     location: str = None,
     app_id: str = None,
     preference: Preference = None,
+    auto_apply_recs: bool = None,
 ):
     """Update a project"""
-    response = update_project(project_id, description, location, app_id, preference)
+    response = update_project(
+        project_id,
+        description=description,
+        s3_url=location,
+        app_id=app_id,
+        prediction_preference=preference,
+        auto_apply_recs=auto_apply_recs,
+    )
     if response.result:
         click.echo("Project updated")
     else: