truefoundry 0.4.4rc2__py3-none-any.whl → 0.4.4rc3__py3-none-any.whl

truefoundry/deploy/python_deploy_codegen.py

@@ -1,15 +1,16 @@
 import ast
 import io
+import json
 import re
-from typing import List
+from typing import Dict, List, Optional
 
 from rich.console import Console
 from rich.pretty import pprint
 
-from truefoundry.deploy import Application
+from truefoundry.deploy import Application, LocalSource
 
 
-def generate_code(
+def generate_deployment_code(
     symbols_to_import: List[str],
     application_type: str,
     spec_repr: str,
@@ -108,6 +109,13 @@ def convert_deployment_config_to_python(workspace_fqn: str, application_spec: di
     application = Application.parse_obj(application_spec)
     application_type = application.__root__.type
 
+    if (
+        hasattr(application.__root__, "image")
+        and application.__root__.image.type == "build"
+        and application.__root__.image.build_source.type == "remote"
+    ):
+        application.__root__.image.build_source = LocalSource(local_build=False)
+
     spec_repr = get_python_repr(application.__root__)
     spec_repr = replace_enums_with_values(spec_repr)
     spec_repr = remove_none_type_fields(spec_repr)
@@ -120,7 +128,7 @@ def convert_deployment_config_to_python(workspace_fqn: str, application_spec: di
     if "GitSource" in symbols_to_import:
         symbols_to_import.append("LocalSource")
 
-    generated_code = generate_code(
+    generated_code = generate_deployment_code(
         symbols_to_import=symbols_to_import,
         application_type=application_type,
         spec_repr=spec_repr,
@@ -131,3 +139,66 @@ def convert_deployment_config_to_python(workspace_fqn: str, application_spec: di
         generated_code = add_local_source_comment(generated_code)
 
     return generated_code
+
+
+def generate_python_snippet_for_trigger_job(
+    application_fqn: str, command: Optional[str], params: Optional[Dict[str, str]]
+):
+    job_run_python_template = """from truefoundry.deploy import trigger_job
+
+response = trigger_job(
+  application_fqn="{{application_fqn}}",
+  # You can pass command or params, but not both
+  # command={{command}}
+  # params={{params}}
+)
+
+print(response.jobRunName)
+"""
+    output_python_str = job_run_python_template.replace(
+        "{{application_fqn}}", application_fqn
+    )
+
+    if command is not None:
+        output_python_str = output_python_str.replace("{{command}}", repr(command))
+        output_python_str = output_python_str.replace("# command", "command")
+    else:
+        output_python_str = output_python_str.replace(
+            "{{command}}", "<Enter Command Here>"
+        )
+
+    if params is not None:
+        output_python_str = output_python_str.replace("{{params}}", repr(params))
+        output_python_str = output_python_str.replace("# params", "params")
+    else:
+        output_python_str = output_python_str.replace(
+            "{{params}}", "<Enter Params(key-value pairs) here as python dict>"
+        )
+
+    return output_python_str
+
+
+def generate_curl_snippet_for_trigger_job(
+    control_plane_url: str,
+    application_id: str,
+    command: Optional[str],
+    params: Optional[Dict[str, str]],
+):
+    job_run_curl_request_template = """curl -X 'POST' \\
+  '{{control_plane_url}}/api/svc/v1/jobs/trigger' \\
+  -H 'accept: */*' \\
+  -H 'Authorization: Bearer <Paste your API key here. You can generate it from the Settings Page>' \\
+  -H 'Content-Type: application/json' \\
+  -d '{
+  "applicationId": "{{application_id}}",
+  "input": {{input}}
+}'
+"""
+    output_curl_str = job_run_curl_request_template.replace(
+        "{{control_plane_url}}", control_plane_url
+    )
+    output_curl_str = output_curl_str.replace("{{application_id}}", application_id)
+    output_curl_str = output_curl_str.replace(
+        "{{input}}", json.dumps({"command": command, "params": params}, indent=2)
+    )
+    return output_curl_str

{truefoundry-0.4.4rc2.dist-info → truefoundry-0.4.4rc3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.4.4rc2
+Version: 0.4.4rc3
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com

{truefoundry-0.4.4rc2.dist-info → truefoundry-0.4.4rc3.dist-info}/RECORD

@@ -107,7 +107,7 @@ truefoundry/deploy/lib/model/entity.py,sha256=fq8hvdJQgQn4uZqxpKrzmaoJhQG53_EbDo
 truefoundry/deploy/lib/session.py,sha256=Vg6rCA315T0yS0xG4ayJ84Ia_9ZfibH8utOSwPBMAmw,4953
 truefoundry/deploy/lib/util.py,sha256=3TapV7yczkheC1MMMfmJDGGzTl2l6e4jCYd_Rr5aoQ8,1330
 truefoundry/deploy/lib/win32.py,sha256=1RcvPTdlOAJ48rt8rCbE2Ufha2ztRqBAE9dueNXArrY,5009
-truefoundry/deploy/python_deploy_codegen.py,sha256=Ok7ufDY2x3aMJv9KpaRqxiS-ZI-kxBWauIUHst-ug7E,4020
+truefoundry/deploy/python_deploy_codegen.py,sha256=Fn2dgn_wcRigaR6Ia4pA7VdhWXaYl5D-o-K8P3953aA,6436
 truefoundry/deploy/v2/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/v2/lib/__init__.py,sha256=WEiVMZXOVljzEE3tpGJil14liIn_PCDoACJ6b3tZ6sI,188
 truefoundry/deploy/v2/lib/deploy.py,sha256=HIcY3SzQ5lWl7avuuKi3J0Z-PBES6Sf4hgMK-m6_53U,11990
@@ -340,7 +340,7 @@ truefoundry/workflow/map_task.py,sha256=2m3qGXQ90k9LdS45q8dqCCECc3qr8t2m_LMCVd1m
 truefoundry/workflow/python_task.py,sha256=SRXRLC4vdBqGjhkwuaY39LEWN6iPCpJAuW17URRdWTY,1128
 truefoundry/workflow/task.py,sha256=ToitYiKcNzFCtOVQwz1W8sRjbR97eVS7vQBdbgUQtKg,1779
 truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.4.4rc2.dist-info/METADATA,sha256=11OX7K_AJ9gbEqtVyiApCxmkGeVbMgrhZ0k4KHh6baY,3101
-truefoundry-0.4.4rc2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-truefoundry-0.4.4rc2.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.4.4rc2.dist-info/RECORD,,
+truefoundry-0.4.4rc3.dist-info/METADATA,sha256=C7iN6mFbrCfs3iLqyPq03a7Lu2QbqmiFwwiAc-G_qhc,3101
+truefoundry-0.4.4rc3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+truefoundry-0.4.4rc3.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.4.4rc3.dist-info/RECORD,,