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级别配置访问控制,然后将它们分配给角色。

相关内容

热门资讯

安卓换鸿蒙系统会卡吗,体验流畅... 最近手机圈可是热闹非凡呢!不少安卓用户都在议论纷纷,说鸿蒙系统要来啦!那么,安卓手机换上鸿蒙系统后,...
app安卓系统登录不了,解锁登... 最近是不是你也遇到了这样的烦恼:手机里那个心爱的APP,突然就登录不上了?别急,让我来帮你一步步排查...
安卓系统拦截短信在哪,安卓系统... 你是不是也遇到了这种情况:手机里突然冒出了很多垃圾短信,烦不胜烦?别急,今天就来教你怎么在安卓系统里...
安卓系统要维护多久,安卓系统维... 你有没有想过,你的安卓手机里那个陪伴你度过了无数日夜的安卓系统,它究竟要陪伴你多久呢?这个问题,估计...
windows官网系统多少钱 Windows官网系统价格一览:了解正版Windows的购买成本Windows 11官方价格解析微软...
安卓系统如何卸载app,轻松掌... 手机里的App越来越多,是不是感觉内存不够用了?别急,今天就来教你怎么轻松卸载安卓系统里的App,让...
怎么复制照片安卓系统,操作步骤... 亲爱的手机控们,是不是有时候想把自己的手机照片分享给朋友,或者备份到电脑上呢?别急,今天就来教你怎么...
安卓系统应用怎么重装,安卓应用... 手机里的安卓应用突然罢工了,是不是让你头疼不已?别急,今天就来手把手教你如何重装安卓系统应用,让你的...
iwatch怎么连接安卓系统,... 你有没有想过,那款时尚又实用的iWatch,竟然只能和iPhone好上好?别急,今天就来给你揭秘,怎...
iphone系统与安卓系统更新... 最近是不是你也遇到了这样的烦恼?手机更新系统总是失败,急得你团团转。别急,今天就来给你揭秘为什么iP...