创建后访问bytesIO对象

Question

I am working on a scrapy spider, trying to extract text multiple pdfs in a directory, using slate ( https://pypi.python.org/pypi/slate ). I have no interest in saving the actual PDF to disk , and so I've been advised to look into the io.bytesIO subclass at https://docs.python.org/2/library/io.html#buffered-streams . Based on Creating bytesIO object , I have initialized the bytesIO class with the pdf body, but now I need to pass the data to the slate module. So far I have:

def save_pdf(self, response):
    in_memory_pdf = BytesIO(response.body)

    with open(in_memory_pdf, 'rb') as f:
        doc = slate.PDF(f)
        print(doc[0])

I'm getting:

in_memory_pdf.read(response.body)
TypeError: integer argument expected, got 'str'

How can I get this working?

edit:

with open(in_memory_pdf, 'rb') as f:
TypeError: coercing to Unicode: need string or buffer, _io.BytesIO found

edit 2:

def save_pdf(self, response):
    in_memory_pdf = BytesIO(bytes(response.body))
    in_memory_pdf.seek(0)
    doc = slate.PDF(in_memory_pdf)
    print(doc)

Answer 1

You already know the answer. It is clearly mentioned in the Python TypeError message and clear from the documentation:

class io.BytesIO([initial_bytes])

BytesIO accepts bytes. And you are passing it contents. ie: response.body which is a string.