示例#1
0
 def test(t):
     try:    # Known bug where first request is rejected
         t.status()
     except APIError:
         pass
     t.clear(type='single', timeout=10)
     t.backup('tmp.tar.gz',backup_type='incremental')
     t.insert({1: {'text': 'aste', 'title': 'lol'}, 2: '<item/>'})
     t.insert(['<page><id>3</id><text>foobar</text></page>','<page><id>4</id></page>'], fully_formed=True)
     t.insert({5: '<text>foobar</text>',6: '<text>baz</text>'})
     t.insert([{'page':{'id':7}},{'page':{'id':8}},{'page':{'id':9}}], fully_formed=True)
     t.replace({'page':{'id':3, 'title': 'one'}}, fully_formed=True)
     t.replace({4:{ 'title': 'two'}})
     t.replace({3: {'title': 'one prom'},4: {'title': 'two prim'}})
     t.partial_replace({'page':{'id':5, 'tabs': 'ss'}}, fully_formed=True)
     t.update({4: {'text':'lorem'}})
     t.update({34: {'title':'Far fetched.', 'text': 'Something.'}})
     t.delete([3,1])
     t.search(query.terms_from_dict({'title':'lorem'}))
     t.search_delete({'title':'lorem'})
     t.insert({11: {'title':'test','text':'11-test'}, 12: {'title':'test','text':'12-test'},\
             13: {'title':'test','text':'13-test'},14: {'title':'test','text':'14-test'},\
             15:{'title':'test','text':'15-test'}})
     t.search({'title':'test'}, docs=3, offset=1, list={'text':'yes', 'title': 'no'})
     t.retrieve([11,12])
     t.similar(12, 2, 1, docs=2)
     t.lookup([12,15], list={'title': 'no'})
     t.alternatives(query.term('test'), cr=1, idif=1, h=1)
     t.list_words(query.term('test'))
     t.list_last(docs=1, offset=0)
     t.list_first(list={'text':'yes'}, docs=1, offset=0)
     t.reindex()
     t.status()
示例#2
0
    def set_documents(self, documents, fully_formed=False):
        """ Wrap documents in the correct root tags, add id fields and convert them to xml strings.

            Args:
                documents -- If fully_formed is False (default), accepts dict where keys are document ids and values can be ether
                            xml string, etree.ElementTree or dict representation of an xml document (see dict_to_etree()).
                            If fully_formed is True, accepts list or single document where ids are integrated in document or
                            not needed and document has the right root tag.

            Keyword args:
                fully_formed  -- If documents are fully formed (contains the right root tags and id fields) set to True
                            to avoid the owerhead of documets beeing parsed at all. If set to True only list of documents or
                            a single document can be pased as 'documents', not a dict of documents. Default is False.
        """
        def add_id(document, id):
            def make_id_tag(root, rel_path, max_depth):
                if max_depth < 0:
                    raise ParameterError("document_id_xpath too deep!")
                if not rel_path:
                    return root
                else:
                    child = root.find(rel_path[0])
                    if child is None:
                        child = ET.Element(rel_path[0])
                        root.append(child)
                    return make_id_tag(child, rel_path[1:], max_depth - 1)

            make_id_tag(document, doc_id_xpath, 10).text = str(id)

        if fully_formed:  # documents is a list or single document that contians root tags and id fields.
            if not isinstance(documents, list):
                documents = [documents]
        else:  # documents is dict with ids as keys and documents as values.
            doc_root_tag = self.connection.document_root_xpath  # Local scope is faster.
            doc_id_xpath = self.connection.document_id_xpath.split('/')
            # Convert to etrees.
            documents = dict([
                (id,
                 to_etree((document if document is not None else query.term(
                     '', doc_root_tag)), doc_root_tag))
                for id, document in documents.items()
            ])  # TODO: possibly ineficient
            # If root not the same as given xpath, make new root and append to it.
            for id, document in documents.items():
                if document.tag != doc_root_tag:
                    documents[id] = ET.Element(doc_root_tag)
                    documents[id].append(
                        document)  # documents is still the old reference
            # Insert ids in documents and collapse to a list of documents.
            for id, document in documents.items():
                add_id(document, id)
            documents = documents.values()
        self._documents = map(to_raw_xml, documents)
示例#3
0
文件: request.py 项目: Pecisk/pycps
    def set_documents(self, documents, fully_formed=False):
        """ Wrap documents in the correct root tags, add id fields and convert them to xml strings.

            Args:
                documents -- If fully_formed is False (default), accepts dict where keys are document ids and values can be ether
                            xml string, etree.ElementTree or dict representation of an xml document (see dict_to_etree()).
                            If fully_formed is True, accepts list or single document where ids are integrated in document or
                            not needed and document has the right root tag.

            Keyword args:
                fully_formed  -- If documents are fully formed (contains the right root tags and id fields) set to True
                            to avoid the owerhead of documets beeing parsed at all. If set to True only list of documents or
                            a single document can be pased as 'documents', not a dict of documents. Default is False.
        """
        def add_id(document, id):
            def make_id_tag(root, rel_path, max_depth):
                if max_depth < 0:
                    raise ParameterError("document_id_xpath too deep!")
                if not rel_path:
                    return root
                else:
                    child = root.find(rel_path[0])
                    if child is None:
                        child = ET.Element(rel_path[0])
                        root.append(child)
                    return make_id_tag(child, rel_path[1:], max_depth - 1)
            make_id_tag(document, doc_id_xpath, 10).text = str(id)

        if fully_formed: # documents is a list or single document that contians root tags and id fields. 
            if not isinstance(documents, list):
                documents = [documents]
        else: # documents is dict with ids as keys and documents as values.
            doc_root_tag = self.connection.document_root_xpath  # Local scope is faster.
            doc_id_xpath = self.connection.document_id_xpath.split('/')
            # Convert to etrees.
            documents = dict([(id, to_etree((document if document is not None else
                                             query.term('', doc_root_tag)), doc_root_tag))
                             for id, document in documents.items()])     # TODO: possibly ineficient
            # If root not the same as given xpath, make new root and append to it.
            for id, document in documents.items():
                if document.tag != doc_root_tag:
                    documents[id] = ET.Element(doc_root_tag)
                    documents[id].append(document)  # documents is still the old reference
            # Insert ids in documents and collapse to a list of documents.
            for id, document in documents.items():
                add_id(document, id)
            documents = documents.values()
        self._documents = map(to_raw_xml, documents)
示例#4
0
 def set_list(self, value):
     if value is not None:
         self._content['list'] = '\n'.join(
             [query.term(value, key) for (key, value) in value.items()])
示例#5
0
文件: request.py 项目: Pecisk/pycps
 def set_list(self, value):
     if value is not None:
         self._content['list'] = '\n'.join([query.term(value, key) for (key, value) in value.items()])