paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-08-14 00:26:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

add split-manifest option to administration exporter

Browse Source
This commit is contained in:
Matthieu Helleboid
2023-01-14 11:54:54 +01:00
committed by Trenton H
parent 896304ccaa
commit 4cb4bd13ad
4 changed files with 66 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/documents/management/commands/document_exporter.py
View File

@@ -53,6 +53,14 @@ class Command(BaseCommand):
def add_arguments(self, parser):
parser.add_argument("target")
parser.add_argument(
"-sm",
"--split-manifest",
default=False,
action="store_true",
help="Export document information in individual manifest json files.",
)
parser.add_argument(
"-c",
"--compare-checksums",
@@ -125,6 +133,7 @@ class Command(BaseCommand):
def __init__(self, *args, **kwargs):
BaseCommand.__init__(self, *args, **kwargs)
self.target: Path = None
self.split_manifest = None
self.files_in_export_dir: Set[Path] = set()
self.exported_files: List[Path] = []
self.compare_checksums = False
@@ -137,6 +146,7 @@ class Command(BaseCommand):
def handle(self, *args, **options):
self.target = Path(options["target"]).resolve()
self.split_manifest = options["split_manifest"]
self.compare_checksums = options["compare_checksums"]
self.use_filename_format = options["use_filename_format"]
self.use_filename_prefix = options["use_filename_prefix"]
@@ -217,7 +227,8 @@ class Command(BaseCommand):
documents = Document.objects.order_by("id")
document_map = {d.pk: d for d in documents}
document_manifest = json.loads(serializers.serialize("json", documents))
manifest += document_manifest
if not self.split_manifest:
manifest += document_manifest
manifest += json.loads(
serializers.serialize("json", MailAccount.objects.all()),
@@ -334,6 +345,15 @@ class Command(BaseCommand):
archive_target,
)
if self.split_manifest:
manifest_name = base_name + "-manifest.json"
if self.use_filename_prefix:
manifest_name = os.path.join("json", manifest_name)
manifest_name = os.path.join(self.target, manifest_name)
os.makedirs(os.path.dirname(manifest_name), exist_ok=True)
with open(manifest_name, "w") as f:
json.dump([document_manifest[index]], f, indent=2)
# 4.1 write manifest to target folder
manifest_path = (self.target / Path("manifest.json")).resolve()
manifest_path.write_text(json.dumps(manifest, indent=2))