Add Python decorator example #84
Open
+267
−0
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
pietern
reviewed
May 26, 2025
| # Python decorator | ||
|
|
||
| This example shows how to package a **Python decorator–based job** so you can | ||
| call regular Python functions from a Databricks job with minimal boilerplate. |
Contributor
There was a problem hiding this comment.
Can you make it more clear this is about Python support for DABs?
E.g. include a doc link in this paragraph.
| raise ValueError("Only keyword arguments are supported") | ||
|
|
||
| func_full_name = f"{self.func.__module__}:{self.func.__name__}" | ||
| parameters: list = [func_full_name] + [f"--{k}={v}" for k, v in kwargs.items()] |
Contributor
There was a problem hiding this comment.
With string conversion + argparse turning it from a string back into the type that the function accepts, does this limit the function argument types in a particular way?
| from my_project.python_wheel_task import python_wheel_task | ||
|
|
||
|
|
||
| @python_wheel_task |
Contributor
There was a problem hiding this comment.
For this mini-framework, should this not just be called @task? In a more extensive version, we could always decide to use subtypes like @task.sql or move to top-level task types like @sql_file_task.
|
|
||
| # Makes the value available to downstream tasks as | ||
| # '{{ tasks.<task_key>.values.message }}' | ||
| dbutils.jobs.taskValues.set("message", "Hello World") |
Contributor
There was a problem hiding this comment.
Could you make this example do something Sparky, like spark.read.table("samples.nyctaxi.trips")?
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Add an example of using Python decorators in Databricks jobs, see README.md for more information.