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