Updated profile interaction methods
This commit is contained in:
24
GLPIAPI.py
24
GLPIAPI.py
@@ -14,7 +14,7 @@ class GLPIAPI:
|
||||
self.ActiveProfile = None
|
||||
self.InitConnection()
|
||||
if(Profile == None):
|
||||
self.ActiveProfile = self.GetActiveProfile()
|
||||
self.ActiveProfile = self.GetActiveProfile()[0]
|
||||
else:
|
||||
self.ActiveProfile = Profile
|
||||
profileId = self.GetItems('Profile', 'Profile.name', Profile)[1]['2']
|
||||
@@ -39,9 +39,12 @@ class GLPIAPI:
|
||||
"Session-Token": self.SessionToken,
|
||||
"App-Token": self.AppToken
|
||||
}
|
||||
if(self.GetActiveProfile() != self.ActiveProfile and self.ActiveProfile != None):
|
||||
profileId = self.GetItems('Profile', 'Profile.name', self.ActiveProfile)[1]['2']
|
||||
self.SetActiveProfile(profileId)
|
||||
if(self.GetActiveProfile()[0] != self.ActiveProfile and self.ActiveProfile != None):
|
||||
profile = [p for p in self.GetUserProfiles()['myprofiles'] if p["name"] == self.ActiveProfile]
|
||||
if(profile != []):
|
||||
self.SetActiveProfile(profile[0]["id"])
|
||||
else:
|
||||
raise Exception(f"{self.ActiveProfile} is not in user profiles")
|
||||
else:
|
||||
raise Exception(f"{result.status_code} - {result.json()[0]}")
|
||||
|
||||
@@ -50,7 +53,7 @@ class GLPIAPI:
|
||||
result = requests.get(sessionUri, headers=self.Headers)
|
||||
if(result.status_code != 200 and result.json()[0] == 'ERROR_SESSION_TOKEN_INVALID'):
|
||||
self.InitConnection()
|
||||
return
|
||||
return result.status_code
|
||||
|
||||
def GetUserProfiles(self):
|
||||
self.CheckConnection()
|
||||
@@ -66,12 +69,19 @@ class GLPIAPI:
|
||||
uri = f"{self.Server}/apirest.php/getActiveProfile"
|
||||
req = requests.get(uri, headers=self.Headers)
|
||||
if(req.status_code == 200):
|
||||
return req.json()['active_profile']['name']
|
||||
return (req.json()['active_profile']['name'],req.json()['active_profile']['id'])
|
||||
else:
|
||||
return req.status_code
|
||||
|
||||
def SetActiveProfile(self, profileId):
|
||||
def SetActiveProfile(self, profile):
|
||||
self.CheckConnection()
|
||||
if(type(profile) == str):
|
||||
profileFound = [p for p in self.GetUserProfiles()['myprofiles'] if p["name"] == profile]
|
||||
if(profileFound != []):
|
||||
print(profileFound)
|
||||
profileId = profileFound[0]["id"]
|
||||
else:
|
||||
profileId = profile
|
||||
body = {
|
||||
"profiles_id" : profileId
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user