对于Airflow GitHub认证失败的问题,以下是一种可能的解决方法,其中包含代码示例:
airflow.cfg
或airflow.yaml
)中,找到并设置以下配置参数:[github_enterprise_conn]
host =
login =
password =
请确保将
替换为您的GitHub企业主机URL,
替换为您的GitHub用户名,以及
替换为您的个人访问令牌。
BaseHook
类创建一个自定义的GitHub认证Hook。创建一个新的Python文件(例如github_hook.py
),并添加以下代码:from airflow.hooks.base_hook import BaseHook
from airflow.exceptions import AirflowException
from github import Github
class GitHubHook(BaseHook):
def __init__(self, github_conn_id='github_default'):
self.github_conn_id = github_conn_id
self.conn = self.get_connection(github_conn_id)
self.host = self.conn.host
self.login = self.conn.login
self.password = self.conn.password
def get_github_client(self):
try:
g = Github(base_url=self.host, login_or_token=self.password)
return g
except Exception as e:
raise AirflowException(f"Failed to create GitHub client: {str(e)}")
def get_repos(self):
g = self.get_github_client()
repos = g.get_user().get_repos()
return [repo.name for repo in repos]
from airflow import DAG
from airflow.utils.dates import days_ago
from github_hook import GitHubHook
default_args = {
'owner': 'airflow',
'start_date': days_ago(1)
}
with DAG('github_dag', default_args=default_args, schedule_interval='@daily') as dag:
github_hook = GitHubHook()
repos = github_hook.get_repos()
for repo in repos:
# Do something with each repo
# ...
上述代码示例中,我们首先导入自定义的GitHub认证Hook GitHubHook
。然后,我们在DAG中创建一个GitHubHook
实例,并使用get_repos
方法获取GitHub仓库列表。最后,我们可以在DAG中执行任何与GitHub仓库相关的操作。
请记住,您需要根据自己的实际情况修改代码和配置参数。这只是一种可能的解决方法,具体取决于您的Airflow和GitHub配置。