25 lines
1.0 KiB
Python
25 lines
1.0 KiB
Python
|
from pythorhead import Lemmy
|
||
|
|
from loguru import logger
|
||
|
|
import os
|
||
|
|
import secrets
|
||
|
|
import overseer.exceptions as e
|
||
|
|
|
||
|
|
overctrl_lemmy = Lemmy(f"https://{os.getenv('OVERSEER_LEMMY_DOMAIN')}")
|
||
|
|
_login = overctrl_lemmy.log_in(os.getenv('OVERSEER_LEMMY_USERNAME'),os.getenv('OVERSEER_LEMMY_PASSWORD'))
|
||
|
|
if not _login:
|
||
|
|
raise Exception("Failed to login to overctrl")
|
||
|
|
overseer_lemmy_user = overctrl_lemmy.user.get(username=os.getenv('OVERSEER_LEMMY_USERNAME'))
|
||
|
|
|
||
|
|
def pm_new_api_key(domain: str):
|
||
|
|
api_key = secrets.token_urlsafe(16)
|
||
|
|
pm_content = f"The API Key for domain {domain} is {api_key}.\n\nUse this to perform operations on the overseer."
|
||
|
|
domain_username = domain.replace(".", "_")
|
||
|
|
domain_user = overctrl_lemmy.user.get(username=domain_username)
|
||
|
|
if not domain_user:
|
||
|
|
raise e.BadRequest(f"Could not find domain user '{domain_username}'")
|
||
|
|
pm = overctrl_lemmy.private_message(pm_content,domain_user["person_view"]["person"]["id"])
|
||
|
|
if not pm:
|
||
|
|
raise e.BadRequest("API Key PM failed")
|
||
|
|
return api_key
|
||
|
|
|