gremlinpython - вернуть идентификатор и пометить как строку - PullRequest
Новая
       14

gremlinpython - вернуть идентификатор и пометить как строку

1 голос
/ 03 мая 2019

Я использую gremlinpython 3.4.1 на Python 3.7.2, и когда получаю ответы вершин / ребер, он предоставляет <T.id: > для идентификатора и <T.label: 3> для метки. Как мне получить вместо него строковое значение для идентификатора и метки в ответе? Моя цель - взять вывод и сгенерировать JSON

Выход: 

python3 stackoverflow.py
[{'name': ['USA'], 'parentname': ['USA'], 'shortname': ['US'], <T.id: 1>: 'country-us', 'parentid': ['country-us'], <T.label: 3>: 'Country'}]

код: 

from gremlin_python import statics
from gremlin_python.process.anonymous_traversal import traversal
from gremlin_python.process.graph_traversal import __
from gremlin_python.process.strategies import *
from gremlin_python.driver.driver_remote_connection import DriverRemoteConnection
from gremlin_python.process.traversal import T
from gremlin_python.process.traversal import Order
from gremlin_python.process.traversal import Cardinality
from gremlin_python.process.traversal import Column
from gremlin_python.process.traversal import Direction
from gremlin_python.process.traversal import Operator
from gremlin_python.process.traversal import P
from gremlin_python.process.traversal import Pop
from gremlin_python.process.traversal import Scope
from gremlin_python.process.traversal import Barrier
from gremlin_python.process.traversal import Bindings
from gremlin_python.process.traversal import WithOptions

CLUSTER_ENDPOINT = "removed"
PORT = "8182"
g = traversal().withRemote(DriverRemoteConnection('wss://' + CLUSTER_ENDPOINT + ':' + PORT + '/gremlin','g'))

response = g.V().has('name', 'USA').limit(1000).hasLabel('Country').valueMap(True).toList()
print(response)

Кстати - я пытался использовать .with_(WithOptions.ids), например:

response = g.V().has('name', 'USA').limit(1000).hasLabel('Country').valueMap(True).with_(WithOptions.ids).toList()

, для которого я получаю следующую ошибку:

gremlin_python.driver.protocol.GremlinServerError: 599: {"requestId":"bf74df44-f064-4411-a1cb-78b30f9d2cf6","code":"InternalFailureException","detailedMessage":"Could not locate method: NeptuneGraphTraversal.with([1])"}

Ответы [ 3 ]

1 голос
/ 03 мая 2019

Кроме того, в приведенном примере project() вы можете сделать что-то вроде следующего, если не можете или не хотите указывать имена свойств: 

g.V().has('name', 'USA').limit(1000).hasLabel('Country').
  map(union(project('id','label').
              by(id).
              by(label),
            valueMap()).unfold().
      group().
        by(keys).
        by(select(values))) // select(values).unfold() if you only have single values
0 голосов
/ 03 мая 2019

Вы могли бы заменить свои ключи DICT EnumMeta фактическими значениями. Вам нужно добавить Unold () после вашего valueMap, чтобы использовать эту функцию. 

from gremlin_python.process.traversal import T

def get_query_result_without_enum_metas(query_result):
    return [replace_enum_metas(d) for d in query_result]

def replace_enum_metas(dict):
      dict_key = (*dict,)[0]
      if type(dict_key) is str:
        return dict
      elif type(dict_key) is T:
        return {dict_key.name: dict[dict_key]}

input: [{'vertex_property': ['Summary']}, {<T.id: 1>: '4b30f448ee2527204a050596b'}, {<T.label: 3>: 'VertexLabel'}]

output: [{'vertex_property': ['Summary']}, {'id': '4b30f448ee2527204a050596b'}, {'label': 'VertexLabel'}]
0 голосов
/ 03 мая 2019

Вы можете попробовать project результаты. 

g.V().has('name', 'USA').limit(1000).hasLabel('Country') \
    .project('id', 'label', 'name', 'parentname', 'shortname', 'parentid') \
    .by(id) \
    .by(label) \
    .by('name') \
    .by('parentname') \
    .by('shortname') \
    .by('parentid') \
    .toList()
...