читать PDF-файл с URL-адреса в память, не сохраняя файл в локальный файл

Question

У меня проблема с пропуском сохранения файла в локальный файл. Моя проблема в том, как напрямую передать содержимое URL в модуль pdfplumber. Мой код выглядит следующим образом:

from pathlib import Path
import urllib3
import pdfplumber

def read_pdf(url: str) -> str:
    http = urllib3.PoolManager()
    temp = Path('temp.pdf')
    temp.write_bytes(http.request("GET", url).data)
    with pdfplumber.open(temp) as pdf:
        text = pdf.pages[0].extract_text()
    temp.unlink()
    return text

text = read_pdf('https://sample.com/test.pdf')

Он работает, но я хотел бы избежать сохранения файла на локальном компьютере, а делать это только в ОЗУ. Я пробовал io.BytesIO, но он не работает

Answer 1

Вот пример, который использует io.BytesIO() вместо создания любых файлов на жестком диске.
Обратите внимание, что pdfplumber.load(...) используется вместо pdfplumber.open(...).

import urllib3
import pdfplumber
import io


def read_pdf(url: str) -> str:
    http = urllib3.PoolManager()
    temp = io.BytesIO()
    temp.write(http.request("GET", url).data)
    with pdfplumber.load(temp) as pdf:
        text = pdf.pages[0].extract_text()
    return text

text = read_pdf('https://sample.com/test.pdf')