使用python驱动程序

时间:2016-12-23 13:56:57

标签: python cassandra tuples

大家,

我在通过DataStax提供的python-driver将数据插入Cassandra表中的元组值字段时遇到问题。

要么我没有得到如何正确地将元组参数传递给Session.execute命令,要么驱动程序在将内容转换为Cassandra时以错误的方式在内部转换元组 - 因为相同的插入在cqlsh会话中执行时效果很好。

不涉及元组而是列表的等效插入在代码和cqlsh中都可以正常工作。

我正在使用Python 2.7.10和cassandra-driver 3.7.1。 Python执行中引发的错误是InvalidRequest: Error from server: code=2200 [Invalid query] message="Invalid list literal for tuplefield of type frozen<tuple<int, int, int>>"

我粘贴了一个最小的工作代码,可以重现我看到的问题。有人可以帮我弄清楚我是否做错了什么?

(注意:我已经尝试将普通列表传递给Session.execute代替元组参数,但也没有运气。)

非常感谢。

'''
    Run with:
        python 2.7.10
        cassandra-driver==3.7.1 installed
'''

from cassandra.cluster import Cluster

if __name__=='__main__':

    serverAddress='SERVER_ADDRESS'
    keyspacename='tupletests'
    tablecreation='''create table tabletest (
                            rowid int,
                            tuplefield tuple < int, int, int >,
                            listfield list < int >,
                        primary key (rowid)
                    );'''

    # connect and create keyspace
    clu=Cluster([serverAddress])
    session=clu.connect()
    session.execute("create keyspace %s with replication = \
        {'class': 'SimpleStrategy', 'replication_factor': 1};" % keyspacename)

    # use keyspace; create a sample table
    session.set_keyspace(keyspacename)
    session.execute(tablecreation)

    # insert a row with rowid,listfield
    session.execute('insert into tabletest (rowid, listfield) values (%s,%s)', \
        [999, [10,11,12]])
    # succeeds

    # insert a row with rowid,tuplefield
    session.execute('insert into tabletest (rowid, tuplefield) values (%s,%s)', \
        [111, tuple([6,5,4])])
    # raises: *** InvalidRequest: Error from server: code=2200 [Invalid query]
    # message="Invalid list literal for tuplefield of type frozen<tuple<int, int, int>>"

    # Compare with analogous statements in cqlsh, which *both* succeed:
    # List:    insert into tabletest (rowid, listfield) values (999, [21,22,23]);
    # Tuple:   insert into tabletest (rowid, tuplefield) values (765, (121,122,123));

    # delete test keyspace
    session.execute("drop keyspace %s;" % keyspacename)    
    print "Test Finished."

1 个答案:

答案 0 :(得分:4)

好的,我刚刚解决了这个问题。在这里为每个人发布解决方案。 为了让驱动程序知道在将元组传递给Cassandra时应该如何格式化元组,如果坚持对插入使用非预处理语句,则必须指定在调用元素之前解析元组时使用的编码器:

session.encoder.mapping[tuple] = session.encoder.cql_encode_tuple

在此语句之后,出现在Session.execute插入语句中的元组会以正确的方式自动转换为Cassandra,而Cassandra不会再抱怨了。

(据我所知,最好采用Prepared语句,其中包含足够的信息,使得单独指定编码器变得多余。)