ephraimbuddy commented on code in PR #27266:
URL: https://github.com/apache/airflow/pull/27266#discussion_r1005532101
##########
airflow/utils/db.py:
##########
@@ -930,6 +930,36 @@ def check_conn_id_duplicates(session: Session) ->
Iterable[str]:
)
+def check_username_duplicates(session: Session) -> Iterable[str]:
+ """
+ Check unique username in User & RegisterUser table
+
+ :param session: session of the sqlalchemy
+ :rtype: str
+ """
+ from airflow.www.fab_security.sqla.models import RegisterUser, User
+
+ for model in [User, RegisterUser]:
+ dups = []
+ try:
+ dups = (
+ session.query(model.username) # type: ignore[attr-defined]
+ .group_by(model.username) # type: ignore[attr-defined]
+ .having(func.count() > 1)
+ .all()
+ )
+ except (exc.OperationalError, exc.ProgrammingError):
+ # fallback if tables hasn't been created yet
+ session.rollback()
+ if dups:
+ yield (
+ f'Seems you have non unique username in {model.__table__.name}
table.\n' # type: ignore
+ 'You have to manage those duplicate usernames '
Review Comment:
Addressed. You can take another look. Thanks
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
