kopia lustrzana https://dev.funkwhale.audio/funkwhale/funkwhale
76 wiersze
2.1 KiB
Python
76 wiersze
2.1 KiB
Python
import datetime
|
|
import logging
|
|
|
|
from django.conf import settings
|
|
from django.utils import timezone
|
|
|
|
from funkwhale_api.common import preferences, session
|
|
from funkwhale_api.users import models as users_models
|
|
|
|
from . import keys, models, serializers
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def get_actor_data(actor_url):
|
|
logger.debug("Fetching actor %s", actor_url)
|
|
response = session.get_session().get(
|
|
actor_url,
|
|
headers={"Accept": "application/activity+json"},
|
|
)
|
|
response.raise_for_status()
|
|
try:
|
|
return response.json()
|
|
except Exception:
|
|
raise ValueError(f"Invalid actor payload: {response.text}")
|
|
|
|
|
|
def get_actor(fid, skip_cache=False):
|
|
if not skip_cache:
|
|
try:
|
|
actor = models.Actor.objects.select_related().get(fid=fid)
|
|
except models.Actor.DoesNotExist:
|
|
actor = None
|
|
fetch_delta = datetime.timedelta(
|
|
minutes=preferences.get("federation__actor_fetch_delay")
|
|
)
|
|
if actor and actor.last_fetch_date > timezone.now() - fetch_delta:
|
|
# cache is hot, we can return as is
|
|
return actor
|
|
data = get_actor_data(fid)
|
|
serializer = serializers.ActorSerializer(data=data)
|
|
serializer.is_valid(raise_exception=True)
|
|
|
|
return serializer.save(last_fetch_date=timezone.now())
|
|
|
|
|
|
_CACHE = {}
|
|
|
|
|
|
def get_service_actor(cache=True):
|
|
if cache and "service_actor" in _CACHE:
|
|
return _CACHE["service_actor"]
|
|
|
|
name, domain = (
|
|
settings.FEDERATION_SERVICE_ACTOR_USERNAME,
|
|
settings.FEDERATION_HOSTNAME,
|
|
)
|
|
try:
|
|
actor = models.Actor.objects.select_related().get(
|
|
preferred_username=name, domain__name=domain
|
|
)
|
|
except models.Actor.DoesNotExist:
|
|
pass
|
|
else:
|
|
_CACHE["service_actor"] = actor
|
|
return actor
|
|
|
|
args = users_models.get_actor_data(name)
|
|
private, public = keys.get_key_pair()
|
|
args["private_key"] = private.decode("utf-8")
|
|
args["public_key"] = public.decode("utf-8")
|
|
args["type"] = "Service"
|
|
actor = models.Actor.objects.create(**args)
|
|
_CACHE["service_actor"] = actor
|
|
return actor
|