Airflow-DAG级别的访问控制
创始人
2024-08-01 13:01:17
0

Airflow支持对DAG进行访问控制。可以通过以下方式实现:

  1. 创建角色和用户: 将所有用户分配给角色,并根据需要分配角色权限。
from airflow import models
from airflow.models import DAG, Variable, Connection
from airflow.models import DAG, DagBag, TaskInstance, TaskReschedule
from airflow.security import permissions, permissions as perms
from airflow.models.baseoperator import BaseOperator
from airflow.utils.db import provide_session
from datetime import datetime

# 创建一个“管理员”角色并授予权限
admin_role = models.Role(
    role_name="admin",
    permissions=[
        (permissions.ACTION_CAN_READ, permissions.RESOURCE_DAG),
        (permissions.ACTION_CAN_EDIT, permissions.RESOURCE_DAG),
        (permissions.ACTION_CAN_READ, permissions.RESOURCE_TASK_INSTANCE),
        (permissions.ACTION_CAN_EDIT, permissions.RESOURCE_TASK_INSTANCE),
    ],
)
# 创建一个名为“admin”的用户并将其分配给“管理员”角色。
admin_user = models.User(
    email='admin@example.com',
    username='admin',
    password='your_password',
    roles=[admin_role],
)

# 创建一个“开发”角色并授予权限
developer_role = models.Role(
    role_name="developer",
    permissions=[
        (permissions.ACTION_CAN_READ, permissions.RESOURCE_DAG),
        (permissions.ACTION_CAN_EDIT, permissions.RESOURCE_TASK_INSTANCE),
    ],
)
# 创建一个名为“dev”的用户并将其分配给“开发人员”角色。
dev_user = models.User(
    email='dev@example.com',
    username='dev',
    password='your_password',
    roles=[developer_role],
)
  1. 编写DAG,并添加访问控制: 使用DAG中的access_control方法实现访问控制。
from airflow import DAG
from airflow.operators.python_operator import PythonOperator

# 用于演示的任务,打印“Hello World!”
def my_task():
    print("Hello World!")

# 创建一个DAG,命名为“my_dag”,调度每天上午9点00分。
dag = DAG(
    "my_dag",
    default_args={"owner": "me"},
    schedule_interval="@daily 9:00",
)

# 添加一个名为“run_my_task”的任务,并将其设置为PythonOperator
run_my_task = PythonOperator(
    task_id="run_my_task",
    python_callable=my_task,
    dag=dag,
)

# DAG需要管理员角色才能读取和编辑它
dag.access_control = {
    "admin": {"can_read", "can_edit"},
    "developer": {"can_read"},
}

定义access_control之后,只有具有适当权限的用户才能编辑和读取DAG。 此外,还可以在DAG和Operator级别配置访问控制,然后将它们分配给角色。

相关内容

热门资讯

安卓系统怎么连不上carlif... 安卓系统无法连接CarLife的原因及解决方法随着智能手机的普及,CarLife这一车载互联功能为驾...
iwatch怎么连接安卓系统,... 你有没有想过,那款时尚又实用的iWatch,竟然只能和iPhone好上好?别急,今天就来给你揭秘,怎...
oppo手机安卓系统换成苹果系... OPPO手机安卓系统换成苹果系统:现实吗?如何操作?随着智能手机市场的不断发展,用户对于手机系统的需...
安卓平板改windows 系统... 你有没有想过,你的安卓平板电脑是不是也能变身成Windows系统的超级英雄呢?想象在同一个设备上,你...
iphone系统与安卓系统更新... 最近是不是你也遇到了这样的烦恼?手机更新系统总是失败,急得你团团转。别急,今天就来给你揭秘为什么iP...
安卓系统上滑按键,便捷生活与高... 你有没有发现,现在手机屏幕越来越大,操作起来却越来越方便了呢?这都得归功于安卓系统上的那些神奇的上滑...
安卓系统连接耳机模式,蓝牙、有... 亲爱的手机控们,你们有没有遇到过这种情况:手机突然变成了“耳机模式”,明明耳机没插,声音却只从耳机孔...
希沃系统怎么装安卓系统,解锁更... 亲爱的读者们,你是否也像我一样,对希沃一体机上的安卓系统充满了好奇呢?想象在教室里,你的希沃一体机不...
安装了Anaconda之后找不... 在安装Anaconda后,如果找不到Jupyter Notebook,可以尝试以下解决方法:检查环境...
安卓换鸿蒙系统会卡吗,体验流畅... 最近手机圈可是热闹非凡呢!不少安卓用户都在议论纷纷,说鸿蒙系统要来啦!那么,安卓手机换上鸿蒙系统后,...