def create_user(username: str, password: str, email: str, is_superuser: bool = True): """ Method to create Airflow user :param username: (str) - Airflow User name :param password: (str) - Airflow Password :param email: (str) - Email of the user :param is_superuser: (str) - Is super user - True/False Note: Modified date: 10-04-2021 Author: TB """ user = PasswordUser(models.User()) user.username = username user.email = email user.password = password user.superuser = is_superuser session = settings.Session() session.add(user) session.commit() session.close()
def create_superuser(): """ Creates a superuser (admin level) for accessing the web UI for Airflow :return: None """ # get path to config file relative to this file current_dir = os.path.dirname(__file__) filepath = os.path.join(current_dir, 'config.ini') # read config.ini file config = ConfigParser() config.read(filepath) # create User object user = PasswordUser(models.User()) user.username = config['web_authentication']['username'] user.email = config['web_authentication']['email'] user.password = config['web_authentication']['password'] user.superuser = True # Add user to Airflow session session = settings.Session() session.add(user) session.commit() session.close()
def generate_admin_user(password): """ Creates the new Airflow admin user. """ user = PasswordUser(models.User()) user.username = '******' user.email = get_email_address() user.password = password user.superuser = True return user
def make_user(username, email): user = PasswordUser(models.User()) user.username = username user.email = email user.superuser = True user.password = click.prompt("Password") session = settings.Session() session.add(user) session.commit() session.close()
def make_user(username, email, superuser, password): user = PasswordUser(models.User()) user.username = username user.email = email user.superuser = superuser user.password = password session = settings.Session() session.add(user) session.commit() session.close()
def add_default_user(session=None): from airflow import models from airflow.contrib.auth.backends.password_auth import PasswordUser user = PasswordUser(models.User()) user.username = '******' user.email = '*****@*****.**' user.password = '******' user.superuser = True session.add(user) session.commit()
def create_user(opts): from airflow.contrib.auth.backends.password_auth import PasswordUser from airflow import models, settings u = PasswordUser(models.User()) u.username = opts['username'] u.email = opts['email'] u.password = opts['password'] u.superuser = opts['super_user'] s = settings.Session() s.add(u) s.commit() s.close()
def createuser(): userdata = json.loads(request.data) username = userdata.get('user') or '' email = userdata.get('email') or '' password = userdata.get('password') or username + '123' superuser = userdata.get('superuser') or False if not superuser: superuser = 0 else: superuser = 1 data = {} try: if not username: data['status'] = '200' data[ 'message'] = "Username has missed, unable to create user in Airflow server" return jsonify(data) session = settings.Session() user = session.query( models.User).filter(models.User.username == username).first() if user: user.superuser = superuser session.commit() data['status'] = '200' data['message'] = 'SuperUser status is updated successfully!' else: user = PasswordUser(models.User()) user.username = username user.email = email user.password = password user.superuser = superuser session.add(user) session.commit() data['status'] = '200' data['message'] = "User Created successfully!" except Exception as e: data['status'] = '400' data['message'] = e.message return jsonify(data)
def createuser(): username = request.args.get('user') email = request.args.get('email', '') superuser = request.args.get('superuser', False) if superuser == 'False': superuser = 0 else: superuser = 1 try: user = PasswordUser(models.User()) user.username = username user.email = email user.password = username + '123' user.superuser = superuser session = settings.Session() session.add(user) session.commit() return jsonify("User Created") except: return jsonify("Raised Exception")
from airflow import models, settings from airflow.contrib.auth.backends.password_auth import PasswordUser from flask_bcrypt import generate_password_hash from sqlalchemy import exists session = settings.Session() isUserPresent = not session.query( exists().where(models.User.username == 'airflow')).scalar() if (isUserPresent): user = PasswordUser(models.User()) user.username = '******' user.email = 'airflow@fab_airflow.com' user.superuser = True user._password = generate_password_hash('airflow', 12) session.add(user) session.commit() session.close() exit()
import airflow from airflow import models, settings from airflow.contrib.auth.backends.password_auth import PasswordUser from sqlalchemy import create_engine user = PasswordUser(models.User()) user.username = '' user.email = '' user.password = '' # Make the value true if you want the user to be a administrator user.superuser = False engine = create_engine("postgresql://*****:*****@postgres:5432/airflow") session = settings.Session(bind=engine) session.add(user) session.commit() session.close() exit()
def create_airflow_rest_connection(): from airflow.contrib.auth.backends.password_auth import PasswordUser import base64 import os session = settings.Session() exists = session.query(models.User).filter(models.User.username == 'application').scalar() if exists is None: LoggingMixin().log.info("creating 'application' user for mini-BRS...") # create 'application' user random_key = str(base64.urlsafe_b64encode(os.urandom(32))) user = PasswordUser(models.User()) user.username = '******' user.email = '*****@*****.**' user.password = random_key session.add(user) session.commit() session.close() # create 'application' airflow connection rest = Connection( conn_id='rest', login='******', password=random_key ) session = settings.Session() session.add(rest) session.commit() session.close() # create 'admin' user # admin_password = str(base64.urlsafe_b64encode(os.urandom(32))) config_parser = configuration.AirflowConfigParser() config_parser.read( configuration.get_airflow_config( configuration.get_airflow_home() ) ) u = config_parser.get( section='core', key='username' ) p = config_parser.get( section='core', key='password' ) user = PasswordUser(models.User()) user.username = u user.email = '*****@*****.**' user.password = p user.superuser = True session = settings.Session() session.add(user) session.commit() session.close() config_parser.remove_option( section='core', option='username' ) config_parser.remove_option( section='core', option='password' ) file = open(configuration.get_airflow_config(configuration.get_airflow_home()), 'w') config_parser.write(file) file.close()