PDF merge
This commit is contained in:
bvn13
commit
69a38caa79
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
@ -0,0 +1 @@
|
|||||||
|
троицкое
|
||||||
76
README.md
Normal file
76
README.md
Normal file
@ -0,0 +1,76 @@
|
|||||||
|
# pdf-merge
|
||||||
|
|
||||||
|
Скрипт для сборки PDF из страниц нескольких файлов по YAML-конфигу.
|
||||||
|
Форматы страниц (A4, A3 и др.) сохраняются без изменений.
|
||||||
|
|
||||||
|
## Требования
|
||||||
|
|
||||||
|
- [uv](https://docs.astral.sh/uv/getting-started/installation/) — менеджер Python-окружений
|
||||||
|
```bash
|
||||||
|
curl -LsSf https://astral.sh/uv/install.sh | sh
|
||||||
|
# или на Arch:
|
||||||
|
sudo pacman -S uv
|
||||||
|
```
|
||||||
|
|
||||||
|
Зависимости (`pypdf`, `pyyaml`) устанавливаются автоматически при первом запуске.
|
||||||
|
|
||||||
|
## Использование
|
||||||
|
|
||||||
|
```bash
|
||||||
|
chmod +x run.sh
|
||||||
|
|
||||||
|
# Запуск с конфигом по умолчанию (merge.yaml в текущей папке)
|
||||||
|
./run.sh
|
||||||
|
|
||||||
|
# Явно указать конфиг
|
||||||
|
./run.sh my_project.yaml
|
||||||
|
|
||||||
|
# Или напрямую через uv
|
||||||
|
uv run pdf_merge.py my_project.yaml
|
||||||
|
```
|
||||||
|
|
||||||
|
## Конфиг (merge.yaml)
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
output: result.pdf # куда сохранить результат
|
||||||
|
|
||||||
|
pages:
|
||||||
|
- file: document_a4.pdf
|
||||||
|
pages: "1-5" # диапазон
|
||||||
|
|
||||||
|
- file: drawing_a3.pdf
|
||||||
|
pages: "1" # одна страница
|
||||||
|
|
||||||
|
- file: document_a4.pdf
|
||||||
|
pages: "3-15,20" # диапазон + отдельная страница
|
||||||
|
|
||||||
|
- file: appendix.pdf
|
||||||
|
pages: [1, 2, 5] # YAML-список
|
||||||
|
label: "приложение" # опциональная метка для читаемости
|
||||||
|
|
||||||
|
- file: other.pdf # pages не указан — берёт все страницы
|
||||||
|
```
|
||||||
|
|
||||||
|
### Форматы поля `pages`
|
||||||
|
|
||||||
|
| Запись | Результат |
|
||||||
|
|----------------|----------------------------------|
|
||||||
|
| `"1-5"` | страницы 1, 2, 3, 4, 5 |
|
||||||
|
| `"1,3,7"` | страницы 1, 3, 7 |
|
||||||
|
| `"1-3,7,9-11"` | страницы 1, 2, 3, 7, 9, 10, 11 |
|
||||||
|
| `[1, 3, 5]` | страницы 1, 3, 5 |
|
||||||
|
| `4` | страница 4 |
|
||||||
|
| `"all"` / пусто| все страницы файла |
|
||||||
|
|
||||||
|
Нумерация с **1** (как в PDF-ридере).
|
||||||
|
|
||||||
|
## Пример вывода
|
||||||
|
|
||||||
|
```
|
||||||
|
Конфиг: merge.yaml
|
||||||
|
+ document_a4.pdf [титул + оглавление]: страниц 2 (из 42) — индексы [1, 2]
|
||||||
|
+ drawing_a3.pdf [чертёж А3]: страниц 1 (из 3) — индексы [1]
|
||||||
|
+ document_a4.pdf [основной текст]: страниц 14 (из 42) — индексы [3, 4, ..., 20]
|
||||||
|
|
||||||
|
Готово: result.pdf (17 стр.)
|
||||||
|
```
|
||||||
39
merge.yaml
Normal file
39
merge.yaml
Normal file
@ -0,0 +1,39 @@
|
|||||||
|
# Итоговый файл (путь относительно этого конфига или абсолютный)
|
||||||
|
output: result.pdf
|
||||||
|
|
||||||
|
# Порядок страниц в итоговом PDF.
|
||||||
|
# Поле "pages" может быть:
|
||||||
|
# - "all" — все страницы файла (значение по умолчанию если поле опущено)
|
||||||
|
# - "1-5" — диапазон
|
||||||
|
# - "1,3,7" — конкретные страницы через запятую
|
||||||
|
# - "1-3,7,9-11" — смешанный формат
|
||||||
|
# - [1, 3, 5] — список чисел (YAML-список)
|
||||||
|
# - 4 — одна страница числом
|
||||||
|
# Нумерация страниц — с 1 (как в PDF-ридере).
|
||||||
|
# Поле "label" опциональное — только для читаемости конфига.
|
||||||
|
|
||||||
|
pages:
|
||||||
|
# Титульный лист и оглавление из первого документа
|
||||||
|
- file: document_a4.pdf
|
||||||
|
pages: "1-2"
|
||||||
|
label: "титул + оглавление"
|
||||||
|
|
||||||
|
# Разворот чертежа в формате A3
|
||||||
|
- file: drawing_a3.pdf
|
||||||
|
pages: "1"
|
||||||
|
label: "чертёж А3"
|
||||||
|
|
||||||
|
# Основной текст — страницы 3–15 и отдельно страница 20
|
||||||
|
- file: document_a4.pdf
|
||||||
|
pages: "3-15,20"
|
||||||
|
label: "основной текст"
|
||||||
|
|
||||||
|
# Ещё один чертёж А3 — все страницы
|
||||||
|
- file: drawing_a3.pdf
|
||||||
|
pages: all
|
||||||
|
label: "все чертежи"
|
||||||
|
|
||||||
|
# Приложение из третьего файла — конкретные страницы списком
|
||||||
|
- file: appendix.pdf
|
||||||
|
pages: [1, 2, 5]
|
||||||
|
label: "приложение"
|
||||||
155
pdf_merge.py
Executable file
155
pdf_merge.py
Executable file
@ -0,0 +1,155 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# /// script
|
||||||
|
# requires-python = ">=3.10"
|
||||||
|
# dependencies = [
|
||||||
|
# "pypdf>=4.0.0",
|
||||||
|
# "pyyaml>=6.0",
|
||||||
|
# ]
|
||||||
|
# ///
|
||||||
|
"""
|
||||||
|
pdf_merge.py — собирает PDF из кусков других PDF по YAML-конфигу.
|
||||||
|
|
||||||
|
Использование:
|
||||||
|
uv run pdf_merge.py [config.yaml]
|
||||||
|
|
||||||
|
Если конфиг не указан явно — ищет merge.yaml в текущей директории.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import sys
|
||||||
|
import re
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
import yaml
|
||||||
|
from pypdf import PdfReader, PdfWriter
|
||||||
|
|
||||||
|
|
||||||
|
# ---------------------------------------------------------------------------
|
||||||
|
# Парсинг спецификации страниц
|
||||||
|
# ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
def parse_pages(spec: str | int | list, total: int) -> list[int]:
|
||||||
|
"""
|
||||||
|
Принимает спецификацию страниц и возвращает список 0-based индексов.
|
||||||
|
|
||||||
|
Форматы (1-based нумерация в конфиге):
|
||||||
|
"1-5" -> [0, 1, 2, 3, 4]
|
||||||
|
"1,3,5" -> [0, 2, 4]
|
||||||
|
"1-3,7,9-11" -> [0, 1, 2, 6, 8, 9, 10]
|
||||||
|
"all" -> все страницы
|
||||||
|
3 -> [2] (одно число как int)
|
||||||
|
[1, 3, 5] -> [0, 2, 4] (список чисел)
|
||||||
|
"""
|
||||||
|
if spec is None or str(spec).strip().lower() == "all":
|
||||||
|
return list(range(total))
|
||||||
|
|
||||||
|
if isinstance(spec, int):
|
||||||
|
spec = str(spec)
|
||||||
|
|
||||||
|
if isinstance(spec, list):
|
||||||
|
spec = ",".join(str(s) for s in spec)
|
||||||
|
|
||||||
|
pages = []
|
||||||
|
for part in re.split(r"[,;]\s*", str(spec).strip()):
|
||||||
|
part = part.strip()
|
||||||
|
if not part:
|
||||||
|
continue
|
||||||
|
m = re.fullmatch(r"(\d+)\s*-\s*(\d+)", part)
|
||||||
|
if m:
|
||||||
|
start, end = int(m.group(1)), int(m.group(2))
|
||||||
|
if start < 1 or end > total or start > end:
|
||||||
|
raise ValueError(
|
||||||
|
f"Диапазон {part} выходит за границы документа (всего {total} стр.)"
|
||||||
|
)
|
||||||
|
pages.extend(range(start - 1, end))
|
||||||
|
elif re.fullmatch(r"\d+", part):
|
||||||
|
n = int(part)
|
||||||
|
if n < 1 or n > total:
|
||||||
|
raise ValueError(
|
||||||
|
f"Страница {n} не существует (всего {total} стр.)"
|
||||||
|
)
|
||||||
|
pages.append(n - 1)
|
||||||
|
else:
|
||||||
|
raise ValueError(f"Не удалось разобрать спецификацию страниц: '{part}'")
|
||||||
|
|
||||||
|
return pages
|
||||||
|
|
||||||
|
|
||||||
|
# ---------------------------------------------------------------------------
|
||||||
|
# Основная логика
|
||||||
|
# ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
def merge(config_path: Path) -> None:
|
||||||
|
with open(config_path, encoding="utf-8") as f:
|
||||||
|
cfg = yaml.safe_load(f)
|
||||||
|
|
||||||
|
output_path = Path(cfg.get("output", "output.pdf"))
|
||||||
|
base_dir = config_path.parent # файлы ищем относительно конфига
|
||||||
|
|
||||||
|
steps: list[dict] = cfg.get("pages", [])
|
||||||
|
if not steps:
|
||||||
|
print("Ошибка: в конфиге нет секции 'pages'.", file=sys.stderr)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
writer = PdfWriter()
|
||||||
|
total_added = 0
|
||||||
|
|
||||||
|
for i, step in enumerate(steps, 1):
|
||||||
|
if "file" not in step:
|
||||||
|
print(f"Шаг {i}: нет поля 'file', пропускаю.", file=sys.stderr)
|
||||||
|
continue
|
||||||
|
|
||||||
|
pdf_path = base_dir / step["file"]
|
||||||
|
if not pdf_path.exists():
|
||||||
|
print(f"Шаг {i}: файл не найден: {pdf_path}", file=sys.stderr)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
reader = PdfReader(str(pdf_path))
|
||||||
|
total_pages = len(reader.pages)
|
||||||
|
spec = step.get("pages", "all")
|
||||||
|
|
||||||
|
try:
|
||||||
|
indices = parse_pages(spec, total_pages)
|
||||||
|
except ValueError as e:
|
||||||
|
print(f"Шаг {i} ({step['file']}): {e}", file=sys.stderr)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
for idx in indices:
|
||||||
|
writer.add_page(reader.pages[idx])
|
||||||
|
|
||||||
|
label = step.get("label", "")
|
||||||
|
label_str = f" [{label}]" if label else ""
|
||||||
|
print(
|
||||||
|
f" + {step['file']}{label_str}: "
|
||||||
|
f"страниц {len(indices)} (из {total_pages}) — "
|
||||||
|
f"индексы {[x+1 for x in indices]}"
|
||||||
|
)
|
||||||
|
total_added += len(indices)
|
||||||
|
|
||||||
|
output_path.parent.mkdir(parents=True, exist_ok=True)
|
||||||
|
with open(output_path, "wb") as out:
|
||||||
|
writer.write(out)
|
||||||
|
|
||||||
|
print(f"\nГотово: {output_path} ({total_added} стр.)")
|
||||||
|
|
||||||
|
|
||||||
|
# ---------------------------------------------------------------------------
|
||||||
|
# Entrypoint
|
||||||
|
# ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
def main() -> None:
|
||||||
|
if len(sys.argv) > 1:
|
||||||
|
config_path = Path(sys.argv[1])
|
||||||
|
else:
|
||||||
|
config_path = Path("merge.yaml")
|
||||||
|
|
||||||
|
if not config_path.exists():
|
||||||
|
print(f"Конфиг не найден: {config_path}", file=sys.stderr)
|
||||||
|
print("Использование: uv run pdf_merge.py [config.yaml]", file=sys.stderr)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
print(f"Конфиг: {config_path}")
|
||||||
|
merge(config_path)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
23
run.sh
Executable file
23
run.sh
Executable file
@ -0,0 +1,23 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
# run.sh — запуск pdf_merge через uv
|
||||||
|
#
|
||||||
|
# Использование:
|
||||||
|
# ./run.sh # берёт merge.yaml из текущей директории
|
||||||
|
# ./run.sh my_config.yaml # указать конфиг явно
|
||||||
|
# ./run.sh path/to/config.yaml # любой путь
|
||||||
|
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
SCRIPT="$SCRIPT_DIR/pdf_merge.py"
|
||||||
|
|
||||||
|
# Проверяем наличие uv
|
||||||
|
if ! command -v uv &>/dev/null; then
|
||||||
|
echo "uv не найден. Установи: https://docs.astral.sh/uv/getting-started/installation/"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
CONFIG="${1:-merge.yaml}"
|
||||||
|
|
||||||
|
echo "=== pdf-merge ==="
|
||||||
|
uv run "$SCRIPT" "$CONFIG"
|
||||||
Loading…
x
Reference in New Issue
Block a user