A script to automatically setup a admin user for airflow which can be run at infrastructure setup time.

setup_airflow_auth.py

import os
import airflow
from airflow import models, settings
from airflow.contrib.auth.backends.password_auth import PasswordUser


def create_admin_user(uname, pwd, email):
  user = PasswordUser(models.User())
  user.username = uname
  user.email = email
  user.password = pwd
  session = settings.Session()
  session.add(user)
  session.commit()
  session.close()


if __name__ == '__main__':
  # you can generate username password dynamically or parse it using argparser
  create_admin_user(some_username, some_password, some_email)