Почему я получаю TypeError при передаче байтовой строки в magic.from_buffer ()? - PullRequest
Новая
       75

Почему я получаю TypeError при передаче байтовой строки в magic.from_buffer ()?

1 голос
/ 01 мая 2019

Я пытаюсь получить тип файла изображения из URL с помощью Python-magic's from_buffer, и он хорошо работает на практике, но не удается, когда я пытаюсь запустить unittest.

Это мойfunction: 

def get_uri_from_url(url):
    if url != '':
        response = requests.get(url)
        data_type = magic.from_buffer(response.content, mime=True)
        image_bytes = str(base64.b64encode(response.content).decode('utf-8'))
        return f'data:{data_type};base64,{image_bytes}'
    return url

И это пока что тест: 

    def test_get_uri_from_url(self):
        test_bytes = b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00\x05\x00\x00\x00\x05\x08\x06\x00\x00\x00\x8do&' \
                     b'\xe5\x00\x00\x00\x1cIDAT\x08\xd7c\xf8\xff\xff?\xc3\x7f\x06 \x05\xc3 \x12\x84\xd01\xf1\x82X\xcd' \
                     b'\x04\x00\x0e\xf55\xcb\xd1\x8e\x0e\x1f\x00\x00\x00\x00IEND\xaeB`\x82'

        class Requests:
            content = test_bytes

            def get(self, url):
                return self

        with patch('path_to_my_function.requests', return_value=Requests):
            self.assertEqual(
                get_uri_from_url('any_url'),
                f'data:image/png;base64,{test_bytes}'
            )

Этот тест требует много времени для запуска и, наконец, выдает ошибки со следующей трассировкой: 

Traceback (most recent call last):
  File "/<path>/test_file.py", line 90, in test_get_uri_from_url
    get_uri_from_url('any_url'),
  File "/<path>/file.py", line 165, in get_uri_from_url
    data_type = magic.from_buffer(response.content, mime=True)
  File "/<path_to_python>/python3.6/site-packages/magic.py", line 148, in from_buffer
    return m.from_buffer(buffer)
  File "/<path_to_python>/python3.6/site-packages/magic.py", line 80, in from_buffer
    return maybe_decode(magic_buffer(self.cookie, buf))
  File "/<path_to_python>/python3.6/site-packages/magic.py", line 255, in magic_buffer
    return _magic_buffer(cookie, buf, len(buf))
ctypes.ArgumentError: argument 2: <class 'TypeError'>: wrong type

Я не вижу, что я делаю неправильно, когда я запускаю это в терминале, оно работает нормально, я должен что-то упустить. 

In [17]: response = requests.get('https://upload.wikimedia.org/wikipedia/commons/3/31/Red-dot-5px.png')                                                                           

In [18]: response.content                                                                                                                                                         
Out[18]: b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00\x05\x00\x00\x00\x05\x08\x06\x00\x00\x00\x8do&\xe5\x00\x00\x00\x1cIDAT\x08\xd7c\xf8\xff\xff?\xc3\x7f\x06 \x05\xc3 \x12\x84\xd01\xf1\x82X\xcd\x04\x00\x0e\xf55\xcb\xd1\x8e\x0e\x1f\x00\x00\x00\x00IEND\xaeB`\x82'

In [19]: magic.from_buffer(response.content, mime=True)                                                                                                                           
Out[19]: 'image/png'

1 Ответ

0 голосов
/ 01 мая 2019

Мне удалось заставить его работать, используя следующее: 

    def test_get_uri_from_url(self):

        class Response:
            content = b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00\x05\x00\x00\x00\x05\x08\x06\x00\x00\x00\x8do&' \
                      b'\xe5\x00\x00\x00\x1cIDAT\x08\xd7c\xf8\xff\xff?\xc3\x7f\x06 \x05\xc3 \x12\x84\xd01\xf1\x82X' \
                      b'\xcd\x04\x00\x0e\xf55\xcb\xd1\x8e\x0e\x1f\x00\x00\x00\x00IEND\xaeB`\x82'

        with patch('touchsurgery.apps.apiv3.helpers.requests.get', return_value=Response):
            self.assertEqual(
                get_uri_from_url('any_url'),
                'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAYAAACNbyblAAAAHElEQVQI12P4//8/w38GIAXDIBKE0DHx'
                'gljNBAAO9TXL0Y4OHwAAAABJRU5ErkJggg=='
            )

Я считаю, что проблема была в моих издевательствах и исправлениях.Поскольку я пропатчил возвращаемое значение 'request', но сам запрос никогда не вызывался явно (), объект, передаваемый в magic.from_buffer, был просто объектом magicmock, поэтому он не прошел.

Эта функция теперь работает, потому что get() явно вызывается в функции.Если кто-то еще сможет пролить свет на это или дать более подробное объяснение, это будет с благодарностью.

...