88 lines
2.3 KiB
Python
88 lines
2.3 KiB
Python
"""transfer common classes"""
|
|
from dataclasses import asdict, dataclass, field, is_dataclass
|
|
from enum import Enum
|
|
from typing import Any
|
|
from uuid import UUID
|
|
|
|
from django.core.serializers.json import DjangoJSONEncoder
|
|
|
|
from authentik.lib.models import SerializerModel
|
|
from authentik.lib.sentry import SentryIgnoredException
|
|
|
|
|
|
def get_attrs(obj: SerializerModel) -> dict[str, Any]:
|
|
"""Get object's attributes via their serializer, and covert it to a normal dict"""
|
|
data = dict(obj.serializer(obj).data)
|
|
to_remove = (
|
|
"policies",
|
|
"stages",
|
|
"pk",
|
|
"background",
|
|
"group",
|
|
"user",
|
|
"verbose_name",
|
|
"verbose_name_plural",
|
|
"object_type",
|
|
"flow_set",
|
|
"promptstage_set",
|
|
)
|
|
for to_remove_name in to_remove:
|
|
if to_remove_name in data:
|
|
data.pop(to_remove_name)
|
|
for key in list(data.keys()):
|
|
if key.endswith("_obj"):
|
|
data.pop(key)
|
|
return data
|
|
|
|
|
|
@dataclass
|
|
class FlowBundleEntry:
|
|
"""Single entry of a bundle"""
|
|
|
|
identifiers: dict[str, Any]
|
|
model: str
|
|
attrs: dict[str, Any]
|
|
|
|
@staticmethod
|
|
def from_model(
|
|
model: SerializerModel, *extra_identifier_names: str
|
|
) -> "FlowBundleEntry":
|
|
"""Convert a SerializerModel instance to a Bundle Entry"""
|
|
identifiers = {
|
|
"pk": model.pk,
|
|
}
|
|
all_attrs = get_attrs(model)
|
|
|
|
for extra_identifier_name in extra_identifier_names:
|
|
identifiers[extra_identifier_name] = all_attrs.pop(extra_identifier_name)
|
|
return FlowBundleEntry(
|
|
identifiers=identifiers,
|
|
model=f"{model._meta.app_label}.{model._meta.model_name}",
|
|
attrs=all_attrs,
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class FlowBundle:
|
|
"""Dataclass used for a full export"""
|
|
|
|
version: int = field(default=1)
|
|
entries: list[FlowBundleEntry] = field(default_factory=list)
|
|
|
|
|
|
class DataclassEncoder(DjangoJSONEncoder):
|
|
"""Convert FlowBundleEntry to json"""
|
|
|
|
def default(self, o):
|
|
if is_dataclass(o):
|
|
return asdict(o)
|
|
if isinstance(o, UUID):
|
|
return str(o)
|
|
if isinstance(o, Enum):
|
|
return o.value
|
|
return super().default(o) # pragma: no cover
|
|
|
|
|
|
class EntryInvalidError(SentryIgnoredException):
|
|
"""Error raised when an entry is invalid"""
|