Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Service Principal OAuth for Databricks. #33005

Merged
merged 4 commits into from
Aug 14, 2023
Merged

Add Service Principal OAuth for Databricks. #33005

merged 4 commits into from
Aug 14, 2023

Conversation

johnbrandborg
Copy link
Contributor

@johnbrandborg johnbrandborg commented Aug 1, 2023

Update the Databricks Hook so that a Service Principal OAuth Client ID and Client Secret can be used like a username and password. The Hook will perform the token retrieval as needed.

closes: #32969

@boring-cyborg
Copy link

boring-cyborg bot commented Aug 1, 2023

Congratulations on your first Pull Request and welcome to the Apache Airflow community! If you have any issues or are unsure about any anything please check our Contribution Guide (https://github.com/apache/airflow/blob/main/CONTRIBUTING.rst)
Here are some useful points:

  • Pay attention to the quality of your code (ruff, mypy and type annotations). Our pre-commits will help you with that.
  • In case of a new feature add useful documentation (in docstrings or in docs/ directory). Adding a new operator? Check this short guide Consider adding an example DAG that shows how users should use it.
  • Consider using Breeze environment for testing locally, it's a heavy docker but it ships with a working Airflow and a lot of integrations.
  • Be patient and persistent. It might take some time to get a review or get the final approval from Committers.
  • Please follow ASF Code of Conduct for all communication including (but not limited to) comments on Pull Requests, Mailing list and Slack.
  • Be sure to read the Airflow Coding style.
    Apache Airflow is a community-driven project and together we are making it better 🚀.
    In case of doubts contact the developers at:
    Mailing List: [email protected]
    Slack: https://s.apache.org/airflow-slack

@eladkal eladkal changed the title Add Service Principal Oauth for Databricks. apache/airflow#32969 Add Service Principal Oauth for Databricks. Aug 4, 2023
@eladkal
Copy link
Contributor

eladkal commented Aug 4, 2023

cc @alexott
can you take a look?

@alexott
Copy link
Contributor

alexott commented Aug 4, 2023

Really we need to prioritize of adoption of new Databricks SDK, although it doesn't have async support yet. Then we'll get all auth support automatically

@jlaneve
Copy link
Contributor

jlaneve commented Aug 4, 2023

Really we need to prioritize of adoption of new Databricks SDK, although it doesn't have async support yet. Then we'll get all auth support automatically

The Databricks SDK is still in beta with an unstable interface, right? I assume we're waiting until the SDK is stable to adopt it?

@alexott
Copy link
Contributor

alexott commented Aug 4, 2023

Authentication part is stable, unstable primarily belongs to specific APIs

Copy link
Contributor

@alexott alexott left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

in general looks good for me, but let unify code, fix tests, etc.

@johnbrandborg johnbrandborg changed the title Add Service Principal Oauth for Databricks. Add Service Principal OAuth for Databricks. Aug 7, 2023
@johnbrandborg johnbrandborg requested a review from alexott August 8, 2023 09:01
Copy link
Contributor

@alexott alexott left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

minor comment on documentation.

@alexott
Copy link
Contributor

alexott commented Aug 13, 2023

@johnbrandborg thank you for doing it! Changes looks good for me

@potiuk
Copy link
Member

potiuk commented Aug 13, 2023

docs need fixing though

@johnbrandborg
Copy link
Contributor Author

docs need fixing though

Link name updated to not conflict.

@potiuk potiuk merged commit 8bf53dd into apache:main Aug 14, 2023
@boring-cyborg
Copy link

boring-cyborg bot commented Aug 14, 2023

Awesome work, congrats on your first merged pull request! You are invited to check our Issue Tracker for additional contributions.

amoghrajesh pushed a commit to amoghrajesh/airflow that referenced this pull request Aug 14, 2023
* Add Service Principal Oauth for Databricks. (apache#32969)

* Consolidate OAuth validation and storage (apache#32969)

* Formatting update and added API and Spark to Init Import

* Update documentation with doc ref and note this is for AWS deployments
ferruzzi pushed a commit to aws-mwaa/upstream-to-airflow that referenced this pull request Aug 17, 2023
* Add Service Principal Oauth for Databricks. (apache#32969)

* Consolidate OAuth validation and storage (apache#32969)

* Formatting update and added API and Spark to Init Import

* Update documentation with doc ref and note this is for AWS deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Databricks support for Service Principal Oauth
5 participants