o NDi- @sddlZddlmZddlmZddlmZmZddlm Z m Z m Z m Z ddl mZmZgdZdZed d eDZd ZGd d d eZGdddeZeD])ZedfddZedfddZedfddZeeedeeeeedqOGdddeZGdddeeZddZ ddZ!d d!Z"d"d#Z#Gd$d%d%eZ$Gd&d'd'e$Z%Gd(d)d)e$Z&e%Z'e&Z(Gd*d+d+eZ)dS),N)warn)ConsistencyLevel) StatementSimpleStatement)VertexEdgePathVertexProperty)GraphSON2ReaderGraphSON3Reader) GraphProtocol GraphOptionsGraphStatementSimpleGraphStatementsingle_object_row_factorygraph_result_row_factorygraph_object_row_factorygraph_graphson2_row_factoryResultgraph_graphson3_row_factory)) graph_namezname of the targeted graph.z graph-name) graph_sourcezAchoose the graph traversal source, configured on the server side.z graph-source)graph_languagez;the language used in the queries (default "gremlin-groovy")zgraph-language)graph_protocolzXthe graph protocol that the server should use for query results (default "graphson-1-0")z graph-results)graph_read_consistency_levelaread `cassandra.ConsistencyLevel `_ for graph queries (if distinct from session default). Setting this overrides the native `Statement.consistency_level `_ for read operations from Cassandra persistencegraph-read-consistency)graph_write_consistency_levelawrite `cassandra.ConsistencyLevel `_ for graph queries (if distinct from session default). Setting this overrides the native `Statement.consistency_level `_ for write operations to Cassandra persistence.graph-write-consistencyccs|]}|dVqdS)rN).0optionrrW/var/www/Datamplify/venv/lib/python3.10/site-packages/cassandra/datastax/graph/query.py )sr"zrequest-timeoutc@seZdZdZ dZ dZdS)r s graphson-1.0s graphson-2.0s graphson-3.0N)__name__ __module__ __qualname__ GRAPHSON_1_0 GRAPHSON_2_0 GRAPHSON_3_0rrrr!r /sr c@seZdZdZejZdZddZddZ ddZ dd d Z d d Z ddZ ddZddZeddZeddZeddZd S)r z. Options for DSE Graph Query handler. sgremlin-groovycKsXi|_|dd|dtj|D]\}}|tvr#td|t|||qdS)Nrgrz7Unknown keyword argument received for GraphOptions: {0}) _graph_options setdefaultr DEFAULT_GRAPH_LANGUAGEitems_graph_option_namesrformatsetattr)selfkwargsattrvaluerrr!__init__Js zGraphOptions.__init__cCst}|j|_|SN)r r*copy)r1 new_optionsrrr!r7Ss zGraphOptions.copycCs|j|jdSr6)r*update)r1optionsrrr!r9XzGraphOptions.updateNcCsL|j}|r ||jdD]}||}|dur#tj|||<q|S)zx Returns a map for these options updated with other options, and mapped to graph payload types. )rrN)r*r7r9getr value_to_nameencode)r1 other_optionsr:clcl_enumrrr!get_options_map[s   zGraphOptions.get_options_mapcC d|_dS)zb Sets ``graph_source`` to the server-defined default traversal source ('default') defaultNrr1rrr!set_source_defaultk zGraphOptions.set_source_defaultcCrC)z] Sets ``graph_source`` to the server-defined analytic traversal source ('a') aNrErFrrr!set_source_analyticsqrHz!GraphOptions.set_source_analyticscCrC)zZ Sets ``graph_source`` to the server-defined graph traversal source ('g') r)NrErFrrr!set_source_graphwrHzGraphOptions.set_source_graphcCs ||_dS)z~ Sets ``graph_protocol`` as server graph results format (See :class:`cassandra.datastax.graph.GraphProtocol`) N)r)r1protocolrrr!set_graph_protocol}rHzGraphOptions.set_graph_protocolcCs |jdvS)N)sdefaultNrErFrrr!is_default_sources zGraphOptions.is_default_sourcecC |jdkS)zh True if ``graph_source`` is set to the server-defined analytics traversal source ('a') arErFrrr!is_analytics_source z GraphOptions.is_analytics_sourcecCrO)zd True if ``graph_source`` is set to the server-defined graph traversal source ('g') grErFrrr!is_graph_sourcerRzGraphOptions.is_graph_sourcer6)r#r$r%__doc__r r&DEFAULT_GRAPH_PROTOCOLr,r5r7r9rBrGrJrKrMpropertyrNrQrTrrrr!r As$    r cCs |j|Sr6)r*r<r1keyrrr!r< r<cCs:|durt|tr |}||j|<dS|j|ddSr6) isinstancestrr>r*pop)r1r4rZrrr!sets  r_cCs|j|ddSr6)r*r^rYrrr!deleter;r`c@s(eZdZdZeddZddZeZdS)rz. An abstract class representing a graph query.cCstr6)NotImplementedErrorrFrrr!queryzGraphStatement.querycCs d|jS)Nz)r/rcrFrrr!__str__r[zGraphStatement.__str__N)r#r$r%rUrWrcre__repr__rrrr!rs  rc@seZdZdZeddZdS)rz Simple graph statement for :meth:`.Session.execute_graph`. Takes the same parameters as :class:`.SimpleStatement`. cCs|jSr6) _query_stringrFrrr!rcrdzSimpleGraphStatement.queryN)r#r$r%rUrWrcrrrr!rsrcCdd|DS)z8 returns the JSON string value of graph results cSsg|]}|dqS)rrrrowrrr! sz-single_object_row_factory..r column_namesrowsrrr!rsrcCrh)z Returns a :class:`Result ` object that can load graph results and produce specific types. The Result JSON is deserialized and unpacked from the top-level 'result' dict. cSs"g|] }tt|ddqSrresult)rjsonloadsrirrr!rks"z,graph_result_row_factory..rrlrrr!rsrcCstdd|DS)a7 Like :func:`~.graph_result_row_factory`, except known element types (:class:`~.Vertex`, :class:`~.Edge`) are converted to their simplified objects. Some low-level metadata is shed in this conversion. Unknown result types are still returned as :class:`Result `. css"|] }t|ddVqdS)rrpN)rqrrrirrr!r"s z+graph_object_row_factory..)_graph_object_sequencerlrrr!rsrccsV|D]%}t|}t|tr%|jd}|dkr|}n|dkr%|}|VqdS)Ntypevertexedge)rr\dictr4r< as_vertexas_edge)objectsorestyprrr!rss   rsc@s$eZdZdZdZddZddZdS)_GraphSONContextRowFactoryNcCs*d|i}|jpi}|j|fi||_dS)Ncluster)graphson_reader_kwargsgraphson_reader_classgraphson_reader)r1rcontextr2rrr!r5s z#_GraphSONContextRowFactory.__init__csfdd|DS)Ncs g|] }j|ddqSro)rreadrirFrr!rks z7_GraphSONContextRowFactory.__call__..r)r1rmrnrrFr!__call__r;z#_GraphSONContextRowFactory.__call__)r#r$r%rrr5rrrrr!r~s  r~c@eZdZdZeZdS)_GraphSON2RowFactoryz-Row factory to deserialize GraphSON2 results.N)r#r$r%rUr rrrrr!rrc@r)_GraphSON3RowFactoryz-Row factory to deserialize GraphSON3 results.N)r#r$r%rUr rrrrr!rrrc@sfeZdZdZdZ ddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZdS)rzl Represents deserialized graph results. Property and item getters are provided for convenience. NcCs ||_dSr6r4)r1r4rrr!r5  zResult.__init__cCs6t|jts td||jvr|j|Std|f)Nz"Value cannot be accessed as a dictz$Result has no top-level attribute %r)r\r4rw ValueErrorAttributeError)r1r3rrr! __getattr__ s   zResult.__getattr__cCsNt|jtrt|tr|j|St|jtr t|tr |j|Std|f)NzResult cannot be indexed by %r)r\r4rwr]listintr)r1itemrrr! __getitem__s   zResult.__getitem__cCs t|jSr6)r]r4rFrrr!rerzResult.__str__cCsdtj|jfS)Nz%s(%r))rr#r4rFrrr!rf!szResult.__repr__cCs |j|jkSr6r)r1otherrrr!__eq__$r[z Result.__eq__c CsDzt|j|j|j|jdiWStttfy!td|fw)z Return a :class:`Vertex` parsed from this result Raises TypeError if parsing fails (i.e. the result structure is not valid). propertieszCould not create Vertex from %r) ridlabelrtr4r<rr TypeErrorrFrrr!rx's  zResult.as_vertexc CsTzt|j|j|j|jdi|j|j|j|j WSt t t fy)t d|fw)z Return a :class:`Edge` parsed from this result Raises TypeError if parsing fails (i.e. the result structure is not valid). rzCould not create Edge from %r) rrrrtr4r<inVinVLabeloutV outVLabelrrrrFrrr!ry2szResult.as_edgec Cs4zt|j|jWStttfytd|fw)z Return a :class:`Path` parsed from this result Raises TypeError if parsing fails (i.e. the result structure is not valid). zCould not create Path from %r)rlabelsrzrrrrFrrr!as_path>s zResult.as_pathcCs&t|jd|jd|jdiS)Nrr4r)r r4r<rFrrr!as_vertex_propertyIs&zResult.as_vertex_property)r#r$r%rUr4r5rrrerfrrxryrrrrrr!rs   r)*rqwarningsr cassandrarcassandra.queryrrcassandra.datastax.graph.typesrrrr !cassandra.datastax.graph.graphsonr r __all__r*tupler._request_timeout_keyobjectr r optr<r_r`r0rWrrrrrrsr~rrrrrrrrr!s:   U