41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
from collections import Counter
|
|
from datetime import UTC, datetime
|
|
|
|
from django.conf import settings
|
|
|
|
from netbox_export.models import InstanceIdentity
|
|
|
|
from .archive import build_archive
|
|
from .codec import serialize_object
|
|
from .graph import ObjectGraph, seed_scope
|
|
|
|
|
|
def export_scope(scope_type: str, scope_id: int, *, max_objects: int, signing_key: str = ""):
|
|
root, seeds = seed_scope(scope_type, scope_id)
|
|
graph = ObjectGraph(max_objects=max_objects).collect(seeds)
|
|
objects = graph.objects
|
|
assets = {}
|
|
records = [
|
|
serialize_object(obj, set(objects), assets)
|
|
for _, obj in sorted(objects.items())
|
|
]
|
|
counts = Counter(record["model"] for record in records)
|
|
manifest = {
|
|
"created_at": datetime.now(UTC).isoformat(),
|
|
"source_instance": str(InstanceIdentity.local_id()),
|
|
"source_netbox_version": getattr(getattr(settings, "RELEASE", None), "version", "4.6"),
|
|
"plugin_version": "0.2.0",
|
|
"scope": {
|
|
"type": scope_type,
|
|
"source_pk": str(scope_id),
|
|
"label": str(root),
|
|
},
|
|
"object_count": len(records),
|
|
"member_count": len(graph.members),
|
|
"dependency_count": len(graph.dependencies),
|
|
"models": dict(sorted(counts.items())),
|
|
}
|
|
return build_archive(manifest, records, assets, signing_key=signing_key), manifest
|