98 lines
3.1 KiB
Python
98 lines
3.1 KiB
Python
import argparse
|
|
import sys
|
|
from collections.abc import Sequence
|
|
from pathlib import Path
|
|
|
|
from copienator import (
|
|
EvaluationWorkspace,
|
|
ExitCode,
|
|
evaluation_parser,
|
|
execute,
|
|
workspace_from_args,
|
|
)
|
|
from copienator.configuration import EXPORT_DIR
|
|
from copienator.platform import replace_with_link_or_copy
|
|
|
|
ANNOTATION_DIRECTORIES = ("BGnot", "Bnot", "Anot")
|
|
|
|
|
|
def export_directory(
|
|
workspace: EvaluationWorkspace,
|
|
source_dir_name: str,
|
|
) -> ExitCode:
|
|
workspace.require_directories(source_dir_name)
|
|
source_dir = workspace.annotation_dir("refaire") if source_dir_name == "BRnot" else workspace.root / source_dir_name
|
|
session_id = workspace.refaire_session_id if source_dir_name == "BRnot" else None
|
|
prefix = f"{session_id}__" if session_id else ""
|
|
sync_dir = Path(EXPORT_DIR).expanduser() / workspace.name
|
|
if session_id:
|
|
sync_dir /= session_id
|
|
sync_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
subdirs = [directory for directory in source_dir.iterdir() if directory.is_dir()]
|
|
if source_dir_name == "BGnot" and subdirs:
|
|
all_start_with_copie = all(directory.name.startswith("Copie") for directory in subdirs)
|
|
if not all_start_with_copie:
|
|
subdirs = [directory for directory in subdirs if not directory.name.startswith("Copie")]
|
|
|
|
missing_outputs = 0
|
|
for subdir in subdirs:
|
|
concat_file = next(
|
|
(
|
|
candidate
|
|
for candidate in (subdir / "Concat.pdf", subdir / "Concat.jpg")
|
|
if candidate.is_file()
|
|
),
|
|
None,
|
|
)
|
|
if concat_file is None:
|
|
print(
|
|
f"Warning: no Concat.pdf or Concat.jpg found in {subdir}",
|
|
file=sys.stderr,
|
|
)
|
|
missing_outputs += 1
|
|
continue
|
|
destination = sync_dir / f"{prefix}{subdir.name}{concat_file.suffix.lower()}"
|
|
method = replace_with_link_or_copy(concat_file, destination, prefer="hardlink")
|
|
print(f"Exported: {destination} ({method})")
|
|
return ExitCode.PARTIAL if missing_outputs else ExitCode.SUCCESS
|
|
|
|
|
|
def build_parser() -> argparse.ArgumentParser:
|
|
parser = evaluation_parser("Export annotated PDFs to the tablet directory.")
|
|
parser.add_argument(
|
|
"annotation_dir",
|
|
nargs="?",
|
|
choices=ANNOTATION_DIRECTORIES,
|
|
default="BGnot",
|
|
help="Annotation directory to export (default: BGnot)",
|
|
)
|
|
parser.add_argument("--refaire", action="store_true", help="Process only copies/labels defined in refaire.json")
|
|
return parser
|
|
|
|
|
|
def run(
|
|
workspace: EvaluationWorkspace,
|
|
*,
|
|
annotation_dir: str = "BGnot",
|
|
refaire: bool = False,
|
|
) -> ExitCode:
|
|
return export_directory(workspace, "BRnot" if refaire else annotation_dir)
|
|
|
|
|
|
def main(argv: Sequence[str] | None = None) -> int:
|
|
parser = build_parser()
|
|
return execute(
|
|
parser,
|
|
argv,
|
|
lambda args: run(
|
|
workspace_from_args(args),
|
|
annotation_dir=args.annotation_dir,
|
|
refaire=args.refaire,
|
|
),
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|