- return response_users\r
-\r
- # create grafana team with member\r
- def create_grafana_teams_members(self, project_name, user_name, is_admin, proj_list):\r
- # check if user exist in grafana or not\r
- user_response = requests.request("GET", self.url + "/api/users/lookup?loginOrEmail={}".format(user_name),\r
- headers=self.headers)\r
- user_obj = json.loads(user_response.text)\r
- if user_response.status_code != 200:\r
- user_response = self.create_grafana_users(user_name)\r
- user_obj = json.loads(user_response.text)\r
-\r
- user_id = user_obj["id"]\r
-\r
- # Get Teams for user\r
- team_objs = requests.request("GET", self.url + "/api/users/{}/teams".format(user_id), headers=self.headers)\r
- team_obj = json.loads(team_objs.text)\r
- team_list = []\r
- if len(team_obj):\r
- for team in team_obj:\r
- team_list.append(team["name"])\r
-\r
- proj_unlink = set(team_list) - set(proj_list)\r
- for prj in proj_unlink:\r
- response_team = requests.request("GET", self.url + "/api/teams/search?name={}".format(prj),\r
- headers=self.headers)\r
- team_id = json.loads(response_team.text)["teams"][0]["id"]\r
- requests.request("DELETE", self.url + "/api/teams/{}/members/{}".format(team_id, user_id),\r
- headers=self.headers)\r
- # add member to team\r
- response_team = requests.request("GET", self.url + "/api/teams/search?name={}".format(project_name),\r
- headers=self.headers)\r
- team_id = json.loads(response_team.text)["teams"][0]["id"]\r
- if project_name not in team_list:\r
- member_payload = {\r
- "userId": user_id\r
- }\r
- requests.request("POST", self.url + "/api/teams/{}/members".format(team_id), json=member_payload,\r
- headers=self.headers)\r
- # if role is admin change permission to admin\r
- if is_admin:\r
- url = "/api/org/users/{}/".format(user_id)\r
- permission_payload = {"role": "Admin", }\r
- requests.request("PATCH", self.url + url, json=permission_payload, headers=self.headers)\r
+ return response_users
+
+ # Get Grafana users
+ def get_grafana_users(self):
+ response_users = requests.request(
+ "GET",
+ self.url + "/api/users",
+ headers=self.headers,
+ )
+ user_list = []
+ users = json.loads(response_users.text)
+ for user in users:
+ if user["name"] and user["name"] != "admin":
+ user_list.append(user["name"])
+ return user_list
+
+ # Create Grafana team with member
+ def create_grafana_teams_members(
+ self, project_name, user_name, is_admin, proj_list
+ ):
+ # Check if user exist in Grafana
+ user_response = requests.request(
+ "GET",
+ self.url + "/api/users/lookup?loginOrEmail={}".format(user_name),
+ headers=self.headers,
+ )
+ user_obj = json.loads(user_response.text)
+ if user_response.status_code != 200:
+ user_response = self.create_grafana_users(user_name)
+ user_obj = json.loads(user_response.text)
+
+ user_id = user_obj["id"]
+
+ # Get teams for user
+ team_objs = requests.request(
+ "GET",
+ self.url + "/api/users/{}/teams".format(user_id),
+ headers=self.headers,
+ )
+ team_obj = json.loads(team_objs.text)
+ team_list = []
+ if len(team_obj):
+ for team in team_obj:
+ team_list.append(team["name"])
+
+ proj_unlink = set(team_list) - set(proj_list)
+ for prj in proj_unlink:
+ response_team = requests.request(
+ "GET",
+ self.url + "/api/teams/search?name={}".format(prj),
+ headers=self.headers,
+ )
+ team_id = json.loads(response_team.text)["teams"][0]["id"]
+ requests.request(
+ "DELETE",
+ self.url + "/api/teams/{}/members/{}".format(team_id, user_id),
+ headers=self.headers,
+ )
+ if project_name != "admin":
+ # Add member to team
+ response_team = requests.request(
+ "GET",
+ self.url + "/api/teams/search?name={}".format(project_name),
+ headers=self.headers,
+ )
+
+ # Search if team in Grafana corresponding to the project already exists
+ if not json.loads(response_team.text)["teams"]:
+ self.create_grafana_teams(project_name)
+ response_team = requests.request(
+ "GET",
+ self.url + "/api/teams/search?name={}".format(project_name),
+ headers=self.headers,
+ )
+ team_id = json.loads(response_team.text)["teams"][0]["id"]
+ if project_name not in team_list:
+ # Create a team in Grafana corresponding to the project as it doesn't exist
+ member_payload = {"userId": user_id}
+ requests.request(
+ "POST",
+ self.url + "/api/teams/{}/members".format(team_id),
+ json=member_payload,
+ headers=self.headers,
+ )
+ # Check if user role or project name is admin
+ if is_admin or project_name == "admin":
+ # Give admin righsts to user
+ url = "/api/org/users/{}/".format(user_id)
+ permission_payload = {
+ "role": "Admin",
+ }
+ requests.request(
+ "PATCH", self.url + url, json=permission_payload, headers=self.headers
+ )