chore(metrics): Added more metrics

uc-cdis · Jun 26, 2024 · 8a84593 · 8a84593
1 parent e38798d
commit 8a84593
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 8 deletions.
diff --git a/fence/auth.py b/fence/auth.py
@@ -132,7 +132,12 @@ def set_flask_session_values(user):
     current_app.scoped_session().add(user)
     current_app.scoped_session().commit()
 
-    from fence.metrics import login_counter
+    from fence.metrics import (
+        login_counter,
+        fence_login_counter,
+        google_login_counter,
+        ras_login_counter,
+    )
 
     login_counter.inc()
 

diff --git a/fence/blueprints/login/base.py b/fence/blueprints/login/base.py
@@ -179,6 +179,3 @@ def _login(username, idp_name, email=None, id_from_idp=None):
     if flask.session.get("redirect"):
         return flask.redirect(flask.session.get("redirect"))
     return flask.jsonify({"username": username})
-    from fence.metrics import login_counter
-
-    login_counter.inc()
diff --git a/fence/blueprints/login/fence_login.py b/fence/blueprints/login/fence_login.py
@@ -124,8 +124,6 @@ def get(self):
         )
         self.post_login()
 
-        from fence.metrics import fence_login_counter
-
         fence_login_counter.inc()
 
         if config["REGISTER_USERS_ON"]:

diff --git a/fence/blueprints/login/ras.py b/fence/blueprints/login/ras.py
@@ -110,8 +110,6 @@ def post_login(self, user=None, token_result=None, id_from_idp=None):
                 expires,
             )
 
-        from fence.metrics import ras_login_counter
-
         ras_login_counter.inc()
 
         flask.current_app.ras_client.store_refresh_token(