Skip to content

Commit

Permalink
consider realm when activating project by name; also added ability to…
Browse files Browse the repository at this point in the history
… filter projects by realm
  • Loading branch information
dlohmeier authored and jthurner committed Jul 11, 2024
1 parent a615933 commit 30fba2e
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions pandahub/lib/PandaHub.py
Original file line number Diff line number Diff line change
Expand Up @@ -257,15 +257,16 @@ def delete_project(self, i_know_this_action_is_final=False, project_id=None):
self.mongo_client.user_management.projects.delete_one({"_id": project_id})
self.active_project = None

def get_projects(self):
def get_projects(self, realm: Optional[str] = None):
filter_dict = {}
if self.user_id is not None:
user = self._get_user()
if user["is_superuser"]:
filter_dict = {}
else:
if not user["is_superuser"]:
filter_dict = {"users.{}".format(self.user_id): {"$exists": True}}
else:
filter_dict = {"users": {"$exists": False}}
if realm is not None:
filter_dict["realm"] = realm
db = self.mongo_client["user_management"]
projects = list(db["projects"].find(filter_dict))
return [
Expand All @@ -287,7 +288,7 @@ def get_projects(self):
]

def set_active_project(self, project_name:str, realm=None):
projects = self.get_projects()
projects = self.get_projects(realm=realm)
active_projects = [
project for project in projects if project["name"] == project_name
]
Expand Down

0 comments on commit 30fba2e

Please sign in to comment.