2024-04-05 02:45:27 +00:00
|
|
|
import redis
|
|
|
|
import json
|
2024-04-05 04:53:31 +00:00
|
|
|
import traceback
|
2024-04-05 02:45:27 +00:00
|
|
|
from bugtracker.launchpad_singleton import get_launchpad
|
|
|
|
|
|
|
|
# Connect to Redis
|
|
|
|
cache = redis.Redis(host='localhost', port=6379, db=0)
|
|
|
|
|
2024-04-05 03:49:37 +00:00
|
|
|
def fetch_individual_or_team_members(person_or_team, launchpad):
|
|
|
|
members = []
|
|
|
|
if person_or_team.is_team:
|
|
|
|
# Recursively fetch members for a team
|
|
|
|
group = launchpad.people[person_or_team.name]
|
|
|
|
for person in group.members:
|
|
|
|
if person.is_team:
|
|
|
|
ext = fetch_individual_or_team_members(person, launchpad)
|
|
|
|
if ext and not person.is_team:
|
2024-04-05 04:53:31 +00:00
|
|
|
members.append(person.name)
|
2024-04-05 03:49:37 +00:00
|
|
|
else:
|
2024-04-05 04:53:31 +00:00
|
|
|
members.append(person.name)
|
2024-04-05 03:49:37 +00:00
|
|
|
else:
|
2024-04-05 04:53:31 +00:00
|
|
|
members.append(person_or_team.name)
|
2024-04-05 03:49:37 +00:00
|
|
|
return members
|
|
|
|
|
2024-04-05 02:45:27 +00:00
|
|
|
def fetch_group_members(group_name):
|
|
|
|
try:
|
2024-04-05 03:49:37 +00:00
|
|
|
#Try to fetch from cache first
|
2024-04-05 02:45:27 +00:00
|
|
|
cached_result = cache.get(f"group_members_{group_name}")
|
|
|
|
if cached_result:
|
|
|
|
return json.loads(cached_result)
|
|
|
|
|
|
|
|
launchpad = get_launchpad()
|
|
|
|
group = launchpad.people[group_name]
|
2024-04-05 03:49:37 +00:00
|
|
|
|
2024-04-05 04:53:31 +00:00
|
|
|
group_members = set()
|
2024-04-05 03:49:37 +00:00
|
|
|
for person in group.members:
|
|
|
|
ext = fetch_individual_or_team_members(person, launchpad)
|
|
|
|
if ext:
|
2024-04-05 04:53:31 +00:00
|
|
|
for member in ext:
|
|
|
|
group_members.add(member)
|
2024-04-05 03:49:37 +00:00
|
|
|
|
|
|
|
# MXIDs should be generated for individuals only
|
2024-04-05 04:53:31 +00:00
|
|
|
print(group_members)
|
|
|
|
mxids = ['@' + member + ':ubuntu.com' for member in group_members]
|
|
|
|
result = {'group_members': tuple(group_members), 'group_name': group_name, 'mxids': mxids}
|
2024-04-05 02:45:27 +00:00
|
|
|
|
|
|
|
# Cache the result with expiration time of 30 minutes (1800 seconds)
|
|
|
|
cache.setex(f"group_members_{group_name}", 1800, json.dumps(result))
|
|
|
|
|
|
|
|
return result
|
|
|
|
except KeyError as e:
|
|
|
|
print(f"Group with name {group_name} was not found. Error: {e}")
|
2024-04-05 04:53:31 +00:00
|
|
|
print(traceback.format_exc())
|
2024-04-05 02:45:27 +00:00
|
|
|
return False
|
|
|
|
except Exception as e:
|
|
|
|
print(f"An error occurred: {e}")
|
2024-04-05 04:53:31 +00:00
|
|
|
print(traceback.format_exc())
|
2024-04-05 02:45:27 +00:00
|
|
|
return False
|