Проблемы с добавлением данных в Solr с помощью Python?

Question

Я использую Python 3.4.3 и Sunburnt, чтобы добавить документ в Solr(5.2.1). Следующий код прямо из документации Sunburnt:

import sunburnt

si=sunburnt.SolrInterface("http://localhost:8983/solr/")

document = {"id":"0553573403",
        "cat":"book",
        "name":"A Game of Thrones",
        "price":7.99,
        "inStock": True,
        "author_t":
        "George R.R. Martin",
        "series_t":"A Song of Ice and Fire",
        "sequence_i":1,
        "genre_s":"fantasy"}

si.add(document)

и когда я запускаю вышеупомянутую команду, я получаю следующее:

NameError                                 Traceback (most recent call last)
<ipython-input-1-1008a9ce394f> in <module>()----> 1 import sunburnt
  2 
  3 si= sunburnt.SolrInterface("http://localhost:8983/solr/")
  4 
  5 document = {"id":"0553573403",

/Users/rmohan/venv_py3/lib/python3.4/site-packages/sunburnt/__init__.py in <module>()
  1 from __future__ import absolute_import
  2 
----> 3 from .strings import RawString
  4 from .sunburnt import SolrError, SolrInterface
  5 

/Users/rmohan/venv_py3/lib/python3.4/site-packages/sunburnt/strings.py in <module>()
  2 
  3 
----> 4 class SolrString(unicode):
  5     # The behaviour below is only really relevant for String fields rather
  6     # than Text fields - most queryparsers will strip these characters out

NameError: name 'unicode' is not defined

Поэтому я попробовал тот же документ с pysolr следующим образом:

import pysolr
solr = pysolr.Solr('http://localhost:8983/solr/', timeout=10)

document = [{"id":"0553573403",
        "cat":"book",
        "name":"A Game of Thrones",
        "price":7.99,
        "inStock": True,
        "author_t":
        "George R.R. Martin",
        "series_t":"A Song of Ice and Fire",
        "sequence_i":1,
        "genre_s":"fantasy"}]

solr.add(document)

которые дают следующее:

/Users/rmohan/venv_py3/lib/python3.4/site-packages/pysolr.py in _scrape_response(self, headers, response)
443         dom_tree = None
444 
--> 445         if response.startswith('<?xml'):
446             # Try a strict XML parse
447             try:

TypeError: startswith first arg must be bytes or a tuple of bytes, not str

Я немного погуглил, но не смог найти однозначного ответа о том, как решить проблему Unicode или байтов с входными данными. Я пытался преобразовать строки в байты и юникод, но ничего не помогло.

Если кто-то знает лучший способ вставить документы в SOLR, пожалуйста, поделитесь. Любая помощь будет оценена.

Answer 1

Я разобрался, файл pysolr.py имеет

if response.startswith('<?xml'):

который должен быть изменен на

if response.startswith(b'<?xml'):

Дополнительная информация по адресу: https://github.com/toastdriven/pysolr/issues/159