Skip to content

Instantly share code, notes, and snippets.

@kouk

kouk/airflow_set_password.py

Created April 5, 2021 15:52
Show Gist options
  • Save kouk/998195bd59a50e32b688ce4421336c41 to your computer and use it in GitHub Desktop.
Save kouk/998195bd59a50e32b688ce4421336c41 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. kouk created this gist Apr 5, 2021.
    20 changes: 20 additions & 0 deletions airflow_set_password.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,20 @@
    import airflow
    import getpass
    import logging
    from airflow import models, settings
    from airflow.contrib.auth.backends.password_auth import PasswordUser


    def set_password(name, password):
    session = settings.Session()
    u = session.query(PasswordUser).filter_by(username=name).first()
    u.password = password
    session.add(u)
    session.commit()
    session.close()

    if __name__ == "__main__":
    logging.basicConfig()
    logging.getLogger('sqlalchemy.engine').setLevel(logging.INFO)

    set_password("johndoe", getpass.getpass())