mirror of
https://github.com/snapotter-hq/SnapOtter.git
synced 2026-08-03 07:46:42 +02:00
32 lines
900 B
Python
32 lines
900 B
Python
"""Extract plain text. Args: {"path": in, "out": out-txt-path}. Prints {"chars": N}."""
|
|
import json
|
|
import sys
|
|
|
|
|
|
def main():
|
|
args = json.loads(sys.argv[1]) if len(sys.argv) > 1 else {}
|
|
path, out = args.get("path"), args.get("out")
|
|
if not path or not out:
|
|
print(json.dumps({"error": "missing path/out"}))
|
|
sys.exit(1)
|
|
try:
|
|
import fitz
|
|
except ImportError:
|
|
print(json.dumps({"error": "PyMuPDF not installed"}))
|
|
sys.exit(1)
|
|
try:
|
|
doc = fitz.open(path)
|
|
parts = [page.get_text() for page in doc]
|
|
doc.close()
|
|
text = "\n".join(parts)
|
|
with open(out, "w", encoding="utf-8") as fh:
|
|
fh.write(text)
|
|
print(json.dumps({"chars": len(text)}))
|
|
except Exception as exc: # noqa: BLE001
|
|
print(json.dumps({"error": str(exc)}))
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|