80 lines
2.7 KiB
Python
80 lines
2.7 KiB
Python
import redis
|
|
import json
|
|
import traceback
|
|
from . launchpad_singleton import get_launchpad
|
|
import requests
|
|
from bs4 import BeautifulSoup
|
|
|
|
# Connect to Redis
|
|
cache = redis.Redis(host='localhost', port=6379, db=0)
|
|
|
|
def fetch_matrix_accounts(profile_id):
|
|
try:
|
|
#Try to fetch from cache first
|
|
cached_result = cache.get(f"matrix_{profile_id}")
|
|
if cached_result:
|
|
return json.loads(cached_result)
|
|
|
|
# Fetch the page
|
|
url = "https://launchpad.net/~" + profile_id
|
|
response = requests.get(url)
|
|
page_content = response.content
|
|
|
|
# Parse the page content
|
|
soup = BeautifulSoup(page_content, 'html.parser')
|
|
|
|
# Locate the elements containing the Matrix IDs
|
|
matrix_ids_elements = soup.find_all('dd', class_='user-social-accounts__item matrix-account')
|
|
|
|
# Extract the Matrix IDs
|
|
matrix_ids = []
|
|
for element in matrix_ids_elements:
|
|
matrix_id = element.find('a').text.strip()
|
|
matrix_ids.append(matrix_id)
|
|
# Cache the result with expiration time of 30 minutes (1800 seconds)
|
|
cache.setex(f"matrix_{profile_id}", 1800, json.dumps(matrix_ids))
|
|
return matrix_ids
|
|
|
|
except KeyError as e:
|
|
print(f"Profule with name {profile_id} was not found. Error: {e}")
|
|
print(traceback.format_exc())
|
|
return False
|
|
except Exception as e:
|
|
print(f"An error occurred: {e}")
|
|
print(traceback.format_exc())
|
|
return False
|
|
|
|
|
|
def fetch_group_members(group_name, recurse=False):
|
|
try:
|
|
#Try to fetch from cache first
|
|
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]
|
|
|
|
group_members = set()
|
|
for person in group.members:
|
|
print(person)
|
|
if not person.is_team:
|
|
group_members.add(person.name)
|
|
continue
|
|
|
|
# MXIDs should be generated for individuals only
|
|
mxids = ['@' + member + ':ubuntu.com' for member in group_members]
|
|
result = {'group_members': tuple(group_members), 'group_name': group_name, 'mxids': mxids}
|
|
|
|
# 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}")
|
|
print(traceback.format_exc())
|
|
return False
|
|
except Exception as e:
|
|
print(f"An error occurred: {e}")
|
|
print(traceback.format_exc())
|
|
return False
|