o NDiz@sddlZddlmZmZddlmZddlmZddlm Z ddl m Z ddl mZmZmZeeZgdZGdd d eZGd d d ee ZGd d d ee ZeZeZdddZGdddeZGdddeZdS)N)SimpleGraphStatement GraphProtocol)EXEC_PROFILE_GRAPH_DEFAULT)GraphTraversal)GraphSONWriter)GremlinUserTypeIOdse_graphson2_serializersdse_graphson3_serializers)TraversalBatch_query_from_traversal_DefaultTraversalBatchcs<eZdZfddZddZddZddZd d ZZS) _GremlinGraphSONWriterAdapterc s&tt|jdi|||_d|_dS)N)superr __init__context user_types)selfrkwargs __class__r_/var/www/Datamplify/venv/lib/python3.10/site-packages/cassandra/datastax/graph/fluent/_query.pyr#s z&_GremlinGraphSONWriterAdapter.__init__cCs ||SN)toDict)rvalue_rrr serialize( z'_GremlinGraphSONWriterAdapter.serializecCs d}z |jt|}Wnty%|jD] \}}t||r"|}qYnw|jrn|jdurRz|jdj|jd}tt t ||_Wn tyQi|_Ynwt |jvrc||jt urct |dsl|snt||jvrnt }|rz||}W|StyY|Sw|S)Ncluster graph_name_fields) serializerstypeKeyErroritems isinstancerr _user_typesdictmapreversedtuplehasattrrget_specialized_serializerAttributeError)rr serializerkeyserrrrrget_serializer+s@     "  z,_GremlinGraphSONWriterAdapter.get_serializercCs||}|r |||S|Sr)r1dictify)robjr.rrrrKs z$_GremlinGraphSONWriterAdapter.toDictcCs||}|||Sr)r1 definition)rrr.rrrr4Os  z(_GremlinGraphSONWriterAdapter.definition) __name__ __module__ __qualname__rrr1rr4 __classcell__rrrrr !s   r c@ eZdZdS)GremlinGraphSON2WriterNr5r6r7rrrrr:Tr:c@r9)GremlinGraphSON3WriterNr;rrrrr=Xr<r=cCsz|tjkr t|td}n|tjkr |durtdt|td}ntd|z| |}W|St y<t dw)z From a GraphTraversal, return a query string. :param traversal: The GraphTraversal object :param graphson_protocol: The graph protocol to determine the output format. )serializer_mapNz5Missing context for GraphSON3 serialization requires.zUnknown graph protocol: {}z)Error preparing graphson traversal query:) r GRAPHSON_2_0graphson2_writerr GRAPHSON_3_0 ValueErrorgraphson3_writerr format writeObject Exceptionlog exception) traversalgraph_protocolrgraphson_writerqueryrrrr `s    r c@sdeZdZdZdZdZdddZddZddZd d Z e j fd d Z d dZ ddZddZeZdS)r a$ A `TraversalBatch` is used to execute multiple graph traversals in a single transaction. If any traversal in the batch fails, the entire batch will fail to apply. If a TraversalBatch is bounded to a DSE session, it can be executed using `traversal_batch.execute()`. NcCs||_||_dS)z :param session: (Optional) A DSE session :param execution_profile: (Optional) The execution profile to use for the batch execution N)_session_execution_profile)rsessionexecution_profilerrrrs zTraversalBatch.__init__cCt)zc Add a traversal to the batch. :param traversal: A gremlin GraphTraversal NotImplementedErrorrrIrrraddzTraversalBatch.addcCrQ)z~ Adds a sequence of traversals to the batch. :param traversals: A sequence of gremlin GraphTraversal rR)r traversalsrrradd_allrVzTraversalBatch.add_allcCrQ)zL Execute the traversal batch if bounded to a `DSE Session`. rRrrrrexecutezTraversalBatch.executecCrQ)z Return the traversal batch as GraphStatement. :param graph_protocol: The graph protocol for the GraphSONWriter. Default is GraphProtocol.GRAPHSON_2_0. rR)rrJrrras_graph_statementrVz!TraversalBatch.as_graph_statementcCrQ)z4 Clear a traversal batch for reuse. rRrYrrrclearr[zTraversalBatch.clearcCrQrrRrYrrr__len__szTraversalBatch.__len__cCsdt|S)Nz)rDlenrYrrr__str__szTraversalBatch.__str__)NN)r5r6r7__doc__rMrNrrUrXrZrr?r\r]r^r`__repr__rrrrr ys  r csXeZdZdZfddZddZddZejdfdd Z d d Z d d Z ddZ Z S)r Ncs tt|j|i|g|_dSr)rr r _traversals)rargsrrrrrs z_DefaultTraversalBatch.__init__cCs"t|ts td|j||S)Nz,traversal should be a gremlin GraphTraversal)r%rrBrcappendrTrrrrUs  z_DefaultTraversalBatch.addcCs|D]}||qdSr)rU)rrWrIrrrrXs z_DefaultTraversalBatch.add_allcs.fdd|jD}dd|}t|S)Ncsg|]}t|qSr)r ).0trrJrr sz=_DefaultTraversalBatch.as_graph_statement..z[{0}],)rcrDjoinr)rrJr statementsrLrrhrr\sz)_DefaultTraversalBatch.as_graph_statementcCsr|jdur td|jr|jnt}|j|j}|jj|jd}|jr+|j |j|dn|j |d}|jj ||dS)Nz>A DSE Session must be provided to execute the traversal batch.)rr)r)rP) rMrBrNrget_execution_profile graph_optionsrrrJr\ execute_graph)rrPrnr statementrrrrZs  z_DefaultTraversalBatch.executecCs|jdd=dSr)rcrYrrrr]sz_DefaultTraversalBatch.clearcCs t|jSr)r_rcrYrrrr^rz_DefaultTraversalBatch.__len__)r5r6r7rcrrUrXrr?r\rZr]r^r8rrrrr s r r)loggingcassandra.graphrrcassandra.clusterr&gremlin_python.process.graph_traversalr(gremlin_python.structure.io.graphsonV2d0rGraphSONWriterV2(gremlin_python.structure.io.graphsonV3d0GraphSONWriterV3+cassandra.datastax.graph.fluent.serializersrrr getLoggerr5rG__all__objectr r:r=r@rCr r r rrrrs"     3 A