support airflow 2

fab_oidc/security.py

@@ -20,13 +20,22 @@ class OIDCSecurityManager(OIDCSecurityManagerMixin, SecurityManager):
 
 
 try:
-    from airflow.www_rbac.security import AirflowSecurityManager
+    from airflow.www.security import AirflowSecurityManager
 
     class AirflowOIDCSecurityManager(OIDCSecurityManagerMixin,
                                      AirflowSecurityManager):
         pass
 except ImportError:
-    log.debug('Airflow not installed')
+    log.debug('Airflow 2 not installed')
+
+    try:
+        from airflow.www_rbac.security import AirflowSecurityManager
+
+        class AirflowOIDCSecurityManager(OIDCSecurityManagerMixin,
+                                         AirflowSecurityManager):
+            pass
+    except ImportError:
+        log.debug('Airflow 1 not installed')
 
 try:
     from superset.security import SupersetSecurityManager

setup.py

@@ -17,7 +17,7 @@ def desc():
 
 setup(
     name='fab_oidc',
-    version='0.0.9',
+    version='0.1.0',
     url='https://github.com/ministryofjustice/fab-oidc/',
     license='MIT',
     author='ministryofjustice',