RaphCodec commented on code in PR #64422:
URL: https://github.com/apache/airflow/pull/64422#discussion_r3697416120
##########
providers/git/src/airflow/providers/git/hooks/git.py:
##########
@@ -183,6 +216,82 @@ def _build_ssh_command(self, key_path: str | None = None)
-> str:
return " ".join(parts)
+ def _get_github_app_token(self):
+ try:
+ from github import Auth, GithubIntegration
+ except ImportError as exc:
+ raise ImportError(
+ "The PyGithub library is required for GitHub App
authentication. Please install it with 'pip install
apache-airflow-providers-git[github]'"
+ ) from exc
+
+ auth = Auth.AppAuth(self.github_app_id, self.github_app_private_key)
+ integration = GithubIntegration(auth=auth)
+ access_token =
integration.get_access_token(installation_id=self.github_installation_id)
+ github_app_token_exp = access_token.expires_at
+ log.info(
+ "Successfully obtained GitHub App installation access token
(expires at: %s)",
+ github_app_token_exp,
+ )
+
+ return "x-access-token", access_token.token, github_app_token_exp
+
+ def _ensure_github_app_token(self) -> None:
+ if self.github_app_id is None or self.github_installation_id is None:
+ return
+
+ TOKEN_REFRESH_BUFFER = timedelta(minutes=5)
+ if (
+ self.github_app_token_exp is None
+ or self.github_app_token_exp < datetime.now(timezone.utc) +
TOKEN_REFRESH_BUFFER
+ ):
+ log.info(
+ "GitHub App token is missing or near expiry (expires at: %s).
Refreshing token.",
+ self.github_app_token_exp,
+ )
+ self.user_name, self.auth_token, self.github_app_token_exp =
self._get_github_app_token()
+
+ @contextlib.contextmanager
+ def _github_app_askpass_env(self):
Review Comment:
added `Generator[None]` annotation
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
